arXiv:2602.24013cs.CV2026-02中稿 · manuscript

用有序标签生成糖尿病视网膜病变眼底图像,更真实连续。

Ordinal Diffusion Models for Color Fundus Images

  • 用标量表示疾病程度,替代离散类别,实现平滑过渡。
  • 在EyePACS数据集上,五阶段中四阶段的生成质量提升,一致性指标κ升至0.87。
  • 适合需要连续病理进展模拟的医学图像生成任务。

生成式图像模型如扩散模型可通过提供补充训练数据提升临床相关任务表现。然而,多数条件扩散模型将疾病阶段视为独立类别,忽略了疾病进展的连续性。这一问题在医学影像中尤为突出,因病理过程通常仅以粗粒度、离散但有序的标签记录,如糖尿病视网膜病变(DR)分级。本文提出一种用于生成彩色眼底图像的有序潜空间扩散模型,将DR严重程度的有序结构显式融入生成过程。不同于类别条件,采用标量疾病表示,实现相邻阶段间的平滑过渡。在EyePACS数据集上,通过视觉真实性指标与基于分类的临床一致性分析评估。相比标准条件扩散模型,本方法在五个DR阶段中的四个降低了弗雷谢初始距离,并将二次加权κ从0.79提升至0.87。插值实验表明,模型成功捕捉了由粗粒度有序标签学习到的连续疾病进展谱。代码已公开于https://github.com/berenslab/OrdinalDiffusionModels。

原文摘要 · Abstract (English)

Generative image models such as diffusion models can improve performance on clinically relevant tasks by offering deep learning models supplementary training data. However, most conditional diffusion models treat disease stages as independent classes, ignoring the continuous nature of disease progression. This mismatch is problematic in medical imaging because continuous pathological processes are typically only observed through coarse, discrete but ordered labels as in ophthalmology for diabetic retinopathy (DR). We propose an ordinal latent diffusion model for generating color fundus images that explicitly incorporates the ordered structure of DR severity into the generation process. Instead of categorical conditioning, we used a scalar disease representation, enabling a smooth transition between adjacent stages. We evaluated our approach using visual realism metrics and classification-based clinical consistency analysis on the EyePACS dataset. Compared to a standard conditional diffusion model, our model reduced the Fréchet inception distance for four of the five DR stages and increased the quadratic weighted $κ$ from 0.79 to 0.87. Furthermore, interpolation experiments showed that the model captured a continuous spectrum of disease progression learned from ordered, coarse class labels. Code available at https://github.com/berenslab/OrdinalDiffusionModels.

扩散模型医学图像有序生成眼底图像

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