We demonstrate that Domain Invariant Feature Learning (DIFL) can improve the out-ofdomain generalizability of a deep learning Tuberculosis (TB) screening algorithm. It is well known that state of the art deep learning algorithms often have difficulty generalizing to unseen data distributions due to ''domain shift.'' In the context of medical imaging, this could lead to unintended biases such as the inability to generalize from one patient population to another. We analyze the performance of a ResNet-50 classifier for the purposes of TB screening using the four most popular public datasets with geographically diverse sources of imagery. We show that without domain adaptation, ResNet-50 has difficulty in generalizing between imaging distributions from a number of public TB screening datasets with imagery from geographically distributed regions. However, with the incorporation of DIFL, the out-of-domain performance is greatly enhanced. Analysis criteria includes a comparison of accuracy, sensitivity, specificity and AUC over both the baseline, as well as the DIFL enhanced algorithms. We conclude that DIFL improves generalizability of TB screening while maintaining acceptable accuracy over the source domain imagery when applied across a variety of public datasets.INDEX TERMS Tuberculosis, X-ray imaging, domain adaptation, domain invariant feature learning, generative adversarial networks, deep learning, computer vision, computer aided diagnosis, DIFL.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.
hi@scite.ai
10624 S. Eastern Ave., Ste. A-614
Henderson, NV 89052, USA
Copyright © 2024 scite LLC. All rights reserved.
Made with 💙 for researchers
Part of the Research Solutions Family.