多分支卷积神经网络提升CT影像细微病灶识别准确率
A Novel Multi-branch ConvNeXt Architecture for Identifying Subtle Pathological Features in CT Scans
- 三路并行提取全局平均、最大池化与注意力加权特征
- 在2609张肺部CT上达到AUC 0.9937,准确率97.57%
- 适合医学影像分析、传染病智能诊断等场景
智能医学影像分析对辅助临床诊断至关重要,尤其在识别细微病理特征方面。本文提出一种新型多分支ConvNeXt架构,专为应对医学图像分析中的细微挑战而设计。虽聚焦于新冠肺炎诊断,但该方法可推广至多种病理类型分类。模型采用端到端流程,涵盖精细的数据预处理与增强,以及基于迁移学习的两阶段训练策略。架构融合三条并行分支提取的特征:全局平均池化、全局最大池化,以及一种新型注意力加权池化机制。模型在两个独立数据集合并的2,609张CT切片上训练与验证。实验结果表明,在验证集上取得0.9937的ROC-AUC、0.9757的准确率和0.9825的F1分数,优于此前所有报告模型。结果证明,结合精心数据处理的现代多分支架构,可实现媲美或超越当前先进水平的性能,验证了深度学习技术在稳健医疗诊断中的有效性。
原文摘要 · Abstract (English)
Intelligent analysis of medical imaging plays a crucial role in assisting clinical diagnosis, especially for identifying subtle pathological features. This paper introduces a novel multi-branch ConvNeXt architecture designed specifically for the nuanced challenges of medical image analysis. While applied here to the specific problem of COVID-19 diagnosis, the methodology offers a generalizable framework for classifying a wide range of pathologies from CT scans. The proposed model incorporates a rigorous end-to-end pipeline, from meticulous data preprocessing and augmentation to a disciplined two-phase training strategy that leverages transfer learning effectively. The architecture uniquely integrates features extracted from three parallel branches: Global Average Pooling, Global Max Pooling, and a new Attention-weighted Pooling mechanism. The model was trained and validated on a combined dataset of 2,609 CT slices derived from two distinct datasets. Experimental results demonstrate a superior performance on the validation set, achieving a final ROC-AUC of 0.9937, a validation accuracy of 0.9757, and an F1-score of 0.9825 for COVID-19 cases, outperforming all previously reported models on this dataset. These findings indicate that a modern, multi-branch architecture, coupled with careful data handling, can achieve performance comparable to or exceeding contemporary state-of-the-art models, thereby proving the efficacy of advanced deep learning techniques for robust medical diagnostics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。