基于双重注意力机制的间质性肺病高分辨率CT图像分类方法
DOI:
作者:
作者单位:

1.云南大学;2.昆明医科大学第一附属医院

作者简介:

通讯作者:

中图分类号:

A

基金项目:

国家自然科学基金资助项目(62063034,61841112)


High Resolution CT Image Classification Method For Interstitial Lung Disease Based On Dual Attention Mechanism
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为了更精确地分类间质性疾病,本文提出了一种基于深度学习的分类网络,首先将多头自注意力机制模块与 DenseNet-121结合,使得模型能够同时关注多个重点区域。然后采用卷积注意力模块实现更高效的特征提取,提升网络的空间感知能力,从而增强分类性能。最后,添加改进的空间金字塔池化层将不同尺度的特征图拼接起来以捕获更丰富的空间信息。此外针对高分辨率CT 图像数据集类别不均衡问题,引入Focal Loss损失函数,使得模型在训练时更专注于难分类的样本,从而进一步增强模型的分类能力。所提方法在本文未经训练的数据集上进行测试,达到了88.28%的准确率。相较于原始 DenseNet-121 在准确率、召回率、精确率、F1 分数和 Kappa 系数提高了4.65%、5.08%、5.82%、5.45%和6.38% 实验结果表明该方法具有特征提取能力强和分类准确率高的特点。

    Abstract:

    Abstract In order to classify interstitial diseases more accurately, this paper proposes a classification network based on deep learning Xi, which first combines the multi-head self-attention mechanism module with DenseNet-121, so that the model can focus on multiple key regions at the same time. Then, the convolutional attention module is used to achieve more efficient feature extraction and spatial perception capabilities, so as to improve the classification ability of the network. Finally, an improved spatial pyramid pooling layer is added to stitch together the feature maps of different scales to capture richer spatial information. In addition, aiming at the problem of category imbalance of high-resolution CT image datasets, the Focal Loss function is introduced, which makes the model focus more on the difficult samples during training, so as to further improve the classification ability of the model. The proposed method is tested on the untrained dataset in this paper, and the accuracy rate reaches 88.28%. Compared with the original DenseNet-121, the accuracy, recall, precision, F1 score and Kappa coefficient are increased by 4.65%, 5.08%, 5.82%, 5.45% and 6.38% Experimental results show that the proposed method has the characteristics of strong feature extraction ability and high classification accuracy.

    参考文献
    相似文献
    引证文献
引用本文
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-01-08
  • 最后修改日期:2024-03-20
  • 录用日期:2024-04-11
  • 在线发布日期:
  • 出版日期: