Mitigating domain shift in AI-based tuberculosis screening with unsupervised domain adaptation

by   Nishanjan Ravin, et al.

We demonstrate that Domain Invariant Feature Learning (DIFL) can improve the out-of-domain generalizability of a deep learning Tuberculosis 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 Tuberculosis 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 Tuberculosis 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 Tuberculosis screening while maintaining acceptable accuracy over the source domain imagery when applied across a variety of public datasets.


page 4

page 5

page 6

page 8

page 9

page 10

page 11

page 14


Unsupervised domain adaptation for medical imaging segmentation with self-ensembling

Recent deep learning methods for the medical imaging domain have reached...

Data Augmentation-Based Unsupervised Domain Adaptation In Medical Imaging

Deep learning-based models in medical imaging often struggle to generali...

Theoretic Analysis and Extremely Easy Algorithms for Domain Adaptive Feature Learning

Domain adaptation problems arise in a variety of applications, where a t...

Structure Preserving Cycle-GAN for Unsupervised Medical Image Domain Adaptation

The presence of domain shift in medical imaging is a common issue, which...

Weakly Supervised Domain Adaptation for Built-up Region Segmentation in Aerial and Satellite Imagery

This paper proposes a novel domain adaptation algorithm to handle the ch...

Learning Transferable Parameters for Unsupervised Domain Adaptation

Unsupervised domain adaptation (UDA) enables a learning machine to adapt...

Inclusive FinTech Lending via Contrastive Learning and Domain Adaptation

FinTech lending (e.g., micro-lending) has played a significant role in f...

Please sign up or login with your details

Forgot password? Click here to reset