通过形状归一化提升肺部影像识别准确率
Accurate Recognition of Pneumonia and COVID-19 by Geometric Shape Normalization of Lung Region using Automatic Landmark Detection and Piecewise Affine Warping

- 用15个关键点检测+分段仿射变换对肺部形状标准化
- 在新冠数据库上达98.60%准确率,优于未对齐图像
- 适合医学影像分析、临床辅助诊断场景
本文提出一种自动系统,通过肺部区域的几何归一化识别胸部X光中的肺部疾病。方法包含三个模块:(1) 基于ResNet-18与坐标注意力的地标检测器,预测15个肺轮廓关键点,通过四个模型集成与测试时增强,平均定位误差为3.61像素;(2) 基于广义普鲁克斯特分析、德劳内三角剖分与分段仿射变换的几何归一化器,将每个肺区映射到标准形状;(3) 基于ResNet-18的分类器,结合迁移学习与SAHS对比度增强,将图像分类为新冠肺炎、病毒性肺炎或正常。在COVID-19 Radiography Database上,归一化图像分类器五折交叉验证下准确率达98.60±0.26%,F1-Macro为98.00%。尽管原始图像原始准确率略高,但梯度加权类激活图(Grad-CAM)与裁剪实验表明该优势部分受成像伪影影响。相比之下,几何归一化图像在相同数据集上(98.60% vs. 96.24%)及包含儿科病例的混合数据集(94.67% vs. 94.17%)中均表现更优,表明解剖对齐可提供更稳定、抗伪影的表征。
原文摘要 · Abstract (English)
This paper presents an automatic system for recognizing pulmonary diseases in chest X-rays using geometric normalization of the lung region. The method combines three modules: (1) a ResNet-18 landmark detector with coordinate attention that predicts 15 lung-contour landmarks, achieving a mean localization error of 3.61 pixels through an ensemble of four models with test-time augmentation; (2) a geometric normalizer based on Generalized Procrustes Analysis, Delaunay triangulation, and piecewise affine warping to map each lung region to a standardized shape; and (3) a ResNet-18 classifier with transfer learning and SAHS contrast enhancement to classify images as COVID-19, Viral Pneumonia, or Normal. On the COVID-19 Radiography Database, the normalized-image classifier achieved 98.60+/-0.26% accuracy and 98.00% F1-Macro using five-fold cross-validation. Although original images produced slightly higher raw accuracy, Grad-CAM and cropping experiments suggest that this advantage is partly influenced by acquisition artifacts. In contrast, geometrically normalized images outperformed artifact-masked/cropped unaligned images on both the COVID-19 Radiography Database (98.60% vs. 96.24%) and a balanced adult-pediatric mixed dataset including pediatric cases from the Kermany dataset (94.67% vs. 94.17%). These results suggest that anatomical alignment can provide a more controlled and artifact-resistant representation for pulmonary disease recognition.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。