Physics-Informed Neural Koopman Machine for Interpretable Longitudinal Personalized Alzheimer's Disease Forecasting

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Hrusanov, Georgi, Vu, Duy-Thanh, Can, Duy-Cat, Tascedda, Sophie, Ryan, Margaret, Bodelet, Julien, Koscielska, Katarzyna, Magnus, Carsten, Chén, Oliver Y.
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915657924214784
author Hrusanov, Georgi
Vu, Duy-Thanh
Can, Duy-Cat
Tascedda, Sophie
Ryan, Margaret
Bodelet, Julien
Koscielska, Katarzyna
Magnus, Carsten
Chén, Oliver Y.
author_facet Hrusanov, Georgi
Vu, Duy-Thanh
Can, Duy-Cat
Tascedda, Sophie
Ryan, Margaret
Bodelet, Julien
Koscielska, Katarzyna
Magnus, Carsten
Chén, Oliver Y.
contents Early forecasting of individual cognitive decline in Alzheimer's disease (AD) is central to disease evaluation and management. Despite advances, it is as of yet challenging for existing methodological frameworks to integrate multimodal data for longitudinal personalized forecasting while maintaining interpretability. To address this gap, we present the Neural Koopman Machine (NKM), a new machine learning architecture inspired by dynamical systems and attention mechanisms, designed to forecast multiple cognitive scores simultaneously using multimodal genetic, neuroimaging, proteomic, and demographic data. NKM integrates analytical ($α$) and biological ($β$) knowledge to guide feature grouping and control the hierarchical attention mechanisms to extract relevant patterns. By implementing Fusion Group-Aware Hierarchical Attention within the Koopman operator framework, NKM transforms complex nonlinear trajectories into interpretable linear representations. To demonstrate NKM's efficacy, we applied it to study the Alzheimer's Disease Neuroimaging Initiative (ADNI) dataset. Our results suggest that NKM consistently outperforms both traditional machine learning methods and deep learning models in forecasting trajectories of cognitive decline. Specifically, NKM (1) forecasts changes of multiple cognitive scores simultaneously, (2) quantifies differential biomarker contributions to predicting distinctive cognitive scores, and (3) identifies brain regions most predictive of cognitive deterioration. Together, NKM advances personalized, interpretable forecasting of future cognitive decline in AD using past multimodal data through an explainable, explicit system and reveals potential multimodal biological underpinnings of AD progression.
format Preprint
id arxiv_https___arxiv_org_abs_2512_06134
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Physics-Informed Neural Koopman Machine for Interpretable Longitudinal Personalized Alzheimer's Disease Forecasting
Hrusanov, Georgi
Vu, Duy-Thanh
Can, Duy-Cat
Tascedda, Sophie
Ryan, Margaret
Bodelet, Julien
Koscielska, Katarzyna
Magnus, Carsten
Chén, Oliver Y.
Machine Learning
Artificial Intelligence
Neurons and Cognition
Quantitative Methods
Early forecasting of individual cognitive decline in Alzheimer's disease (AD) is central to disease evaluation and management. Despite advances, it is as of yet challenging for existing methodological frameworks to integrate multimodal data for longitudinal personalized forecasting while maintaining interpretability. To address this gap, we present the Neural Koopman Machine (NKM), a new machine learning architecture inspired by dynamical systems and attention mechanisms, designed to forecast multiple cognitive scores simultaneously using multimodal genetic, neuroimaging, proteomic, and demographic data. NKM integrates analytical ($α$) and biological ($β$) knowledge to guide feature grouping and control the hierarchical attention mechanisms to extract relevant patterns. By implementing Fusion Group-Aware Hierarchical Attention within the Koopman operator framework, NKM transforms complex nonlinear trajectories into interpretable linear representations. To demonstrate NKM's efficacy, we applied it to study the Alzheimer's Disease Neuroimaging Initiative (ADNI) dataset. Our results suggest that NKM consistently outperforms both traditional machine learning methods and deep learning models in forecasting trajectories of cognitive decline. Specifically, NKM (1) forecasts changes of multiple cognitive scores simultaneously, (2) quantifies differential biomarker contributions to predicting distinctive cognitive scores, and (3) identifies brain regions most predictive of cognitive deterioration. Together, NKM advances personalized, interpretable forecasting of future cognitive decline in AD using past multimodal data through an explainable, explicit system and reveals potential multimodal biological underpinnings of AD progression.
title Physics-Informed Neural Koopman Machine for Interpretable Longitudinal Personalized Alzheimer's Disease Forecasting
topic Machine Learning
Artificial Intelligence
Neurons and Cognition
Quantitative Methods
url https://arxiv.org/abs/2512.06134