针对糖尿病视网膜病变分级中的类别不均衡问题,提出新型生成式模型提升小样本病程阶段识别精度。
Ordinal Label-Distribution Learning with Constrained Asymmetric Priors for Imbalanced Retinal Grading
- 设计非对称先验的WAE框架,保留少数类长尾分布特征
- 通过方向感知损失与动态权重,使模型更重视低估病程的错误
- 在公开数据集上实现最佳综合性能,适合医疗影像诊断场景
糖尿病视网膜病变分级具有固有的序数性和长尾分布特征,少数病程阶段样本稀少、异质性强且临床重要。传统方法多依赖各向同性高斯先验和对称损失函数,与任务的非对称特性不匹配。本文提出约束非对称先验的Wasserstein自编码器(CAP-WAE),通过三个关键创新解决该问题:首先,采用将聚合后验与非对称先验对齐的WAE,保持少数类的重尾和偏斜结构;其次,引入边界感知正交性与紧凑性(MAOC)损失,确保潜在空间中等级有序可分;第三,在监督层面设计方向感知序数损失,轻量级头预测非对称离散度,生成反映临床优先级的软标签,更严厉惩罚低估病程的错误。通过自适应多任务加权方案稳定训练,模型端到端实现,仅需少量调参。在多个公开糖尿病视网膜病变基准测试中,CAP-WAE在加权κ值、准确率和宏平均F1上均达到当前最优,超越传统序数分类与潜在生成基线。t-SNE可视化显示,本方法将潜在流形重构为紧凑、有序、重叠更少的聚类。
原文摘要 · Abstract (English)
Diabetic retinopathy grading is inherently ordinal and long-tailed, with minority stages being scarce, heterogeneous, and clinically critical to detect accurately. Conventional methods often rely on isotropic Gaussian priors and symmetric loss functions, misaligning latent representations with the task's asymmetric nature. We propose the Constrained Asymmetric Prior Wasserstein Autoencoder (CAP-WAE), a novel framework that addresses these challenges through three key innovations. Our approach employs a Wasserstein Autoencoder (WAE) that aligns its aggregate posterior with a asymmetric prior, preserving the heavy-tailed and skewed structure of minority classes. The latent space is further structured by a Margin-Aware Orthogonality and Compactness (MAOC) loss to ensure grade-ordered separability. At the supervision level, we introduce a direction-aware ordinal loss, where a lightweight head predicts asymmetric dispersions to generate soft labels that reflect clinical priorities by penalizing under-grading more severely. Stabilized by an adaptive multi-task weighting scheme, our end-to-end model requires minimal tuning. Across public DR benchmarks, CAP-WAE consistently achieves state-of-the-art Quadratic Weighted Kappa, accuracy, and macro-F1, surpassing both ordinal classification and latent generative baselines. t-SNE visualizations further reveal that our method reshapes the latent manifold into compact, grade-ordered clusters with reduced overlap.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。