A Stage-Aware Mixture of Experts Framework for Neurodegenerative Disease Progression Modelling
- URL: http://arxiv.org/abs/2508.07032v1
- Date: Sat, 09 Aug 2025 16:09:00 GMT
- Title: A Stage-Aware Mixture of Experts Framework for Neurodegenerative Disease Progression Modelling
- Authors: Tiantian He, Keyue Jiang, An Zhao, Anna Schroder, Elinor Thompson, Sonja Soskic, Frederik Barkhof, Daniel C. Alexander,
- Abstract summary: Long-term progression of neurodegenerative diseases is commonly conceptualized as a graph diffusion process.<n>We propose a novel stage-aware Mixture Experts framework that explicitly models how different contributing mechanisms dominate at different disease stages.<n>The resulting IGND-MoE model dynamically integrates these components across temporal states, offering a principled way to understand how stage-specific pathological mechanisms contribute to progression.
- Score: 6.577465893404979
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: The long-term progression of neurodegenerative diseases is commonly conceptualized as a spatiotemporal diffusion process that consists of a graph diffusion process across the structural brain connectome and a localized reaction process within brain regions. However, modeling this progression remains challenging due to 1) the scarcity of longitudinal data obtained through irregular and infrequent subject visits and 2) the complex interplay of pathological mechanisms across brain regions and disease stages, where traditional models assume fixed mechanisms throughout disease progression. To address these limitations, we propose a novel stage-aware Mixture of Experts (MoE) framework that explicitly models how different contributing mechanisms dominate at different disease stages through time-dependent expert weighting.Data-wise, we utilize an iterative dual optimization method to properly estimate the temporal position of individual observations, constructing a co hort-level progression trajectory from irregular snapshots. Model-wise, we enhance the spatial component with an inhomogeneous graph neural diffusion model (IGND) that allows diffusivity to vary based on node states and time, providing more flexible representations of brain networks. We also introduce a localized neural reaction module to capture complex dynamics beyond standard processes.The resulting IGND-MoE model dynamically integrates these components across temporal states, offering a principled way to understand how stage-specific pathological mechanisms contribute to progression. The stage-wise weights yield novel clinical insights that align with literature, suggesting that graph-related processes are more influential at early stages, while other unknown physical processes become dominant later on.
Related papers
- CLARITY: Medical World Model for Guiding Treatment Decisions by Modeling Context-Aware Disease Trajectories in Latent Space [49.74032713886216]
CLARITY is a medical world model that forecasts disease evolution directly within a structured latent space.<n>It explicitly integrates time intervals (temporal context) and patient-specific data (clinical context) to model treatment-conditioned progression as a smooth, interpretable trajectory.
arXiv Detail & Related papers (2025-12-08T20:42:10Z) - LLM enhanced graph inference for long-term disease progression modelling [11.745038114408395]
We present a novel framework that uses Large Language Models (MsLL) as expert guides on the interaction of regional variables to enhance learning of disease progression.<n>We demonstrate the new approach by estimating the pathology propagation using tau-PET imaging data from an Alzheimer's disease cohort.
arXiv Detail & Related papers (2025-11-14T02:03:10Z) - Conditional Neural ODE for Longitudinal Parkinson's Disease Progression Forecasting [51.906871559732245]
Parkinson's disease (PD) shows heterogeneous, evolving brain-morphometry patterns.<n>Modeling these longitudinal trajectories enables mechanistic insight, treatment development, and individualized 'digital-twin' forecasting.<n>We propose CNODE, a novel framework for continuous, individualized PD progression forecasting.
arXiv Detail & Related papers (2025-11-06T20:16:33Z) - Disentangling Neurodegeneration with Brain Age Gap Prediction Models: A Graph Signal Processing Perspective [89.99666725996975]
The brain age gap prediction (BAGP) models estimate the difference between a person's predicted brain age from data and their chronological age.<n>This tutorial article provides an overview of BAGP and introduces a principled framework for this application based on recent advancements in graph signal processing (GSP)<n>VNNs offer strong theoretical grounding and operational interpretability, enabling robust estimation of brain age gap predictions.
arXiv Detail & Related papers (2025-10-14T17:44:45Z) - NEUBORN: The Neurodevelopmental Evolution framework Using BiOmechanical RemodelliNg [1.62955730276521]
We present a novel framework for learning individual growth trajectories from biomechanically constrained, longitudinal, diffeomorphic image registration.<n>The method improves the biological plausibility of warps, generating growth trajectories that better follow population-level trends.<n>This framework opens new possibilities for predictive modeling of brain maturation and early identification of malformations of cortical development.
arXiv Detail & Related papers (2025-08-13T12:36:23Z) - Langevin Flows for Modeling Neural Latent Dynamics [81.81271685018284]
We introduce LangevinFlow, a sequential Variational Auto-Encoder where the time evolution of latent variables is governed by the underdamped Langevin equation.<n>Our approach incorporates physical priors -- such as inertia, damping, a learned potential function, and forces -- to represent both autonomous and non-autonomous processes in neural systems.<n>Our method outperforms state-of-the-art baselines on synthetic neural populations generated by a Lorenz attractor.
arXiv Detail & Related papers (2025-07-15T17:57:48Z) - High-throughput digital twin framework for predicting neurite deterioration using MetaFormer attention [2.0971479389679337]
Neurodevelopmental disorders (NDDs) cover a variety of conditions, including autism spectrum disorder, attention-deficit/hyperactivity disorder, and epilepsy.<n>Their high comorbidity and complex etiologies present significant challenges for accurate diagnosis and effective treatments.<n>This paper introduces a high- throughput digital twin framework for neurite deteriorations associated with NDDs.
arXiv Detail & Related papers (2024-12-18T01:48:50Z) - SkelMamba: A State Space Model for Efficient Skeleton Action Recognition of Neurological Disorders [14.304356695180005]
We introduce a novel state-space model (SSM)-based framework for skeleton-based human action recognition.<n>Our model captures local joint interactions and global motion patterns across multiple body parts.<n>This gait-aware decomposition enhances the ability to identify subtle motion patterns critical in medical diagnosis.
arXiv Detail & Related papers (2024-11-29T08:43:52Z) - Interpretable Spatio-Temporal Embedding for Brain Structural-Effective Network with Ordinary Differential Equation [56.34634121544929]
In this study, we first construct the brain-effective network via the dynamic causal model.
We then introduce an interpretable graph learning framework termed Spatio-Temporal Embedding ODE (STE-ODE)
This framework incorporates specifically designed directed node embedding layers, aiming at capturing the dynamic interplay between structural and effective networks.
arXiv Detail & Related papers (2024-05-21T20:37:07Z) - Persistent-Transient Duality: A Multi-mechanism Approach for Modeling
Human-Object Interaction [58.67761673662716]
Humans are highly adaptable, swiftly switching between different modes to handle different tasks, situations and contexts.
In Human-object interaction (HOI) activities, these modes can be attributed to two mechanisms: (1) the large-scale consistent plan for the whole activity and (2) the small-scale children interactive actions that start and end along the timeline.
This work proposes to model two concurrent mechanisms that jointly control human motion.
arXiv Detail & Related papers (2023-07-24T12:21:33Z) - Extraction and Recovery of Spatio-Temporal Structure in Latent Dynamics
Alignment with Diffusion Models [1.4756031289693907]
In behavior-related brain computation, it is necessary to align neural signals against drastic domains among them.
We propose an alignment method ERDiff, which leverages the expressivity of the diffusion model to preserve the intrinsic-temporal structure of latent dynamics.
Our method consistently manifests its capability of preserving thetemporal structure of latent dynamics and outperforms existing approaches in alignment goodness-of-fit and neural decoding performance.
arXiv Detail & Related papers (2023-06-09T05:53:11Z) - Ranking of Communities in Multiplex Spatiotemporal Models of Brain
Dynamics [0.0]
We propose an interpretation of neural HMMs as multiplex brain state graph models we term Hidden Markov Graph Models (HMs)
This interpretation allows for dynamic brain activity to be analysed using the full repertoire of network analysis techniques.
We produce a new tool for determining important communities of brain regions using a random walk-based procedure.
arXiv Detail & Related papers (2022-03-17T12:14:09Z) - Integrating Expert ODEs into Neural ODEs: Pharmacology and Disease
Progression [71.7560927415706]
latent hybridisation model (LHM) integrates a system of expert-designed ODEs with machine-learned Neural ODEs to fully describe the dynamics of the system.
We evaluate LHM on synthetic data as well as real-world intensive care data of COVID-19 patients.
arXiv Detail & Related papers (2021-06-05T11:42:45Z) - Learning Dynamic and Personalized Comorbidity Networks from Event Data
using Deep Diffusion Processes [102.02672176520382]
Comorbid diseases co-occur and progress via complex temporal patterns that vary among individuals.
In electronic health records we can observe the different diseases a patient has, but can only infer the temporal relationship between each co-morbid condition.
We develop deep diffusion processes to model "dynamic comorbidity networks"
arXiv Detail & Related papers (2020-01-08T15:47:08Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.