Large-scale long-tailed disease diagnosis on radiology images
Developing a generalist radiology diagnosis system can greatly enhance clinical diagnostics. In this paper, we introduce RadDiag, a foundational model supporting 2D and 3D inputs across various modalities and anatomies, using a transformer-based fusion module for comprehensive disease diagnosis. Due to patient privacy concerns and the lack of large-scale radiology diagnosis datasets, we utilize high-quality, clinician-reviewed radiological images available online with diagnosis labels. Our dataset, RP3D-DiagDS, contains 40,936 cases with 195,010 scans covering 5568 disorders (930 unique ICD-10-CM codes). Experimentally, our RadDiag achieves 95.14% AUC on internal evaluation with the knowledge-enhancement strategy. Additionally, RadDiag can be zero-shot applied or fine-tuned to external diagnosis datasets sourced from various medical centers, demonstrating state-of-the-art results. In conclusion, we show that publicly shared medical data on the Internet is a tremendous and valuable resource that can potentially support building strong models for image understanding in healthcare.
Authors
- Xiaoman Zhang (ORCID: https://orcid.org/0000-0002-7696-9366)
- Weidi Xie (ORCID: https://orcid.org/0009-0002-8609-6826)
- Lisong Dai (ORCID: https://orcid.org/0000-0003-3608-9732)
- Chaoyi Wu (ORCID: https://orcid.org/0009-0004-1374-011X)
- Hengyu Guan (ORCID: https://orcid.org/0009-0007-6168-285X)
- Ya Zhang (ORCID: https://orcid.org/0000-0002-5390-9053)
- Yanfeng Wang (ORCID: https://orcid.org/0000-0002-3196-2347)
- Yuehua Li (ORCID: https://orcid.org/0000-0001-8028-248X)
- Weike Zhao
- Qiaoyu Zheng (ORCID: https://orcid.org/0009-0004-7494-7851)
Institutions
- Shanghai Jiao Tong University (CN)
- Renji Hospital (CN)
- Beijing Academy of Artificial Intelligence (CN)
- Shanghai Sixth People's Hospital (CN)
- Shanghai Artificial Intelligence Laboratory (CN)
Publication Details
- Journal
- Nature Communications
- Published
- 2024-11-22
- DOI
- https://doi.org/10.1038/s41467-024-54424-6
- Citations
- 15
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 6.06
Funders
- Science and Technology Commission of Shanghai Municipality