通过小波分解分离视网膜图像的解剖结构与设备差异,提升模型跨域泛化能力。
Decoupling Wavelet Sub-bands for Single Source Domain Generalization in Fundus Image Segmentation
- 利用小波子带分解,分离图像中的解剖结构与设备相关外观特征。
- 在5个未见数据集上,平衡Dice分数领先,95%分位豪斯多夫距离最低。
- 适合医疗图像分割中缺乏多域标注数据的场景,尤其关注跨设备稳定性。
由于设备和临床环境差异导致的眼底图像域间变化,使深度学习模型在未见域上性能下降,而跨域标注数据获取成本高且受隐私限制。现有单源域泛化方法难以区分解剖拓扑与外观特征。本文提出WaveSDG,一种基于小波引导的分割网络,通过小波子带分解实现解剖结构与域特定外观的解耦。设计新型小波不变结构提取与优化(WISER)模块,对编码器特征进行处理:低频子带强化全局解剖结构,高频子带选择性增强方向边缘并抑制噪声。大量消融实验证明了WISER模块的有效性及其解耦策略。在1个源域和5个未见目标域上进行视杯与视盘分割评估,WaveSDG持续优于7种前沿方法,达到最优平衡Dice分数和最低95%分位豪斯多夫距离,且方差更小,表明其精度、鲁棒性和跨域稳定性显著提升。
原文摘要 · Abstract (English)
Domain generalization in fundus imaging is challenging due to variations in acquisition conditions across devices and clinical settings. The inability to adapt to these variations causes performance degradation on unseen domains for deep learning models. Besides, obtaining annotated data across domains is often expensive and privacy constraints restricts their availability. Although single-source domain generalization (SDG) offers a realistic solution to this problem, the existing approaches frequently fail to capture anatomical topology or decouple appearance from anatomical features. This research introduces WaveSDG, a new wavelet-guided segmentation network for SDG. It decouples anatomical structure from domain-specific appearance through a wavelet sub-band decomposition. A novel Wavelet-based Invariant Structure Extraction and Refinement (WISER) module is proposed to process encoder features by leveraging distinct semantic roles of each wavelet sub-band. The module refines low-frequency components to anchor global anatomy, while selectively enhancing directional edges and suppressing noise within the high-frequency sub-bands. Extensive ablation studies validate the effectiveness of the WISER module and its decoupling strategy. Our evaluations on optic cup and optic disc segmentation across one source and five unseen target datasets show that WaveSDG consistently outperforms seven state-of-the-art methods. Notably, it achieves the best balanced Dice score and lowest 95th percentile Hausdorff distance with reduced variance, indicating improved accuracy, robustness, and cross-domain stability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。