arXiv:2601.13417cs.CV2026-01

用快速几何对齐方法提升眼底图像质量,更准确保留疾病特征

SGW-GAN: Sliced Gromov-Wasserstein Guided GANs for Retinal Fundus Image Enhancement

论文配图:SGW-GAN: Sliced Gromov-Wasserstein Guided GANs for Retinal Fundus Image Enhancement
图 1 · 摘自论文原文
  • 引入切片格罗莫夫-沃瑟斯坦距离,低成本保持图像内在结构
  • 在糖尿病视网膜病变分级任务中表现最佳,GW差异最低
  • 适合医疗图像增强,尤其关注疾病分类精度的研究者

眼底照相对眼科筛查诊断至关重要,但图像常受噪声、伪影和光照不均影响。现有基于GAN和扩散模型的增强方法虽提升感知质量,但会破坏类内几何结构:临床相关样本分散,病种边界模糊,影响分级或病灶检测等下游任务。格罗莫夫-沃瑟斯坦(GW)差异通过内部成对距离对齐分布,可自然保留类内结构,但计算成本高。为此,我们提出首个将切片GW(SGW)用于眼底图像增强的SGW-GAN框架。SGW通过随机投影近似GW,保持关系保真度的同时大幅降低开销。在公开数据集上的实验表明,SGW-GAN生成视觉上令人信服的增强结果,在糖尿病视网膜病变分级中表现优异,并在各病种标签下实现最低的GW差异,验证了其在无配对医学图像增强中的高效性与临床保真度。

原文摘要 · Abstract (English)

Retinal fundus photography is indispensable for ophthalmic screening and diagnosis, yet image quality is often degraded by noise, artifacts, and uneven illumination. Recent GAN- and diffusion-based enhancement methods improve perceptual quality by aligning degraded images with high-quality distributions, but our analysis shows that this focus can distort intra-class geometry: clinically related samples become dispersed, disease-class boundaries blur, and downstream tasks such as grading or lesion detection are harmed. The Gromov Wasserstein (GW) discrepancy offers a principled solution by aligning distributions through internal pairwise distances, naturally preserving intra-class structure, but its high computational cost restricts practical use. To overcome this, we propose SGW-GAN, the first framework to incorporate Sliced GW (SGW) into retinal image enhancement. SGW approximates GW via random projections, retaining relational fidelity while greatly reducing cost. Experiments on public datasets show that SGW-GAN produces visually compelling enhancements, achieves superior diabetic retinopathy grading, and reports the lowest GW discrepancy across disease labels, demonstrating both efficiency and clinical fidelity for unpaired medical image enhancement.

图像增强眼底图像生成对抗网络几何对齐

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