A Biologically Constrained Continuous-Time Framework for Long-Horizon Cognition Forecasting in Alzheimer’s Disease
Discuss this preprint
Start a discussion What are Sciety discussions?Listed in
This article is not in any list yet, why not save it to one of your lists.Abstract
Accurate long-term forecasting of cognitive trajectories across the Alzheimer’s disease continuum is essential for early intervention, personalized prognosis, patient stratification, and clinical trial enrichment. Despite the promising predictive performance of recent longitudinal forecasting methods, they remain largely data-driven, struggle with irregularly sampled, incomplete longitudinal data and often neglect established disease biology, leading to biologically implausible trajectories. To address this, we propose a biologically constrained continuous-time framework for long-horizon cognition forecasting from limited baseline observations. The proposed method models the complete amy-loid–tau–vascular–neurodegeneration–cognition (ATVNC) cascade using hierarchical Neural ODEs with biologically motivated monotonicity constraints. Each pathological stream is governed by a dedicated Neural ODE initialized from irregular longitudinal observations using a GRU-D encoder, capturing intrinsic disease evolution while being modulated by directed upstream pathological influences. A bounded cognition readout ensures physiologically valid cognitive score (MoCA) predictions, while teacher–student knowledge distillation improves learning from sparse lon-gitudinal supervision. Evaluated on the ADNI dataset, the proposed framework achieves a long-horizon extrapolation MAE of 2.06 on 188 held-out participants while eliminating biologically implausible trajectory violations. It further demonstrates robust zero-shot cross-cohort generalization on OASIS-3 (MAE 2.68 on 300 participants), with fine-tuning improving MAE to 1.90. The model also supports prognostic enrichment for Alzheimer’s clinical trials, achieving up to 2.70 × enrichment over the cohort base rate. These results demonstrate that embedding biological disease mechanisms within continuous-time deep learning improves the accuracy, biological plausibility, and clinical utility of long-horizon cognitive forecasting. The code is publicly available at: https://github.com/PonDeepika/BEACON .