arXiv:2506.15562eess.IVcs.CV2025-06被引 1

用混合模型提升医院本地MRI肿瘤分割精度,助力放疗精准规划。

Automated MRI Tumor Segmentation using hybrid U-Net with Transformer and Efficient Attention

  • 结合U-Net与Transformer的轻量级架构,引入多种注意力机制增强特征提取。
  • 在6080张本地MRI数据上实现Dice 0.764、IoU 0.736,性能媲美大样本模型。
  • 专为医院部署设计,适配隐私保护数据流程,适合临床AI系统落地。

癌症是具有局部侵袭和远端转移潜力的异常生长。准确自动分割肿瘤及周围正常组织,对优化放疗计划至关重要。当前基于AI的分割模型多在大型公开数据集上训练,缺乏本地患者群体的异质性。尽管这些研究推动了医学图像分割发展,但针对本地数据集的研究仍有必要,以将AI肿瘤分割模型直接集成至医院软件,实现高效精准的肿瘤治疗规划与执行。本研究在严格隐私保护下,利用本地医院采集的磁共振成像(MRI)数据,采用计算高效的混合U-Net-Transformer模型提升肿瘤分割性能。我们构建了稳健的数据处理流程,实现DICOM无缝提取与预处理,并通过大规模图像增强确保模型在多样化临床场景中的泛化能力,最终形成包含6080张图像的训练数据集。新型架构融合基于U-Net的卷积神经网络与Transformer瓶颈,辅以高效注意力、Squeeze-and-Excitation(SE)块、卷积块注意力模块(CBAM)及ResNeXt块。为加速收敛并降低计算开销,采用最大批量大小8,使用预训练ImageNet权重初始化编码器,并在双NVIDIA T4 GPU上通过检查点技术训练,突破Kaggle运行时限制。定量评估显示,该模型在本地MRI数据集上达到Dice相似系数0.764与交并比(IoU)0.736,表现优异,凸显有限数据条件下开展院内专属模型开发的重要性。

原文摘要 · Abstract (English)

Cancer is an abnormal growth with potential to invade locally and metastasize to distant organs. Accurate auto-segmentation of the tumor and surrounding normal tissues is required for radiotherapy treatment plan optimization. Recent AI-based segmentation models are generally trained on large public datasets, which lack the heterogeneity of local patient populations. While these studies advance AI-based medical image segmentation, research on local datasets is necessary to develop and integrate AI tumor segmentation models directly into hospital software for efficient and accurate oncology treatment planning and execution. This study enhances tumor segmentation using computationally efficient hybrid UNet-Transformer models on magnetic resonance imaging (MRI) datasets acquired from a local hospital under strict privacy protection. We developed a robust data pipeline for seamless DICOM extraction and preprocessing, followed by extensive image augmentation to ensure model generalization across diverse clinical settings, resulting in a total dataset of 6080 images for training. Our novel architecture integrates UNet-based convolutional neural networks with a transformer bottleneck and complementary attention modules, including efficient attention, Squeeze-and-Excitation (SE) blocks, Convolutional Block Attention Module (CBAM), and ResNeXt blocks. To accelerate convergence and reduce computational demands, we used a maximum batch size of 8 and initialized the encoder with pretrained ImageNet weights, training the model on dual NVIDIA T4 GPUs via checkpointing to overcome Kaggle's runtime limits. Quantitative evaluation on the local MRI dataset yielded a Dice similarity coefficient of 0.764 and an Intersection over Union (IoU) of 0.736, demonstrating competitive performance despite limited data and underscoring the importance of site-specific model development for clinical deployment.

肿瘤分割MRI轻量化模型临床部署

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