From Classification to Localization and Clinical Validation: Large-Scale Development of a Deep Learning System for Thoracic Disease Detection on Chest Radiographs in Thailand

· Source: cs.CV updates on arXiv.org · Field: Science & Research — Health & Medical Research, Mathematics & Computational Sciences · Depth: Advanced, extended

Summary

Inspectra CXR version 5 is a deep learning system developed for multi-label thoracic disease classification and weakly supervised lesion localization on chest radiographs in Thailand. Built on a DenseNet-121 backbone augmented with Attend-and-Compare Modules (ACM) and a Probabilistic Class Activation Map (PCAM) aggregation layer, the model was trained on 874,858 frontal chest radiographs from Siriraj Hospital. On a held-out in-domain test set of 19,871 cases, it achieved a mean AUROC of 0.994, with 92.4% sensitivity and 98.6% specificity across nine conditions. The system demonstrated robust cross-site generalization, maintaining a mean AUROC of 0.970 on 5,992 cases from 13 diverse Thai hospitals. For localization, it attained a mean lesion-localization fraction (LLF) of 77.9% on 4,549 radiologist-annotated cases. Clinical usability evaluation with five thoracic radiologists showed 93.6% classification concordance, 94.7% localization concordance, and an "excellent" mean System Usability Scale (SUS) score of 89.

Key takeaway

For Machine Learning Engineers developing medical imaging AI, this study demonstrates that focusing on local data adaptation and comprehensive validation is crucial. You should prioritize multi-task architectures that provide both classification and weakly supervised localization, as these enhance clinical utility and radiologist trust. Ensure your validation includes diverse real-world sites and formal usability assessments like the SUS to prove practical applicability and acceptance.

Key insights

A locally adapted deep learning system for CXR achieves high accuracy, cross-site generalization, and radiologist trust through simultaneous classification and localization.

Principles

Method

The system uses a DenseNet-121 + ACM encoder, external attention decoder, and PCAM aggregation layer for multi-task classification and weakly supervised localization.

In practice

Topics

Best for: Computer Vision Engineer, AI Scientist, Research Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.