arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.09463cs.LGcs.AIstat.AP
收藏

Mixed Effects Neural ODE: A Variational Approximation for Analyzing the Dynamics of Panel Data

University of Wisconsin Madison(威斯康星大学麦迪逊分校);Amazon Lab 126(亚马逊126实验室);University of Illinois at Chicago(伊利诺伊大学芝加哥分校)

Jurijs Nazarovs, Rudrasis Chakraborty, Songwong Tasneeyapant, Sathya N. Ravi, Vikas Singh

英文摘要:

Panel data involving longitudinal measurements of the same set of participants taken over multiple time points is common in studies to understand childhood development and disease modeling. Deep hybrid models that marry the predictive power of neural networks with physical simulators such as differential equations, are starting to drive advances in such applications. The task of modeling not just the observations but the hidden dynamics that are captured by the measurements poses interesting statistical/computational questions. We propose a probabilistic model called ME-NODE to incorporate (fixed + random) mixed effects for analyzing such panel data. We show that our model can be derived using smooth approximations of SDEs provided by the Wong-Zakai theorem. We then derive Evidence Based Lower Bounds for ME-NODE, and develop (efficient) training algorithms using MC based sampling methods and numerical ODE solvers. We demonstrate ME-NODE's utility on tasks spanning the spectrum from simulations and toy data to real longitudinal 3D imaging data from an Alzheimer's disease (AD) study, and study its performance in terms of accuracy of reconstruction for interpolation, uncertainty estimates and personalized prediction.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑