arXiv:2410.14388cs.LG2024-10NeurIPS被引 5

用最优传输加速疾病进展建模,实现千倍提速与高维特征分析。

Unscrambling disease progression at scale: fast inference of event permutations with optimal transport

  • 将疾病进程建模为属于Birkhoff多面体的潜在排列矩阵,通过变分下界优化实现快速推断。
  • 推理速度比现有方法快1000倍,支持数百倍更多特征,模拟中精度和抗噪性显著提升。
  • 首次在脑部与眼部影像中实现像素级疾病进展事件解析,适用于各类渐进性疾病。

疾病进展模型可推断患者特征随慢性退行性疾病演进的群体级时间轨迹,为疾病生物学和个体化临床分期提供独特洞见。离散模型将疾病进展视为事件的潜在排列,每个事件对应一个特征变得可测量异常。然而,传统最大似然方法因组合爆炸导致排列推断变得不可行,严重限制了模型维度与实用性。本文借鉴最优传输思想,将疾病进展建模为属于Birkhoff多面体的潜在排列矩阵,通过优化变分下界实现快速推断。该方法使推理速度比当前最先进方法快1000倍,相应支持比现有方法多几个数量级的特征。模拟实验表明其在速度、准确性和抗噪声方面均有提升。真实世界影像数据实验(阿尔茨海默病与年龄相关性黄斑变性)首次实现了脑部与眼部的像素级疾病进展事件可视化。本方法计算量低、可解释性强,适用于任何进行性疾病和数据模态,具有广泛临床应用潜力。

原文摘要 · Abstract (English)

Disease progression models infer group-level temporal trajectories of change in patients' features as a chronic degenerative condition plays out. They provide unique insight into disease biology and staging systems with individual-level clinical utility. Discrete models consider disease progression as a latent permutation of events, where each event corresponds to a feature becoming measurably abnormal. However, permutation inference using traditional maximum likelihood approaches becomes prohibitive due to combinatoric explosion, severely limiting model dimensionality and utility. Here we leverage ideas from optimal transport to model disease progression as a latent permutation matrix of events belonging to the Birkhoff polytope, facilitating fast inference via optimisation of the variational lower bound. This enables a factor of 1000 times faster inference than the current state of the art and, correspondingly, supports models with several orders of magnitude more features than the current state of the art can consider. Experiments demonstrate the increase in speed, accuracy and robustness to noise in simulation. Further experiments with real-world imaging data from two separate datasets, one from Alzheimer's disease patients, the other age-related macular degeneration, showcase, for the first time, pixel-level disease progression events in the brain and eye, respectively. Our method is low compute, interpretable and applicable to any progressive condition and data modality, giving it broad potential clinical utility.

疾病进展最优传输高维建模医学影像

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