提高多标签胸部X射线分类使用改进的排名损失
Muhammad Shehzad Hanif1, Muhammad Bilal1, Abdullah H Alsaggaf2
1Department of Electrical and Computer Engineering, King Abdulaziz University, Jeddah 21589, Saudi Arabia.
Bioengineering (Basel, Switzerland)
|June 26, 2025
概括
这项研究引入了一个新的焦点ZLPR损失函数,以改善胸部X射线图像中胸部疾病的多标签分类,有效处理类失衡并提高诊断准确性.
科学领域:
- 医疗成像医学成像
- 人工智能的人工智能
- 计算机视觉 计算机视觉
背景情况:
- 对胸部疾病的胸部X射线 (CXR) 分析,由于同时出现的病理,提出了多标签分类的挑战.
- 在CXR数据集中的类失衡,其中一些疾病是罕见的,使准确的模型训练复杂化.
- 从头开始训练深度学习模型需要大量的数据集,通常无法用于特定的医疗任务.
研究的目的:
- 开发一种有效的方法,在CXR图像中对胸部疾病进行多标签分类.
- 为应对CXR数据集中类不平衡的挑战.
- 提高深度学习模型在胸部疾病检测中的性能.
主要方法:
- 通过对NIH胸部X射线14数据集上预训练的DenseNet121模型进行微调来利用转移学习.
- 提出了一种新的基于等级的损失函数,焦点ZLPR (FZLPR),灵感来自焦点损失,以解决阶级不平衡.
- 在FZLPR中加入了一个温度参数,以强调难以分类的罕见疾病病例.
主要成果:
- 拟议的FZLPR损失函数超过了标准的损失函数,如二进制交叉 (BCE) 和焦点损失.
- 用FZLPR训练的模型在NIH胸部X射线14数据集上表现出卓越的表现.
- 通过使用FZLPR训练模型的测试时间增强,获得了80.96%的竞争平均AUC.
结论:
- 焦点ZLPR损失函数是改善CXR图像中胸部疾病多标签分类的有希望的方法.
- 该方法有效地处理了阶级不平衡,从而更准确地检测出常见和罕见疾病.
- 开发的模型表现出竞争力的性能,在临床环境中提供了增强诊断支持的潜力.
更多相关视频
02:09Multi-modal Pulmonary Imaging: Using Complementary Information from CT and Hyperpolarized 129Xe MRI to Evaluate Lung Structure-Function
Published on: April 12, 2024
683
13:44Detection of Architectural Distortion in Prior Mammograms via Analysis of Oriented Patterns
Published on: August 30, 2013
43.0K
