提出动态编码网络,精准分割大小形状各异的器官。
MD-RWKV-UNet: Scale-Aware Anatomical Encoding with Cross-Stage Fusion for Multi-Organ Segmentation
- 用可变感受野模块实现多尺度结构自适应建模
- 在Synapse和ACDC数据集上边界精度领先,小器官分割更准
- 适合需要高精度解剖细节的医学图像分析任务
医学影像中多器官分割因解剖差异大、器官间依赖复杂及器官尺度与形状多样而极具挑战。传统编码器-解码器结构难以同时捕捉精细局部细节与长程上下文信息,尤其影响小器官或可变形器官的准确分割。为此,本文提出MD-RWKV-UNet,一种具备尺度感知表征与空间自适应上下文建模能力的动态编码网络。核心为MD-RWKV模块,通过可变形空间偏移与接收门加权键值机制的双路径设计,使感受野能根据局部结构线索动态调整。进一步引入选择性核注意力,实现对不同感受野卷积核的自适应选择,增强多尺度交互并提升对器官大小与形状变化的鲁棒性。同时,跨阶段双注意力融合策略聚合编码器各层级特征,在保留低层结构的同时强化语义一致性。相比堆叠静态卷积或依赖全局注意力的方法,本方案轻量且表达能力强。在Synapse与ACDC数据集上的实验表明,该方法达到当前最优性能,尤其在边界精度与小器官分割方面表现突出。
原文摘要 · Abstract (English)
Multi-organ segmentation in medical imaging remains challenging due to large anatomical variability, complex inter-organ dependencies, and diverse organ scales and shapes. Conventional encoder-decoder architectures often struggle to capture both fine-grained local details and long-range context, which are crucial for accurate delineation - especially for small or deformable organs. To address these limitations, we propose MD-RWKV-UNet, a dynamic encoder network that enables scale-aware representation and spatially adaptive context modeling. At its core is the MD-RWKV block, a dual-path module that integrates deformable spatial shifts with the Receptance Weighted Key Value mechanism, allowing the receptive field to adapt dynamically to local structural cues. We further incorporate Selective Kernel Attention to enable adaptive selection of convolutional kernels with varying receptive fields, enhancing multi-scale interaction and improving robustness to organ size and shape variation. In parallel, a cross-stage dual-attention fusion strategy aggregates multi-level features across the encoder, preserving low-level structure while enhancing semantic consistency. Unlike methods that stack static convolutions or rely heavily on global attention, our approach provides a lightweight yet expressive solution for dynamic organ modeling. Experiments on Synapse and ACDC demonstrate state-of-the-art performance, particularly in boundary precision and small-organ segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。