通过结构与风格增强提升视网膜血管分割泛化能力
DGSSA: Domain generalization with structural and stylistic augmentation for retinal vessel segmentation
- 用空间殖民算法生成逼真血管结构,结合Pix2Pix生成伪图像扩充数据
- 在四个数据集上达到当前最佳性能,显著提升跨域适应性
- 适合医学图像分析、临床辅助诊断等实际应用
视网膜血管形态对糖尿病、青光眼和高血压等疾病的诊断至关重要,准确分割有助于早期干预。传统方法假设训练与测试数据分布一致,但在成像设备和患者群体差异导致的领域偏移下表现不佳。本文提出DGSSA方法,通过结合结构与风格增强策略提升模型泛化能力。利用空间殖民算法生成多样化的血管样结构,再通过改进的Pix2Pix模型生成伪视网膜图像,使分割模型学习更广泛的结构分布。同时采用PixMix实现随机光度增强并引入不确定性扰动,丰富风格多样性,显著提升模型对不同成像条件的适应能力。在DRIVE、CHASEDB、HRF和STARE四个挑战性数据集上验证,性能超越现有方法,证明了该框架的有效性,具备临床自动化视网膜血管分析的应用潜力。
原文摘要 · Abstract (English)
Retinal vascular morphology is crucial for diagnosing diseases such as diabetes, glaucoma, and hypertension, making accurate segmentation of retinal vessels essential for early intervention. Traditional segmentation methods assume that training and testing data share similar distributions, which can lead to poor performance on unseen domains due to domain shifts caused by variations in imaging devices and patient demographics. This paper presents a novel approach, DGSSA, for retinal vessel image segmentation that enhances model generalization by combining structural and style augmentation strategies. We utilize a space colonization algorithm to generate diverse vascular-like structures that closely mimic actual retinal vessels, which are then used to generate pseudo-retinal images with an improved Pix2Pix model, allowing the segmentation model to learn a broader range of structure distributions. Additionally, we utilize PixMix to implement random photometric augmentations and introduce uncertainty perturbations, thereby enriching stylistic diversity and significantly enhancing the model's adaptability to varying imaging conditions. Our framework has been rigorously evaluated on four challenging datasets-DRIVE, CHASEDB, HRF, and STARE-demonstrating state-of-the-art performance that surpasses existing methods. This validates the effectiveness of our proposed approach, highlighting its potential for clinical application in automated retinal vessel analysis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。