用X光预测膝骨关节炎进展,生成未来影像并定位关键部位。
Risk Estimation of Knee Osteoarthritis Progression via Predictive Multi-task Modelling from Efficient Diffusion Model using X-ray Images
- 基于扩散模型在条件潜空间生成未来膝关节影像
- 预测疾病进展准确率AUC达0.71,优于当前最优水平2%
- 可定位解剖标志,提升临床可解释性,推理速度快9%
医学影像在评估膝骨关节炎(OA)风险中至关重要,有助于早期发现和疾病监测。近年来,机器学习方法在风险估计(即预测疾病进展可能性)和预测建模(即基于当前数据预测未来结果)方面取得进展,但因缺乏可解释性,临床应用受限。现有生成未来影像的方法复杂且不实用,且无法精确定位膝关节解剖标志,影响可解释性。本文提出一种新的可解释机器学习方法,通过多任务预测建模,从高效生成的高质量未来影像中分类未来膝骨关节炎严重程度并预测解剖标志。该图像生成利用扩散模型在类别条件潜空间中预测疾病进展,提供健康状况演化的可视化表示。在Osteoarthritis Initiative数据集上,本方法将现有最先进水平提升2%,预测膝骨关节炎进展的AUC达0.71,同时实现约9%的推理速度提升。
原文摘要 · Abstract (English)
Medical imaging plays a crucial role in assessing knee osteoarthritis (OA) risk by enabling early detection and disease monitoring. Recent machine learning methods have improved risk estimation (i.e., predicting the likelihood of disease progression) and predictive modelling (i.e., the forecasting of future outcomes based on current data) using medical images, but clinical adoption remains limited due to their lack of interpretability. Existing approaches that generate future images for risk estimation are complex and impractical. Additionally, previous methods fail to localize anatomical knee landmarks, limiting interpretability. We address these gaps with a new interpretable machine learning method to estimate the risk of knee OA progression via multi-task predictive modelling that classifies future knee OA severity and predicts anatomical knee landmarks from efficiently generated high-quality future images. Such image generation is achieved by leveraging a diffusion model in a class-conditioned latent space to forecast disease progression, offering a visual representation of how particular health conditions may evolve. Applied to the Osteoarthritis Initiative dataset, our approach improves the state-of-the-art (SOTA) by 2\%, achieving an AUC of 0.71 in predicting knee OA progression while offering ~9% faster inference time.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。