利用疾病分类体系提升胸部X光多病种识别准确率。
Taxlifier: Leveraging Disease Taxonomy for Enhanced Multi-Label Classification in Chest Radiography

- 引入疾病层级关系,通过损失函数和概率调整优化多标签分类。
- 在三大数据集上,准确率、AUC、F1值分别提升11%~24%。
- 结果更可解释,适合临床辅助诊断场景使用。
胸部X光(CXR)图像中胸腔疾病的精准高效分类对及时诊疗至关重要。然而,多种病灶视觉特征重叠,给自动化分类系统带来挑战。本文提出两种新型分层多标签分类方法:基于损失的和基于逻辑值的。前者将层级信息融入优化过程,后者根据疾病分类树中父类调整子类预测概率。我们在三个大规模CXRC数据集上评估:CheXpert(224,316张)、PADCHEST(160,000张)和NIH(112,120张)。实验表明,相比基线方法,两种方法在各类病灶上均显著提升准确率(分别+12%和+11%)、AUC(+13%和+10%)、F1分数(+24%和+12%)。我们还进行了全面统计分析,验证了方法的稳健性。融合领域特异性层级知识不仅提升分类性能,也提供更具临床解释性的输出。研究证明,分层多标签分类有望推动胸部X光辅助诊断系统发展。
原文摘要 · Abstract (English)
Accurate and efficient classification of thoracic diseases in chest X-ray (CXR) images is crucial for timely diagnosis and treatment. However, the presence of multiple pathologies with overlapping visual characteristics poses significant challenges for automated classification systems. In this study, we propose two novel hierarchical multi-label classification techniques, namely the loss-based and logit-based methods, to address these challenges by leveraging the hierarchical relationships among different thoracic pathologies. The loss-based technique integrates hierarchical information directly into the optimization process, while the logit-based method adjusts the predicted probabilities of each class based on its parent class in the disease taxonomy. We evaluate the performance of both techniques using three large-scale CXR datasets: CheXpert (224,316 CXRs), PADCHEST (160,000 CXRs), and NIH (112,120 CXRs). The experimental results demonstrate significant improvements in accuracy, AUC, and F1 scores compared to the baseline method across various pathologies. The logit-based and loss-based methods improve accuracy by 12\% and 11\%, AUC by 13\% and 10\%, and F1 scores by 24\% and 12\%, respectively compared to the baseline. These results represent a substantial improvement over the baseline method. Furthermore, we conduct a comprehensive statistical analysis to validate the robustness and reliability of the proposed techniques. The integration of domain-specific hierarchical knowledge not only enhances the classification performance but also provides a more interpretable output for clinical decision support. Our findings highlight the potential of hierarchical multi-label classification in advancing computer-aided diagnosis systems for chest radiography.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。