用多注意力集成模型提升肺结节良恶性判别准确率
Multi-Attention Stacked Ensemble for Lung Cancer Detection in CT Scans
- 融合三类预训练模型,通过双阶段注意力加权决策
- 在LIDC-IDRI数据集上达98.09%准确率,错误率降35%
- 特别擅长处理放射科医生意见不一致的疑难病例
本文针对CT影像中肺结节良恶性二分类难题,提出一种多层级注意力堆叠集成深度神经网络。采用EfficientNet V2 S、MobileViT XXS和DenseNet201三类预训练主干网络,分别适配96×96像素输入并配置定制分类头。设计两阶段注意力机制,从拼接的逻辑输出中学习模型级与类别级重要性得分,并由轻量级元学习器优化最终预测。为缓解类别不平衡并提升泛化能力,引入动态焦点损失(带经验计算的类别权重)、训练时的MixUp增强及推理时的测试增强。在LIDC-IDRI数据集上的实验表明,模型达到98.09%准确率和0.9961 AUC,相比现有最优方法错误率降低35%。敏感度达98.73,特异度为98.96,在放射科医生意见分歧高的难例上表现尤为突出。统计显著性检验验证了结果的稳定性。该方法可作为放射科医生肺癌筛查的可靠自动化辅助工具。
原文摘要 · Abstract (English)
In this work, we address the challenge of binary lung nodule classification (benign vs malignant) using CT images by proposing a multi-level attention stacked ensemble of deep neural networks. Three pretrained backbones -- EfficientNet V2 S, MobileViT XXS, and DenseNet201 -- are each adapted with a custom classification head tailored to 96 x 96 pixel inputs. A two-stage attention mechanism learns both model-wise and class-wise importance scores from concatenated logits, and a lightweight meta-learner refines the final prediction. To mitigate class imbalance and improve generalization, we employ dynamic focal loss with empirically calculated class weights, MixUp augmentation during training, and test-time augmentation at inference. Experiments on the LIDC-IDRI dataset demonstrate exceptional performance, achieving 98.09 accuracy and 0.9961 AUC, representing a 35 percent reduction in error rate compared to state-of-the-art methods. The model exhibits balanced performance across sensitivity (98.73) and specificity (98.96), with particularly strong results on challenging cases where radiologist disagreement was high. Statistical significance testing confirms the robustness of these improvements across multiple experimental runs. Our approach can serve as a robust, automated aid for radiologists in lung cancer screening.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。