提出多尺度注意力密集网络,提升肺癌分类准确率
MSAD-Net: Multiscale and Spatial Attention-based Dense Network for Lung Cancer Classification
- 设计新型密集模块与跨层连接,降低模型复杂度
- 在CT图像上达到98.7%准确率,优于ConvNext-Tiny等模型
- 适合医学影像分析、肺癌早期筛查研究者使用
肺癌是全球癌症死亡的主要原因之一,早期检测至关重要。传统放射科医生通过胸片或CT扫描人工诊断存在挑战,亟需自动化系统。尽管深度学习尤其是卷积神经网络(CNN)已显著提升自动检测能力,但医学图像中的类别不平衡问题仍限制性能。本文提出一种新架构——多尺度密集网络(MSAD-Net),在第4和第5个块中引入新型密集模块,结合3个深度可分离卷积层与1个1×1卷积层,有效降低计算复杂度;同时加入从第3块到第5块的跳跃连接,以及从第4块到全局平均池化(GAP)层的并行分支,后者采用膨胀率为2的空洞卷积以提取多尺度特征。大量实验表明,该模型在肺癌分类任务上显著优于ConvNeXt-Tiny、Vision Transformer(ViT)、Pooling-based ViT(PiT)等主流模型,准确率达到98.7%。
原文摘要 · Abstract (English)
Lung cancer, a severe form of malignant tumor that originates in the tissues of the lungs, can be fatal if not detected in its early stages. It ranks among the top causes of cancer-related mortality worldwide. Detecting lung cancer manually using chest X-Ray image or Computational Tomography (CT) scans image poses significant challenges for radiologists. Hence, there is a need for automatic diagnosis system of lung cancers from radiology images. With the recent emergence of deep learning, particularly through Convolutional Neural Networks (CNNs), the automated detection of lung cancer has become a much simpler task. Nevertheless, numerous researchers have addressed that the performance of conventional CNNs may be hindered due to class imbalance issue, which is prevalent in medical images. In this research work, we have proposed a novel CNN architecture ``Multi-Scale Dense Network (MSD-Net)'' (trained-from-scratch). The novelties we bring in the proposed model are (I) We introduce novel dense modules in the 4th block and 5th block of the CNN model. We have leveraged 3 depthwise separable convolutional (DWSC) layers, and one 1x1 convolutional layer in each dense module, in order to reduce complexity of the model considerably. (II) Additionally, we have incorporated one skip connection from 3rd block to 5th block and one parallel branch connection from 4th block to Global Average Pooling (GAP) layer. We have utilized dilated convolutional layer (with dilation rate=2) in the last parallel branch in order to extract multi-scale features. Extensive experiments reveal that our proposed model has outperformed latest CNN model ConvNext-Tiny, recent trend Vision Transformer (ViT), Pooling-based ViT (PiT), and other existing models by significant margins.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。