Physics-Grounded Disentangled Flow Modeling for Brain Disease Progression Trajectory

• URL: http://arxiv.org/abs/2606.28630v1
• Date: Fri, 26 Jun 2026 22:21:54 GMT
• Title: Physics-Grounded Disentangled Flow Modeling for Brain Disease Progression Trajectory
• Abstract summary: PDF is a Physics-grounded Disentangled Flow matching framework for longitudinal brain disease forecasting.<n>We explicitly decompose the longitudinal modeling of lesion growth into two processes, morphology evolution and intensity evolution.<n>Experiments on three public longitudinal datasets spanning diverse brain diseases demonstrate state-of-the-art performance.
• Score: 8.695330306289884
• License: http://creativecommons.org/licenses/by-sa/4.0/
• Abstract: Forecasting longitudinal brain lesion evolution is critical for disease monitoring and treatment planning. Existing approaches typically learn a direct mapping from a baseline image to a future observation, without explicitly modeling the physical mechanisms underlying the lesion progression. Such an entangled modeling of structural deformation and image intensity variation limits physical plausibility, model generalization, and interpretability. To address this, we propose PDF, a Physics-grounded Disentangled Flow matching framework for longitudinal brain disease forecasting. We explicitly decompose the longitudinal modeling of lesion growth into two processes, each learned by a dedicated flow matching network: morphology evolution, which captures lesion growth and structural deformation; and intensity evolution, which models signal changes driven by variations in lesion concentration. To enforce physics-grounded constraints, we introduce a PDE-regularized loss based on lesion growth dynamics, that enforces a diffusion-reaction-advection formulation for morphological evolution. Experiments on three public longitudinal datasets spanning diverse brain diseases demonstrate state-of-the-art performance, validating the effectiveness of the disentangled modeling framework and physics-grounded learning design. Code is publicly available at https://github.com/jhuldr/PDF.

Related papers

• Progression as Latent Drift: Generative Forecasting of Slow-Evolving Pathologies [25.820056271749788]
  Forecasting the future anatomy of slow-evolving neurodegenerative diseases could enable earlier, more targeted intervention and improve clinical trial design.<n>Latent Drift is a progressive generative framework that learns change in a compressed semantic representation rather than full-resolution anatomy.<n> Experiments on longitudinal 3D brain MRI show that Latent Drift improves patient-specific neuro-forecasting over diffusion and autoregressive transformer baselines.
  arXiv  Detail & Related papers  (2026-07-09T09:13:03Z)
• From Trajectories to Phenotypes: Disease Progression as Structural Priors for Multi-organ Imaging Representation Learning [22.146397986514604]
  We propose a trajectory-aware framework that transfers structural knowledge from a generative disease trajectory Transformer into an organ-wise IDP encoder.<n>Across 159 diseases in the UK Biobank cohort, trajectory-aware pretraining consistently improves both discrimination and time-to-onset prediction.<n>These results suggest that population-scale generative disease models can serve as structural priors for data-limited imaging modalities.
  arXiv  Detail & Related papers  (2026-05-12T11:10:26Z)
• Generative Modeling of Neurodegenerative Brain Anatomy with 4D Longitudinal Diffusion Model [4.65021411381981]
  Most longitudinal neuroimaging datasets are temporally sparse with a few follow-up scans per subject.<n>We propose a novel 4D (3DxT) diffusion-based generative framework that effectively models and synthesizes longitudinal anatomy over time.<n>Our method outperforms state-of-the-art baselines in generating anatomically accurate, temporally consistent, and clinically meaningful brain trajectories.
  arXiv  Detail & Related papers  (2026-04-24T16:33:48Z)
• PhyCustom: Towards Realistic Physical Customization in Text-to-Image Generation [58.02373668073258]
  We propose a fine-tuning framework comprising two novel regularization losses to activate diffusion model to perform physical customization.<n>Specifically, the proposed isometric loss aims at activating diffusion models to learn physical concepts while decouple loss helps to eliminate the mixture learning of independent concepts.
  arXiv  Detail & Related papers  (2025-12-01T16:57:02Z)
• 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)
• 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)
• A Stage-Aware Mixture of Experts Framework for Neurodegenerative Disease Progression Modelling [6.577465893404979]
  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.
  arXiv  Detail & Related papers  (2025-08-09T16:09:00Z)
• Causal Disentanglement for Robust Long-tail Medical Image Generation [80.15257897500578]
  We propose a novel medical image generation framework, which generates independent pathological and structural features.<n>We leverage a diffusion model guided by pathological findings to model pathological features, enabling the generation of diverse counterfactual images.
  arXiv  Detail & Related papers  (2025-04-20T01:54:18Z)
• Latent Drifting in Diffusion Models for Counterfactual Medical Image Synthesis [55.959002385347645]
  Latent Drifting enables diffusion models to be conditioned for medical images fitted for the complex task of counterfactual image generation.<n>We evaluate our method on three public longitudinal benchmark datasets of brain MRI and chest X-rays for counterfactual image generation.
  arXiv  Detail & Related papers  (2024-12-30T01:59:34Z)
• 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)
• Distinguishing Healthy Ageing from Dementia: a Biomechanical Simulation of Brain Atrophy using Deep Networks [5.411313268782566]
  We present a deep learning framework for hyper-elastic strain modelling of brain atrophy, during healthy ageing and in Alzheimer's Disease. The framework directly models the effects of age, disease status, and scan interval to regress regional patterns of atrophy, from which a strain-based model estimates deformations. Results show that the framework can estimate realistic deformations, following the known course of Alzheimer's disease, that clearly differentiate between healthy and demented patterns of ageing.
  arXiv  Detail & Related papers  (2021-08-18T15:58:53Z)
• 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.