Abstract:Generative AI models have recently achieved mainstream attention with the advent of powerful approaches such as stable diffusion, DALL-E and MidJourney. The underlying breakthrough generative mechanism of denoising diffusion modeling can generate high quality synthetic images and can learn the underlying distribution of complex, high-dimensional data. Recent research has begun to extend these models to medical and specifically neuroimaging data. Typical neuroimaging tasks such as diagnostic classification and predictive modeling often rely on deep learning approaches based on convolutional neural networks (CNNs) and vision transformers (ViTs), with additional steps to help in interpreting the results. In our paper, we train conditional latent diffusion models (LDM) and denoising diffusion probabilistic models (DDPM) to provide insight into Alzheimer’s disease (AD) effects on the brain’s anatomy at the individual level. We first created diffusion models that could generate synthetic MRIs, by training them on real 3D T1-weighted MRI scans, and conditioning the generative process on the clinical diagnosis as a context variable. We conducted experiments to overcome limitations in training dataset size, compute time and memory resources, testing different model sizes, effects of pretraining, training duration, and latent diffusion models. We tested the sampling quality of the disease-conditioned diffusion using metrics to assess realism and diversity of the generated synthetic MRIs. We also evaluated the ability of diffusion models to conditionally sample MRI brains using a 3D CNN-based disease classifier relative to real MRIs. In our experiments, the diffusion models generated synthetic data that helped to train an AD classifier (using only 500 real training scans) -and boosted its performance by over 3% when tested on real MRI scans. Further, we used implicit classifier-free guidance to alter the conditioning of an encoded individual scan to its counterfactual (representing a healthy subject of the same age and sex) while preserving subject-specific image details. From this counterfactual image (where the same person appears healthy), a personalized disease map was generated to identify possible disease effects on the brain. Our approach efficiently generates realistic and diverse synthetic data, and may create interpretable AI-based maps for neuroscience research and clinical diagnostic applications.

BDHT: Generative AI Enables Causality Analysis for Mild Cognitive Impairment

Brain Imaging-to-Graph Generation using Adversarial Hierarchical Diffusion Models for MCI Causality Analysis

Atomic Connectomics Signatures for Characterization and Differentiation of Mild Cognitive Impairment

BrainNetDiff: Generative AI Empowers Brain Network Generation via Multimodal Diffusion Model

Functional Connectivity Networks with Latent Distributions for Mild Cognitive Impairment Identification

DiffGAN-F2S: Symmetric and Efficient Denoising Diffusion GANs for Structural Connectivity Prediction from Brain fMRI

Counterfactual MRI Generation with Denoising Diffusion Models for Interpretable Alzheimer's Disease Effect Detection

Topological Properties Analysis and Identification of Mild Cognitive Impairment Based on Individual Morphological Brain Network Connectome.

Functional Connectivity Network Estimation with an Inter-Similarity Prior for Mild Cognitive Impairment Classification

Counterfactual MRI Generation with Denoising Diffusion Models for Interpretable Alzheimer’s Disease Effect Detection

Brain Diffuser: An End-to-End Brain Image to Brain Network Pipeline

Fusing Structural and Functional Connectivities using Disentangled VAE for Detecting MCI

Multigroup Recognition of Dementia Patients with Dynamic Brain Connectivity under Multimodal Cortex Parcellation

Diffusion‐Based Causality‐Preserving Neural Network for Dementia Recognition

Adversarial Learning Based Structural Brain-network Generative Model for Analyzing Mild Cognitive Impairment

Modality-Aware Discriminative Fusion Network for Integrated Analysis of Brain Imaging Genomics

Novel Effective Connectivity Network Inference for MCI Identification

Deep Fusion of Brain Structure-Function in Mild Cognitive Impairment

Brain Network Diffusion-Driven fMRI Connectivity Augmentation for Enhanced Autism Spectrum Disorder Diagnosis

The Added Value of Diffusion-Weighted MRI-Derived Structural Connectome in Evaluating Mild Cognitive Impairment: A Multi-Cohort Validation1.

Structural Connectivity Guided Sparse Effective Connectivity for MCI Identification.