arXiv:2505.17602eess.IV2025-05被引 3

提出新型3D注意力残差UNet,提升肺部与结节分割精度。

Attention-ResUNet and EfficientSASM-UNet: UNet based frameworks for Lung and Nodule segmentation

  • 基于注意力机制和残差块,改进编码器-解码器结构
  • 在LUNA16数据集上Dice得分优于现有方法
  • 适合临床实时肺结节分割,兼顾精度与效率

肺癌是全球主要健康威胁之一,早期检测可显著提高生存率。肺实质(含胸膜旁结节)和肺结节的精确分割是肺部计算机辅助检测(CAD)系统的关键。由于结节类型多样及肺内干扰结构复杂,该任务极具挑战性。传统机器学习/深度学习方法泛化能力不足,近期视觉语言模型虽在解剖层面表现良好,但在细粒度分割任务中性能受限,且半自动特性不适用于实时临床场景。本文提出一种新型3D肺实质与结节分割方法,采用基于注意力的网络架构,每层编码器-解码器均引入残差块;编码器用步进卷积替代最大池化,解码器用转置卷积替代三线性插值,以增加可学习参数;各阶段使用空洞卷积,在不增加计算成本的前提下扩大上下文感受野。该方法在公开最大的数据集LUNA16上进行了全面评估,并与领域内最新方法进行对比,采用Dice分数、交并比(IoU)等标准指标。结果表明,所提方法在多项指标上优于当前最优水平。源代码、数据集及预处理数据可通过链接获取:https://github.com/EMeRALDsNRPU/Attention-Based-3D-ResUNet。

原文摘要 · Abstract (English)

Lung cancer has been one of the major threats across the world with the highest mortalities. Computer-aided detection (CAD) can help in early detection and thus can help increase the survival rate. Accurate lung parenchyma segmentation (to include the juxta-pleural nodules) and lung nodule segmentation, the primary symptom of lung cancer, play a crucial role in the overall accuracy of the Lung CAD pipeline. Lung nodule segmentation is quite challenging because of the diverse nodule types and other inhibit structures present within the lung lobes. Traditional machine/deep learning methods suffer from generalization and robustness. Recent Vision Language Models/Foundation Models perform well on the anatomical level, but they suffer on fine-grained segmentation tasks, and their semi-automatic nature limits their effectiveness in real-time clinical scenarios. In this paper, we propose a novel method for accurate 3D segmentation of lung parenchyma and lung nodules. The proposed architecture is an attention-based network with residual blocks at each encoder-decoder state. Max pooling is replaced by strided convolutions at the encoder, and trilinear interpolation is replaced by transposed convolutions at the decoder to maximize the number of learnable parameters. Dilated convolutions at each encoder-decoder stage allow the model to capture the larger context without increasing computational costs. The proposed method has been evaluated extensively on one of the largest publicly available datasets, namely LUNA16, and is compared with recent notable work in the domain using standard performance metrics like Dice score, IOU, etc. It can be seen from the results that the proposed method achieves better performance than state-of-the-art methods. The source code, datasets, and pre-processed data can be accessed using the link: https://github.com/EMeRALDsNRPU/Attention-Based-3D-ResUNet.

肺结节分割3D分割UNet医学图像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。