Multi-modal Imputation for Alzheimer's Disease Classification
Abhijith Shaji, Tamoghna Chattopadhyay, Sophia I. Thomopoulos, Greg Ver Steeg, Paul M. Thompson, Jose-Luis Ambite
TL;DR
The paper tackles the challenge of incomplete multimodal MRI data for Alzheimer's disease classification by learning a conditional diffusion model to impute missing diffusion-weighted images from T1 scans. A 3D DDPM conditioned on T1 is trained to produce anatomically consistent synthetic DWI volumes, enabling larger multimodal datasets and broader evaluation of uni- and bimodal classifiers. Across extensive experiments, imputation yields improvements in several metrics, particularly for minority classes, but gains are configuration-dependent and not universally observed; in some bimodal setups, gains are largely driven by increased T1 data rather than synthetic DWI. Overall, the work demonstrates the potential and limitations of diffusion-based cross-modality synthesis for neuroimaging classification and outlines concrete directions for expanding paired datasets and exploring newer diffusion bridging methods.
Abstract
Deep learning has been successful in predicting neurodegenerative disorders, such as Alzheimer's disease, from magnetic resonance imaging (MRI). Combining multiple imaging modalities, such as T1-weighted (T1) and diffusion-weighted imaging (DWI) scans, can increase diagnostic performance. However, complete multimodal datasets are not always available. We use a conditional denoising diffusion probabilistic model to impute missing DWI scans from T1 scans. We perform extensive experiments to evaluate whether such imputation improves the accuracy of uni-modal and bi-modal deep learning models for 3-way Alzheimer's disease classification-cognitively normal, mild cognitive impairment, and Alzheimer's disease. We observe improvements in several metrics, particularly those sensitive to minority classes, for several imputation configurations.
