Enhancing Spatiotemporal Disease Progression Models via Latent Diffusion and Prior Knowledge
- URL: http://arxiv.org/abs/2405.03328v1
- Date: Mon, 6 May 2024 10:07:16 GMT
- Title: Enhancing Spatiotemporal Disease Progression Models via Latent Diffusion and Prior Knowledge
- Authors: Lemuel Puglisi, Daniel C. Alexander, Daniele Ravì,
- Abstract summary: We introduce Brain LaLAS (BrLP), atemporal disease model based on latent diffusion.
BrLP is designed to predict the evolution of diseases at the individual level on 3D brain progressions MRI.
- Score: 2.7853513988338108
- License: http://creativecommons.org/licenses/by/4.0/
- 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.
Related papers
- TADM: Temporally-Aware Diffusion Model for Neurodegenerative Progression on Brain MRI [4.414541804340033]
Temporally-Aware Diffusion Model (TADM) learns the distribution of structural changes in terms of intensity differences between scans.
During training, we propose to leverage a pre-trained Brain-Age Estimator (BAE) to refine the model's training process.
Our approach will benefit applications, such as predicting patient outcomes or improving treatments for patients.
arXiv Detail & Related papers (2024-06-18T09:00:49Z) - Predicting Parkinson's disease evolution using deep learning [1.4610685586329806]
Parkinson's disease is a neurological condition that occurs in nearly 1% of the world's population.
There is not a single blood test or biomarker available to diagnose Parkinson's disease.
No AI tools have been designed to identify the stage of progression.
arXiv Detail & Related papers (2023-12-28T10:30:54Z) - Patched Diffusion Models for Unsupervised Anomaly Detection in Brain MRI [55.78588835407174]
We propose a method that reformulates the generation task of diffusion models as a patch-based estimation of healthy brain anatomy.
We evaluate our approach on data of tumors and multiple sclerosis lesions and demonstrate a relative improvement of 25.1% compared to existing baselines.
arXiv Detail & Related papers (2023-03-07T09:40:22Z) - Temporally Adjustable Longitudinal Fluid-Attenuated Inversion Recovery
MRI Estimation / Synthesis for Multiple Sclerosis [0.0]
Multiple Sclerosis (MS) is a chronic progressive neurological disease characterized by the development of lesions in the white matter of the brain.
FLAIR brain magnetic resonance imaging (MRI) provides superior visualization and characterization of MS lesions, relative to other MRI modalities.
Longitudinal brain FLAIR MRI in MS, involving repetitively imaging a patient over time, provides helpful information for clinicians towards monitoring disease progression.
Predicting future whole brain MRI examinations with variable time lag has only been attempted in limited applications, such as healthy aging and structural degeneration in Alzheimer's Disease.
arXiv Detail & Related papers (2022-09-09T12:42:00Z) - Building Brains: Subvolume Recombination for Data Augmentation in Large
Vessel Occlusion Detection [56.67577446132946]
A large training data set is required for a standard deep learning-based model to learn this strategy from data.
We propose an augmentation method that generates artificial training samples by recombining vessel tree segmentations of the hemispheres from different patients.
In line with the augmentation scheme, we use a 3D-DenseNet fed with task-specific input, fostering a side-by-side comparison between the hemispheres.
arXiv Detail & Related papers (2022-05-05T10:31:57Z) - Federated Learning Enables Big Data for Rare Cancer Boundary Detection [98.5549882883963]
We present findings from the largest Federated ML study to-date, involving data from 71 healthcare institutions across 6 continents.
We generate an automatic tumor boundary detector for the rare disease of glioblastoma.
We demonstrate a 33% improvement over a publicly trained model to delineate the surgically targetable tumor, and 23% improvement over the tumor's entire extent.
arXiv Detail & Related papers (2022-04-22T17:27:00Z) - Unsupervised Anomaly Detection in 3D Brain MRI using Deep Learning with
Multi-Task Brain Age Prediction [53.122045119395594]
Unsupervised anomaly detection (UAD) in brain MRI with deep learning has shown promising results.
We propose deep learning for UAD in 3D brain MRI considering additional age information.
Based on our analysis, we propose a novel deep learning approach for UAD with multi-task age prediction.
arXiv Detail & Related papers (2022-01-31T09:39:52Z) - Automatic Classification of Neuromuscular Diseases in Children Using
Photoacoustic Imaging [77.32032399775152]
Neuromuscular diseases (NMDs) cause a significant burden for both healthcare systems and society.
They can lead to severe progressive muscle weakness, muscle degeneration, contracture, deformity and progressive disability.
arXiv Detail & Related papers (2022-01-27T16:37:19Z) - Classification of Brain Tumours in MR Images using Deep Spatiospatial
Models [0.0]
This paper uses twotemporal models, ResNet (2+1)D and ResNet Mixed Convolution, to classify different types of brain tumours.
It was observed that both these models performed superior to the pure 3D convolutional model, ResNet18.
arXiv Detail & Related papers (2021-05-28T19:27:51Z) - Deep Convolutional Neural Network based Classification of Alzheimer's
Disease using MRI data [8.609787905151563]
Alzheimer's disease (AD) is a progressive and incurable neurodegenerative disease which destroys brain cells and causes loss to patient's memory.
In this paper, we have proposed a smart and accurate way of diagnosing AD based on a two-dimensional deep convolutional neural network (2D-DCNN) using imbalanced three-dimensional MRI dataset.
The model classifies MRI into three categories: AD, mild cognitive impairment, and normal control: and has achieved 99.89% classification accuracy with imbalanced classes.
arXiv Detail & Related papers (2021-01-08T06:51:08Z) - Patch-based Brain Age Estimation from MR Images [64.66978138243083]
Brain age estimation from Magnetic Resonance Images (MRI) derives the difference between a subject's biological brain age and their chronological age.
Early detection of neurodegeneration manifesting as a higher brain age can potentially facilitate better medical care and planning for affected individuals.
We develop a new deep learning approach that uses 3D patches of the brain as well as convolutional neural networks (CNNs) to develop a localised brain age estimator.
arXiv Detail & Related papers (2020-08-29T11:50:37Z)
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.