不靠分割和过度增广,也能准确识别新冠肺部X光片。
Are Data Augmentation and Segmentation Always Necessary? Insights from COVID-19 X-Rays and a Methodology Thereof
- 用热力图分析模型决策,验证肺部分割对诊断必要性
- 增广数据超阈值后准确率下降,说明会引发过拟合
- 新方法SDL-COVID精度达95.21%,适合临床辅助诊断
快速可靠的诊断工具对管理呼吸系统疾病(如新冠肺炎)至关重要。胸部X光结合人工智能技术已被证明极具价值。然而,现有大部分研究未考虑肺部分割,对其可靠性提出质疑;同时,部分研究采用不均衡且不切实际的数据增强方式,导致模型泛化能力差、易过拟合。本研究对上述问题进行批判性分析,并提出一种新方法SDL-COVID,以提升胸部X光图像中新冠肺炎分类的可靠性。通过类激活映射(CAM)可视化卷积神经网络(CNN)的预测过程,验证肺部分割在精准诊断中的必要性。在两个层面评估数据增强的影响:一个使用增强数据集,另一个使用非增强数据集。专家医学监督下的图像与热力图分析表明,肺部分割对于准确预测新冠肺炎至关重要。此外,测试准确率在超过某一阈值后随增强图像增加而显著下降,表明模型出现过拟合。所提方法SDL-COVID实现95.21%的精确率和更低的假阴性率,确保其在新冠肺炎胸部X光检测中的可靠性。
原文摘要 · Abstract (English)
Purpose: Rapid and reliable diagnostic tools are crucial for managing respiratory diseases like COVID-19, where chest X-ray analysis coupled with artificial intelligence techniques has proven invaluable. However, most existing works on X-ray images have not considered lung segmentation, raising concerns about their reliability. Additionally, some have employed disproportionate and impractical augmentation techniques, making models less generalized and prone to overfitting. This study presents a critical analysis of both issues and proposes a methodology (SDL-COVID) for more reliable classification of chest X-rays for COVID-19 detection. Methods: We use class activation mapping to obtain a visual understanding of the predictions made by Convolutional Neural Networks (CNNs), validating the necessity of lung segmentation. To analyze the effect of data augmentation, deep learning models are implemented on two levels: one for an augmented dataset and another for a non-augmented dataset. Results: Careful analysis of X-ray images and their corresponding heat maps under expert medical supervision reveals that lung segmentation is necessary for accurate COVID-19 prediction. Regarding data augmentation, test accuracy significantly drops beyond a certain threshold with additional augmented images, indicating model overfitting. Conclusion: Our proposed methodology, SDL-COVID, achieves a precision of 95.21% and a lower false negative rate, ensuring its reliability for COVID-19 detection using chest X-rays.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。