Predicting future disease progression risk from medical images is challenging due to patient heterogeneity, and subtle or unknown imaging biomarkers. Moreover, deep learning (DL) methods for survival analysis are susceptible to image domain shifts across scanners. We tackle these issues in the task of predicting late dry Age-related Macular Degeneration (dAMD) onset from retinal OCT scans. We propose a novel DL method for survival prediction to jointly predict from the current scan a risk score, inversely related to time-to-conversion, and the probability of conversion within a time interval $t$. It uses a family of parallel hyperplanes generated by parameterizing the bias term as a function of $t$. In addition, we develop unsupervised losses based on intra-subject image pairs to ensure that risk scores increase over time and that future conversion predictions are consistent with AMD stage prediction using actual scans of future visits. Such losses enable data-efficient fine-tuning of the trained model on new unlabeled datasets acquired with a different scanner. Extensive evaluation on two large datasets acquired with different scanners resulted in a mean AUROCs of 0.82 for Dataset-1 and 0.83 for Dataset-2, across prediction intervals of 6,12 and 24 months.
翻译:从医学图像预测未来疾病进展风险具有挑战性,原因在于患者异质性以及成像生物标志物的微妙性或未知性。此外,用于生存分析的深度学习方法易受不同扫描仪间图像域偏移的影响。我们在从视网膜OCT扫描预测晚期干性年龄相关性黄斑变性发病的任务中应对这些问题。我们提出了一种新颖的用于生存预测的深度学习方法,能够从当前扫描联合预测一个风险评分(与转化时间成反比)以及在时间间隔$t$内发生转化的概率。该方法使用一族平行超平面,通过将偏置项参数化为$t$的函数来生成。此外,我们开发了基于受试者内部图像对的无监督损失,以确保风险评分随时间增加,并且未来的转化预测与使用未来访视实际扫描进行的AMD分期预测保持一致。此类损失使得训练好的模型能够在由不同扫描仪获取的新未标记数据集上进行数据高效微调。在两个由不同扫描仪获取的大型数据集上的广泛评估结果显示,在6、12和24个月的预测区间内,数据集1的平均AUROC为0.82,数据集2的平均AUROC为0.83。