用证据推理和原型记忆提升膝骨关节炎分级的准确率与可信度
ClinNet: Evidential Ordinal Regression with Bilateral Asymmetry and Prototype Memory for Knee Osteoarthritis Grading
- 通过双侧结构差异建模和原型记忆稳定特征表示
- 在KOA分级上达到0.892的加权卡帕值,显著优于现有方法
- 能有效识别异常样本和误诊,适合临床可信部署
基于放射影像的膝骨关节炎(KOA)分级是关键但极具挑战的任务,因其各阶段间差异细微、标注存在不确定性,且疾病进展具有序数特性。传统深度学习方法通常将其视为确定性多分类问题,忽略了退变过程的连续性及专家标注的不确定性。本文提出ClinNet,一种可信的新型框架,将KOA分级建模为证据序数回归问题。该方法包含三个核心组件:(1) 双侧不对称编码器(BAE),显式建模内侧-外侧结构差异;(2) 诊断记忆库,维护类别级原型以稳定特征表示;(3) 基于正态逆伽马分布(NIG)的证据序数头,联合估计连续KL分级与认知不确定性。大量实验表明,ClinNet在加权卡帕值上达到0.892,准确率达0.768,显著优于当前最优基线(p < 0.001)。更重要的是,模型的不确定性估计能有效识别分布外样本与潜在误诊,为临床安全部署奠定基础。
原文摘要 · Abstract (English)
Knee osteoarthritis (KOA) grading based on radiographic images is a critical yet challenging task due to subtle inter-grade differences, annotation uncertainty, and the inherently ordinal nature of disease progression. Conventional deep learning approaches typically formulate this problem as deterministic multi-class classification, ignoring both the continuous progression of degeneration and the uncertainty in expert annotations. In this work, we propose ClinNet, a novel trustworthy framework that addresses KOA grading as an evidential ordinal regression problem. The proposed method integrates three key components: (1) a Bilateral Asymmetry Encoder (BAE) that explicitly models medial-lateral structural discrepancies; (2) a Diagnostic Memory Bank that maintains class-wise prototypes to stabilize feature representations; and (3) an Evidential Ordinal Head based on the Normal-Inverse-Gamma (NIG) distribution to jointly estimate continuous KL grades and epistemic uncertainty. Extensive experiments demonstrate that ClinNet achieves a Quadratic Weighted Kappa of 0.892 and Accuracy of 0.768, statistically outperforming state-of-the-art baselines (p < 0.001). Crucially, we demonstrate that the model's uncertainty estimates successfully flag out-of-distribution samples and potential misdiagnoses, paving the way for safe clinical deployment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。