面向尿路结石CT图像的DAST-UNet双重注意力自动分割模型OA
Automatic segmentation of urinary stones in CT images using DAST-UNet with dual attention mechanism
目的 腹部CT图像中尿路结石存在形态多样、边界模糊及与周围组织对比度低等问题,给自动分割任务带来了显著挑战,尤其在小目标检测与复杂背景干扰下,传统方法往往难以兼顾精度与鲁棒性.为此,本文提出一种融合双重注意力机制的DAST-UNet模型,用于实现尿路结石的高精度自动分割.方法 所提模型以Swin-Unet为主干结构,设计了CASI(channel and spatial interactive)模块与TDA(token dependency attention)模块,分别从通道-空间维度和局部-全局语义关系两个层面增强模型特征表达能力.整体采用对称的编码器-解码器架构,通过多尺度特征融合与逐步上采样策略提升分割性能.其中,CASI模块增强模型对目标区域的关注并有效抑制背景噪声,TDA模块则通过加强Token之间的依赖关系,有效捕捉细粒度结构特征,优化上下文信息建模.结果 在构建的尿路结石CT图像数据集上,DAST-UNet在多个关键性能指标上均优于Res Unet、SegNet与Swin-Unet等经典模型,最终在测试集上获得Dice系数80.34%、灵敏性80.21%、准确率89.91%和IoU 67.14%.训练过程中模型损失函数快速收敛,验证集loss与训练集趋势一致,表明网络训练过程稳定,未出现明显过拟合.消融实验进一步验证了CASI和TDA模块对性能提升的具体贡献,Swin-Unet+CASI模型在Dice、灵敏性等指标上有明显改进,继续引入TDA后性能进一步提升,最终DAST-UNet在所有组合中表现最佳.结论 本文提出的DAST-UNet模型在结构上实现了对局部细节与全局上下文的协同建模,有效提升了CT图像中尿路结石的自动分割精度.实验结果充分验证了双重注意力机制对小目标识别与复杂背景处理的积极作用,为相关临床应用提供了一种可靠且具有推广价值的自动化解决方案.
Objective Urinary tract stones in abdominal CT images present significant challenges for automatic segmentation,particularly due to their diverse morphologies,blurred boundaries,and low contrast with surrounding tissues.Traditional methods often struggle to achieve a balance of accuracy and robustness,particularly when detecting small objects and encountering complex background interference.Therefore,a DAST-UNet model is proposed that incorporates a dual attention mechanism to achieve high-precision automatic segmentation of urinary tract stones.Methods The proposed model utilizes the Swin-Unet backbone architecture and incorporates a CASI(channel and spatial interactive)module and a TDA(Token dependency attention)module to enhance the model's feature representation capabilities from the channel-spatial dimension and the local-global semantic relationship level,respectively.A symmetric encoder-decoder architecture is employed to improve segmentation performance through multi-scale feature fusion and a progressive upsampling strategy.The CASI module enhances the model's focus on the target region and effectively suppresses background noise,while the TDA module effectively captures fine-grained structural features and optimizes contextual information modeling by strengthening dependencies between Tokens.Results On a constructed urinary tract calculi CT image dataset,DAST-UNet outperforms classic models such as Res Unet,SegNet,and Swin-Unet across multiple key performance metrics,ultimately achieving a Dice coefficient of 80.34%,sensitivity of 80.21%,accuracy of 89.91%,and intersection over union(IoU)of 67.14%on the test set.The model loss function converges rapidly during training,with the validation set loss trend consistent with the training set,indicating that the network training process is stable and free of significant overfitting.Ablation experiments further validate the specific contributions of the CASI and TDA modules to performance improvement.The Swin-Unet+CASI model shows significant improvements in metrics such as Dice and sensitivity.The introduction of TDA further improves performance,resulting in DAST-UNet performing the best among all combinations.Conclusions The proposed DAST-UNet model achieves a coordinated modeling of local details and global context,effectively improving the accuracy of automatic segmentation of urinary tract calculi in CT images.The experimental results fully verify the positive effect of the dual attention mechanism on small target recognition and complex background processing,and provide a reliable and scalable automated solution for related clinical applications.
孙海刚;张叶飞;刘彦斌
太原市中心医院(太原 030009)太原市中心医院(太原 030009)太原市中心医院(太原 030009)
医药卫生
结石分割CT图像DAST-UNetCASITDA
urinary stone segmentationCT imageDAST-UNetCASITDA
《北京生物医学工程》 2026 (2)
127-136,10
国家区域医疗中心科技创新计划项目(202212)资助
评论