TimeFlow: Temporal Conditioning for Longitudinal Brain MRI Registration and Aging Analysis

View PDF HTML (experimental)

Abstract:Longitudinal brain analysis is essential for understanding healthy aging and identifying pathological deviations. Longitudinal registration of sequential brain MRI underpins such analyses. However, existing methods are limited by reliance on densely sampled time series, a trade-off between accuracy and temporal smoothness, and an inability to prospectively forecast future brain states. To overcome these challenges, we introduce \emph{TimeFlow}, a learning-based framework for longitudinal brain MRI registration. TimeFlow uses a U-Net backbone with temporal conditioning to model neuroanatomy as a continuous function of age. Given only two scans from an individual, TimeFlow estimates accurate and temporally coherent deformation fields, enabling non-linear extrapolation to predict future brain states. This is achieved by our proposed inter-/extra-polation consistency constraints applied to both the deformation fields and deformed images. Remarkably, these constraints preserve temporal consistency and continuity without requiring explicit smoothness regularizers or densely sampled sequential data. Extensive experiments demonstrate that TimeFlow outperforms state-of-the-art methods in terms of both future timepoint forecasting and registration accuracy. Moreover, TimeFlow supports novel biological brain aging analyses by differentiating neurodegenerative trajectories from normal aging without requiring segmentation, thereby eliminating the need for labor-intensive annotations and mitigating segmentation inconsistency. TimeFlow offers an accurate, data-efficient, and annotation-free framework for longitudinal analysis of brain aging and chronic diseases, capable of forecasting brain changes beyond the observed study period.

Subjects:

Image and Video Processing (eess.IV); Computer Vision and Pattern Recognition (cs.CV)

Cite as: arXiv:2501.08667 [eess.IV]

(or arXiv:2501.08667v4 [eess.IV] for this version)

https://doi.org/10.48550/arXiv.2501.08667

arXiv-issued DOI via DataCite

Journal reference: IEEE Transactions on Medical Imaging 2026

Submission history

From: Bailiang Jian [view email] [v1] Wed, 15 Jan 2025 09:02:04 UTC (8,262 KB) (withdrawn) [v2] Mon, 7 Jul 2025 22:11:42 UTC (11,553 KB) [v3] Tue, 26 Aug 2025 14:31:39 UTC (5,735 KB) [v4] Mon, 30 Mar 2026 12:07:02 UTC (6,837 KB)