Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

WaveRNet: Wavelet-Guided Frequency Learning for Multi-Source Domain-Generalized Retinal Vessel Segmentation

About

Domain-generalized retinal vessel segmentation is critical for automated ophthalmic diagnosis, yet faces significant challenges from domain shift induced by non-uniform illumination and varying contrast, compounded by the difficulty of preserving fine vessel structures. While the Segment Anything Model (SAM) exhibits remarkable zero-shot capabilities, existing SAM-based methods rely on simple adapter fine-tuning while overlooking frequency-domain information that encodes domain-invariant features, resulting in degraded generalization under illumination and contrast variations. Furthermore, SAM's direct upsampling inevitably loses fine vessel details. To address these limitations, we propose WaveRNet, a wavelet-guided frequency learning framework for robust multi-source domain-generalized retinal vessel segmentation. Specifically, we devise a Spectral-guided Domain Modulator (SDM) that integrates wavelet decomposition with learnable domain tokens, enabling the separation of illumination-robust low-frequency structures from high-frequency vessel boundaries while facilitating domain-specific feature generation. Furthermore, we introduce a Frequency-Adaptive Domain Fusion (FADF) module that performs intelligent test-time domain selection through wavelet-based frequency similarity and soft-weighted fusion. Finally, we present a Hierarchical Mask-Prompt Refiner (HMPR) that overcomes SAM's upsampling limitation through coarse-to-fine refinement with long-range dependency modeling. Extensive experiments under the Leave-One-Domain-Out protocol on four public retinal datasets demonstrate that WaveRNet achieves state-of-the-art generalization performance. The source code is available at https://github.com/Chanchan-Wang/WaveRNet.

Chanchan Wang, Yuanfang Wang, Qing Xu, Guanxin Chen• 2026

Related benchmarks

TaskDatasetResultRank
Retinal Vessel SegmentationCHASE DB1--
47
Retinal Vessel SegmentationSTARE
F1 Score81.45
40
Retinal Vessel SegmentationDRIVE
F1 Score0.7859
33
Retinal Vessel SegmentationRECOVERY FA19
Dice41.75
17
Retinal Vessel SegmentationDRIVE S1 (in-domain)
Dice80.46
15
Retinal Vessel SegmentationSTARE S2 (in-domain)
Dice79.39
15
Retinal Vessel SegmentationCHASE_DB1 S3 (in-domain)
Dice81.94
15
Retinal Vessel SegmentationRECOVERY S4 FA19 (in-domain)
Dice60.33
15
Showing 8 of 8 rows

Other info

Follow for update