arXiv:2511.16162cs.CVcs.GR2025-11

通过分层噪声引导小波重建,提升医学图像分割在扰动下的稳定性。

Layer-wise Noise Guided Selective Wavelet Reconstruction for Robust Medical Image Segmentation

  • 分层注入噪声学习频域先验,指导特征避开敏感方向。
  • 在强攻击下性能下降减少,清洁与鲁棒性指标同步提升。
  • 可无缝集成现有训练流程,适合临床部署应用。

临床部署要求分割模型在分布偏移和扰动下保持稳定。主流方法是对抗训练(AT),但常导致干净精度-鲁棒性权衡及高训练成本,限制了在医学影像中的可扩展性。本文提出分层噪声引导的选通小波重建(LNG-SWR)。训练时,在多层注入零均值小噪声,学习频率偏置先验,使表示远离噪声敏感方向。随后对输入/特征分支进行先验引导的选通小波重建,实现频域自适应:抑制噪声敏感频带,增强方向结构与形状线索,稳定边界响应,同时保持频谱一致性。该框架与主干网络无关,推理开销低,可作为AT的即插即用增强模块,也可独立提升鲁棒性。在CT与超声数据集上,统一采用PGD-$L_{\

原文摘要 · Abstract (English)

Clinical deployment requires segmentation models to stay stable under distribution shifts and perturbations. The mainstream solution is adversarial training (AT) to improve robustness; however, AT often brings a clean--robustness trade-off and high training/tuning cost, which limits scalability and maintainability in medical imaging. We propose \emph{Layer-wise Noise-Guided Selective Wavelet Reconstruction (LNG-SWR)}. During training, we inject small, zero-mean noise at multiple layers to learn a frequency-bias prior that steers representations away from noise-sensitive directions. We then apply prior-guided selective wavelet reconstruction on the input/feature branch to achieve frequency adaptation: suppress noise-sensitive bands, enhance directional structures and shape cues, and stabilize boundary responses while maintaining spectral consistency. The framework is backbone-agnostic and adds low additional inference overhead. It can serve as a plug-in enhancement to AT and also improves robustness without AT. On CT and ultrasound datasets, under a unified protocol with PGD-$L_{\infty}/L_{2}$ and SSAH, LNG-SWR delivers consistent gains on clean Dice/IoU and significantly reduces the performance drop under strong attacks; combining LNG-SWR with AT yields additive gains. When combined with adversarial training, robustness improves further without sacrificing clean accuracy, indicating an engineering-friendly and scalable path to robust segmentation. These results indicate that LNG-SWR provides a simple, effective, and engineering-friendly path to robust medical image segmentation in both adversarial and standard training regimes.

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