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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Retinal Vessel Segmentation | CHASE DB1 | -- | 47 | |
| Retinal Vessel Segmentation | STARE | F1 Score81.45 | 40 | |
| Retinal Vessel Segmentation | DRIVE | F1 Score0.7859 | 33 | |
| Retinal Vessel Segmentation | RECOVERY FA19 | Dice41.75 | 17 | |
| Retinal Vessel Segmentation | DRIVE S1 (in-domain) | Dice80.46 | 15 | |
| Retinal Vessel Segmentation | STARE S2 (in-domain) | Dice79.39 | 15 | |
| Retinal Vessel Segmentation | CHASE_DB1 S3 (in-domain) | Dice81.94 | 15 | |
| Retinal Vessel Segmentation | RECOVERY S4 FA19 (in-domain) | Dice60.33 | 15 |