arXiv:2410.16143eess.IVcs.CV2024-10被引 17

提出可解释的混合模型,提升儿童肺炎胸片检测准确率

An Explainable Contrastive-based Dilated Convolutional Network with Transformer for Pediatric Pneumonia Detection

  • 结合空洞卷积与对比学习变压器,增强特征提取能力
  • 在四个公开数据集上表现优于现有方法,有效应对低剂量图像和数据不平衡
  • 通过注意力可视化实现诊断可解释性,适合医疗AI研发者参考

儿童肺炎仍是全球重大威胁,是五岁以下儿童死亡的主要原因之一,亟需快速诊断。目前以胸片为基础的早期诊断虽为常规手段,但面临图像辐射强度低及数据分布不均等问题。为此,本文提出一种新型可解释的对比学习空洞卷积网络与变压器融合模型(XCCNet),用于儿童肺炎检测。XCCNet利用空洞卷积捕捉空间特征,并通过基于对比学习的变压器获取全局信息,实现有效特征优化。一个强大的胸片预处理模块可缓解低强度图像问题,而对抗式数据增强则缓解数据集中的分布偏斜。此外,通过特征可视化实现可解释性,直接关联注意力区域与肺炎或正常状态的判别。XCCNet在四个公开数据集上进行了全面评估,实验结果表明其性能显著优于现有先进方法。

原文摘要 · Abstract (English)

Pediatric pneumonia remains a significant global threat, posing a larger mortality risk than any other communicable disease. According to UNICEF, it is a leading cause of mortality in children under five and requires prompt diagnosis. Early diagnosis using chest radiographs is the prevalent standard, but limitations include low radiation levels in unprocessed images and data imbalance issues. This necessitates the development of efficient, computer-aided diagnosis techniques. To this end, we propose a novel EXplainable Contrastive-based Dilated Convolutional Network with Transformer (XCCNet) for pediatric pneumonia detection. XCCNet harnesses the spatial power of dilated convolutions and the global insights from contrastive-based transformers for effective feature refinement. A robust chest X-ray processing module tackles low-intensity radiographs, while adversarial-based data augmentation mitigates the skewed distribution of chest X-rays in the dataset. Furthermore, we actively integrate an explainability approach through feature visualization, directly aligning it with the attention region that pinpoints the presence of pneumonia or normality in radiographs. The efficacy of XCCNet is comprehensively assessed on four publicly available datasets. Extensive performance evaluation demonstrates the superiority of XCCNet compared to state-of-the-art methods.

肺炎检测可解释AI医学影像深度学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。