XDT-CXR: Investigating Cross-Disease Transferability in Zero-Shot Binary Classification of Chest X-Rays

Umaima Rahman,Abhishek Basu,Muhammad Uzair Khattak,Aniq Ur Rahman
2024-08-21
Abstract:This study explores the concept of cross-disease transferability (XDT) in medical imaging, focusing on the potential of binary classifiers trained on one disease to perform zero-shot classification on another disease affecting the same organ. Utilizing chest X-rays (CXR) as the primary modality, we investigate whether a model trained on one pulmonary disease can make predictions about another novel pulmonary disease, a scenario with significant implications for medical settings with limited data on emerging diseases. The XDT framework leverages the embedding space of a vision encoder, which, through kernel transformation, aids in distinguishing between diseased and non-diseased classes in the latent space. This capability is especially beneficial in resource-limited environments or in regions with low prevalence of certain diseases, where conventional diagnostic practices may fail. However, the XDT framework is currently limited to binary classification, determining only the presence or absence of a disease rather than differentiating among multiple diseases. This limitation underscores the supplementary role of XDT to traditional diagnostic tests in clinical settings. Furthermore, results show that XDT-CXR as a framework is able to make better predictions compared to other zero-shot learning (ZSL) baselines.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
### What problems does this paper attempt to solve? This paper aims to explore the concept of Cross - Disease Transferability (XDT) in medical imaging, especially to study whether a binary classifier trained on chest X - rays (CXR) can classify other lung diseases in the case of zero - shot learning. Specifically, the authors attempt to answer the following question: **Can a binary classifier trained on one disease use medical images of the same modality to perform zero - shot classification on another disease affecting the same organ?** #### Main problems and background 1. **Diagnostic challenges in resource - limited environments**: - In some resource - limited environments, such as medical institutions in developing countries or low - incidence areas, there may be a lack of sufficient data and laboratory facilities for traditional diagnosis. - In these cases, traditional diagnostic methods may fail, so a new method is needed to assist in diagnosis. 2. **Rapid response to new diseases**: - For newly emerging diseases (such as COVID - 19), there may not be enough training data to develop a specialized diagnostic model in the early stage. - If the existing models can be used to preliminarily classify new diseases, it will be helpful to quickly respond to public health emergencies. 3. **The potential of cross - disease transferability**: - Research shows that different diseases of the same organ may have visual similarities in images (Özger et al., 2020). Therefore, a model trained on a certain disease may have the ability to classify other similar diseases. #### Solutions To achieve this goal, the authors propose the XDT framework, and its core ideas are as follows: - **Embedding space and kernel transformation**: Generate an embedding space through a visual encoder and use kernel transformation to distinguish between disease and non - disease categories in the latent space. - **Zero - shot learning**: Use the knowledge of existing models to reason without specific training data for the target disease. - **Application scenarios**: Especially in resource - limited environments or in areas with a low incidence of certain diseases, XDT can be used as an auxiliary screening tool to help doctors quickly classify patients and optimize resource allocation. #### Limitations and complementary roles Although the XDT framework has shown superior performance, it is currently limited to binary classification tasks (i.e., determining whether a disease exists, rather than distinguishing between multiple diseases). Therefore, it is more of a complementary tool to traditional diagnostic tests, providing additional support in the clinical environment. In this way, the XDT framework can not only provide effective diagnostic support in resource - limited situations, but also provide new ideas for the early identification and management of new diseases.