用前后扫描图提升前列腺分段,少标注也能准。
MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal Prostate MRI Segmentation
- 双输入设计:结合当前与前次扫描,利用历史信息优化分割。
- 在有限标注下达到最佳性能,平均Dice达0.913,优于U-Net和Transformer模型。
- 适合临床长期随访场景,尤其标注稀缺的前列腺MRI分析。
主动监测(AS)是管理低至中度风险前列腺癌(PCa)的重要策略,通过定期MRI和临床随访监控病情进展,避免过度治疗。准确分割前列腺是实现自动化检测与诊断的关键第一步。然而,现有深度学习分割模型多基于单时间点、专家标注数据集训练,难以适用于纵向AS分析——因存在多个时间点且专家标注稀缺,导致模型难以有效微调。为此,我们提出MambaX-Net,一种新型半监督、双扫描3D分割架构,通过融合前一时相的MRI与对应分割掩码来生成当前时间点的分割结果。创新引入两项组件:(i) Mamba增强交叉注意力模块,将Mamba块嵌入交叉注意力机制,高效捕捉时间演变与长程空间依赖;(ii) 形状提取模块,将前一时刻分割掩码编码为潜在解剖表征,以精化各区域边界。此外,采用半监督自训练策略,利用预训练nnU-Net生成伪标签,在无需专家标注条件下实现有效学习。MambaX-Net在纵向主动监测数据集上评估,显著优于现有SOTA的U-Net与Transformer模型,在仅使用有限且带噪声数据的情况下仍实现优越的前列腺分区分割性能。
原文摘要 · Abstract (English)
Active Surveillance (AS) is a treatment option for managing low and intermediate-risk prostate cancer (PCa), aiming to avoid overtreatment while monitoring disease progression through serial MRI and clinical follow-up. Accurate prostate segmentation is an important preliminary step for automating this process, enabling automated detection and diagnosis of PCa. However, existing deep-learning segmentation models are often trained on single-time-point, expertly annotated datasets, making them unsuitable for longitudinal AS analysis, where multiple time points and a scarcity of expert labels hinder effective fine-tuning. To address these challenges, we propose MambaX-Net, a novel semi-supervised, dual-scan 3D segmentation architecture that computes the segmentation for time point t by leveraging the MRI and the corresponding segmentation mask from the previous time point. We introduce two new components: (i) a Mamba-enhanced Cross-Attention Module, which integrates the Mamba block into cross-attention to efficiently capture temporal evolution and long-range spatial dependencies, and (ii) a Shape Extractor Module that encodes the previous segmentation mask into a latent anatomical representation for refined zone delineation. Moreover, we use a semi-supervised self-training strategy that leverages pseudo-labels generated from a pre-trained nnU-Net, enabling effective learning without expert annotations. MambaX-Net was evaluated on a longitudinal AS dataset, and results showed that it significantly outperforms state-of-the-art U-Net and Transformer-based models, achieving superior prostate zone segmentation even when trained on limited and noisy data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。