arXiv:2603.03187cs.CV2026-03被引 1

提出新型跳接门控机制,精准过滤医学图像中的无关噪声。

ProSMA-UNet: Decoder Conditioning for Proximal-Sparse Skip Feature Selection

  • 用解码器条件控制的稀疏特征选择替代传统注意力门
  • 3D分割任务性能提升约20%,显著优于现有方法
  • 适合低对比度临床影像分割,尤其擅长抑制噪声干扰

医学图像分割常用U形编码器-解码器结构(如U-Net),通过跳跃连接将高分辨率编码器特征注入解码器以保留细节。然而,这些跳跃路径也会传递低级纹理、背景杂波和采集噪声,导致无关信息绕过深层语义过滤,尤其在低对比度临床成像中危害显著。尽管引入注意力门可缓解此问题,但通常生成密集的sigmoid掩码,仅软性重加权特征而非显式移除无关激活。本文提出ProSMA-UNet(近端稀疏多尺度注意力U-Net),将跳跃门控重构为解码器条件的稀疏特征选择问题。ProSMA利用轻量级深度可分离空洞卷积构建多尺度兼容场,捕捉局部与上下文尺度的相关性,再通过$\ ext{l}_1$近端算子施加显式稀疏性,结合可学习通道阈值,实现闭式软阈值门控以清除噪声响应。为进一步抑制语义无关通道,引入由全局解码器上下文驱动的通道门控。在具有挑战性的2D与3D基准测试中,实验表明其达到顶尖性能,尤其在复杂3D分割任务上取得约20%的显著提升。

原文摘要 · Abstract (English)

Medical image segmentation commonly relies on U-shaped encoder-decoder architectures such as U-Net, where skip connections preserve fine spatial detail by injecting high-resolution encoder features into the decoder. However, these skip pathways also propagate low-level textures, background clutter, and acquisition noise, allowing irrelevant information to bypass deeper semantic filtering -- an issue that is particularly detrimental in low-contrast clinical imaging. Although attention gates have been introduced to address this limitation, they typically produce dense sigmoid masks that softly reweight features rather than explicitly removing irrelevant activations. We propose ProSMA-UNet (Proximal-Sparse Multi-Scale Attention U-Net), which reformulates skip gating as a decoder-conditioned sparse feature selection problem. ProSMA constructs a multi-scale compatibility field using lightweight depthwise dilated convolutions to capture relevance across local and contextual scales, then enforces explicit sparsity via an $\ell_1$ proximal operator with learnable per-channel thresholds, yielding a closed-form soft-thresholding gate that can remove noisy responses. To further suppress semantically irrelevant channels, ProSMA incorporates decoder-conditioned channel gating driven by global decoder context. Extensive experiments on challenging 2D and 3D benchmarks demonstrate state-of-the-art performance, with particularly large gains ($\approx20$\%) on difficult 3D segmentation tasks. Project page: https://math-ml-x.github.io/ProSMA-UNet/

医学图像分割注意力机制U-Net

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