Out-of-Distribution Detection for Learning-Based Chest X-Ray Diagnosis.

Wenlong Chen,Chuanwen Feng,Ao Ke,Xike Xie,S. Kevin Zhou

DOI: https://doi.org/10.1109/ICASSP48485.2024.10447414

2024-01-01

Abstract:Deep learning has shown prominence in chest radiography interpretation, which is critical in evaluating various lung and chest diseases, such as pneumonia, emphysema, and tuberculosis. Deploying machine learning model, it is important to detect out-of-distribution (OOD) inputs, which are distinct from the training data. For example, a model trained on conventional pneumonia data may not be able to diagnose COVID-19 accurately, letting alone the vast body of possible abnormalities and variation in medical imaging. Thus, a trustworthy diagnosing model should be alerted with such OOD inputs to avoid the overconfidence with samples and therefore the risk of misdiagnosis and severe medical consequences. In this paper, we study the problem of OOD detection for deep learning-based chest X-ray imaging by proposing a distance-based OOD detection method. We show that our method (1) inherits the merits of distance-based methods in being applicable to different models and training losses; and (2) adopts probabilistic score functions with fewer statistics to mitigate the potential distribution assumption on training data. We conduct extensive experiments on benchmarks to demonstrate the superiority of our method. In particular, on challenging tasks, such as Chest X-ray 14 vs. COVID-19, our method achieves 12.62% lower FPR than the state-of-the-art solutions.