Puglisi, L;
Alexander, DC;
Ravì, D;
(2024)
Enhancing Spatiotemporal Disease Progression Models via Latent Diffusion and Prior Knowledge.
In:
Medical Image Computing and Computer Assisted Intervention – MICCAI 2024.
(pp. pp. 173-183).
Springer Nature
Text
0511_paper.pdf - Accepted Version Access restricted to UCL open access staff until 5 October 2025. Download (1MB) |
Abstract
In this work, we introduce Brain Latent Progression (BrLP), a novel spatiotemporal disease progression model based on latent diffusion. BrLP is designed to predict the evolution of diseases at the individual level on 3D brain MRIs. Existing deep generative models developed for this task are primarily data-driven and face challenges in learning disease progressions. BrLP addresses these challenges by incorporating prior knowledge from disease models to enhance the accuracy of predictions. To implement this, we propose to integrate an auxiliary model that infers volumetric changes in various brain regions. Additionally, we introduce Latent Average Stabilization (LAS), a novel technique to improve spatiotemporal consistency of the predicted progression. BrLP is trained and evaluated on a large dataset comprising 11,730 T1-weighted brain MRIs from 2,805 subjects, collected from three publicly available, longitudinal Alzheimer’s Disease (AD) studies. In our experiments, we compare the MRI scans generated by BrLP with the actual follow-up MRIs available from the subjects, in both cross-sectional and longitudinal settings. BrLP demonstrates significant improvements over existing methods, with an increase of 22% in volumetric accuracy across AD-related brain regions and 43% in image similarity to the ground-truth scans. The ability of BrLP to generate conditioned 3D scans at the subject level, along with the novelty of integrating prior knowledge to enhance accuracy, represents a significant advancement in disease progression modeling, opening new avenues for precision medicine. The code of BrLP is available at the following link: https://github.com/LemuelPuglisi/BrLP.
Type: | Proceedings paper |
---|---|
Title: | Enhancing Spatiotemporal Disease Progression Models via Latent Diffusion and Prior Knowledge |
Event: | 27th International Conference - MICCAI 2024 |
ISBN-13: | 978-3-031-72068-0 |
DOI: | 10.1007/978-3-031-72069-7_17 |
Publisher version: | http://dx.doi.org/10.1007/978-3-031-72069-7_17 |
Language: | English |
Additional information: | This version is the author accepted manuscript. For information on re-use, please refer to the publisher's terms and conditions. |
UCL classification: | UCL UCL > Provost and Vice Provost Offices > UCL BEAMS UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science > Dept of Computer Science |
URI: | https://discovery.ucl.ac.uk/id/eprint/10198880 |
Archive Staff Only
View Item |