Learning Patient-Specific Disease Dynamics with Latent Flow Matching for Longitudinal Imaging Generation

Summary

This paper introduces $Δ$-LFM, a novel framework designed to model patient-specific disease progression by treating disease dynamics as a continuous velocity field. It leverages Flow Matching (FM) and a novel patient-specific latent alignment strategy to create a semantically meaningful latent space where trajectories monotonically correlate with disease severity. The method demonstrates strong empirical performance on longitudinal MRI benchmarks and offers enhanced interpretability and visualization of disease dynamics.

Medical Relevance

This research is highly relevant to medicine as it offers a sophisticated computational tool to accurately model and visualize the continuous progression of diseases from longitudinal data, paving the way for more precise early diagnoses and the development of tailored, personalized treatment plans.

AI Health Application

The AI application is to develop models that can learn, interpret, and visualize patient-specific disease progression over time using longitudinal medical imaging data. This enables more precise prediction of disease trajectory, aids in early diagnosis, and informs personalized treatment strategies by correlating latent disease dynamics with clinical severity indicators.

Key Points

• **Problem Addressed:** Prior generative models for disease progression suffer from scattered latent representations and disruption of continuity (e.g., diffusion models), failing to capture the inherently continuous and monotonic nature of disease. Auto-Encoders (AEs) also lack guarantees for cross-patient latent alignment or correlation with clinical severity.
• **Core Methodology - Flow Matching:** The disease dynamic is conceptualized as a continuous velocity field, and Flow Matching (FM) is employed to align the temporal evolution of patient data, allowing for the capture of intrinsic disease dynamics.
• **Novel Latent Alignment Strategy:** A key innovation is learning patient-specific latent alignment. This enforces patient trajectories within the latent space to lie along a consistent, specific axis, with magnitude along this axis increasing monotonically with disease severity indicators (e.g., age, disease conditions).
• **Semantically Meaningful Latent Space:** This alignment strategy constructs a consistent and semantically meaningful latent space, which is crucial for interpretable disease progression modeling.
• **$Δ$-LFM Framework:** The proposed framework, $Δ$-LFM, integrates Flow Matching with this patient-specific latent alignment to model latent progression.
• **Empirical Validation:** $Δ$-LFM demonstrated strong empirical performance across three distinct longitudinal MRI benchmarks.
• **Enhanced Interpretability:** Beyond quantitative performance, the framework provides a new and improved means for interpreting and visualizing complex disease dynamics due to its structured and semantically rich latent representation.

Clinical Impact

The clinical impact of $Δ$-LFM is substantial, enabling more accurate early diagnosis by identifying subtle progression patterns. It supports the development of personalized treatment strategies by providing insights into individual patient trajectories and predicted disease courses. The framework can enhance disease monitoring, allow for better prognosis prediction, and potentially aid in the discovery of novel imaging-based biomarkers. Moreover, it could optimize clinical trial design by facilitating more precise patient stratification based on expected progression.

Limitations

The abstract does not explicitly state limitations of the proposed $Δ$-LFM framework. It primarily highlights challenges in prior methods (scattered latent representations, lack of semantic structure, disruption of continuity by diffusion models, and AE's inability to guarantee alignment with clinical severity) that $Δ$-LFM aims to overcome.

Future Directions

The abstract does not explicitly mention future research directions.

Medical Domains

Keywords

Abstract

Understanding disease progression is a central clinical challenge with direct implications for early diagnosis and personalized treatment. While recent generative approaches have attempted to model progression, key mismatches remain: disease dynamics are inherently continuous and monotonic, yet latent representations are often scattered, lacking semantic structure, and diffusion-based models disrupt continuity with random denoising process. In this work, we propose to treat the disease dynamic as a velocity field and leverage Flow Matching (FM) to align the temporal evolution of patient data. Unlike prior methods, it captures the intrinsic dynamic of disease, making the progression more interpretable. However, a key challenge remains: in latent space, Auto-Encoders (AEs) do not guarantee alignment across patients or correlation with clinical-severity indicators (e.g., age and disease conditions). To address this, we propose to learn patient-specific latent alignment, which enforces patient trajectories to lie along a specific axis, with magnitude increasing monotonically with disease severity. This leads to a consistent and semantically meaningful latent space. Together, we present $Δ$-LFM, a framework for modeling patient-specific latent progression with flow matching. Across three longitudinal MRI benchmarks, $Δ$-LFM demonstrates strong empirical performance and, more importantly, offers a new framework for interpreting and visualizing disease dynamics.

Comments

Under review