提出稳定乘法交互模块,提升医学图像分割精度。
PU-UNet: Stable Multiplicative Interactions for Medical Image Segmentation

- 用稳定映射与对数域截断实现乘法特征建模
- 在多个数据集上Dice分数达0.942~0.959,误报率降为0
- 适合追求高精度医学图像分割的研究者
许多密集预测网络依赖加性特征变换,隐式建模高阶特征交互。乘积单元提供显式乘法建模机制,但其对数-指数形式易引发数值不稳定,限制了在深层密集预测网络中的应用。本文提出产品单元U-Net(PU-UNet),一种将稳定乘积单元残差块融入丰富低分辨率阶段的残差U-Net结构。所提方法结合平滑正性映射与对数域截断,实现稳定乘法特征学习,计算开销可忽略。在ISIC 2018、Kvasir-SEG和BUSI数据集上,PU-UNet分别取得0.942、0.959和最高0.925的Dice分数。相比匹配的残差U-Net基线,其持续提升Dice与IoU,参数量、浮点运算量(FLOPs)和推理延迟几乎不变,并将正常BUSI病例的图像级假阳性率从0.077降至零。消融实验表明,性能提升源于乘积单元交互,低分辨率处效果最显著,且受益于所提稳定设计。结果表明,稳定的乘积单元残差学习是增强U-Net类分割网络显式乘法交互的有效方式。
原文摘要 · Abstract (English)
Many dense prediction networks rely on additive feature transformations and model higher-order feature interactions only implicitly. Product units provide an explicit mechanism for multiplicative feature modeling, but their logarithmic--exponential formulation can cause numerical instability, which has limited their use in deep dense prediction networks. In this work, we propose Product-Unit U-Net (PU-UNet), a residual U-Net that integrates stable product-unit residual blocks into rich low-resolution stages for medical image segmentation. The proposed formulation combines smooth positivity mapping with log-domain clipping, enabling stable multiplicative feature learning with negligible computational overhead. On ISIC 2018, Kvasir-SEG, and BUSI, PU-UNet achieves Dice scores of 0.942, 0.959, and up to 0.925, respectively. Compared with a matched Residual U-Net baseline, PU-UNet consistently improves Dice and IoU while keeping parameters, FLOPs, and inference latency nearly unchanged, and reduces the image-level false-positive rate on normal BUSI cases from 0.077 to zero. Ablation studies suggest that the gains are associated with product-unit interactions, are strongest under low-resolution placement, and benefit from the proposed stabilization design. These results suggest that stable product-unit residual learning can be an effective way to enhance U-Net-style segmentation networks with explicit multiplicative interactions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。