Abstract:By utilizing model interpretability, our deep learning framework enhances assessment of cognitive impairment from MRI scans, paving the way for improved generalizability in machine learning models. Development of deep learning models to evaluate structural brain changes caused by cognitive impairment in MRI scans holds significant translational value. The efficacy of these models often encounters challenges due to variabilities arising from different data generation protocols, imaging equipment, radiological artifacts, and shifts in demographic distributions. Domain generalization (DG) techniques show promise in addressing these challenges by enabling the model to learn from one or more source domains and apply this knowledge to new, unseen target domains. Here we present a framework that utilizes model interpretability to enhance the generalizability of classification models across various cohorts. We used MRI scans and clinical diagnoses from four independent cohorts: Alzheimer's Disease Neuroimaging Initiative (ADNI, n = 1821), the Framingham Heart Study (FHS, n = 304), the Australian Imaging Biomarkers & Lifestyle Study of Ageing (AIBL, n = 661), and the National Alzheimer's Coordinating Center (NACC, n = 4647). With this data, we trained a deep neural network to focus on areas of the brain identified as relevant to the disease for model training. Our approach involved training a classifier to differentiate between structural neurodegeneration in individuals with normal cognition (NC), mild cognitive impairment (MCI), and dementia due to Alzheimer's disease (AD). This was achieved by aligning class‐wise attention with a unified visual saliency prior, which was computed offline for each class using all the training data. Our method not only competes with state‐of‐the‐art approaches but also shows improved correlation with postmortem histology. This alignment with the gold standard evidence is a significant step towards validating the effectiveness of DG frameworks, paving the way for their broader application in the field.

Structure Guided Manifolds for Discovery of Disease Characteristics

Morphological Feature Visualization of Alzheimer's Disease via Multidirectional Perception GAN

A Novel Framework to Synthesize Arterial Spin Labeling Images Using Difference Images

A novel locally-constrained GAN-based ensemble to synthesize arterial spin labeling images

BG-GAN: Generative AI Enables Representing Brain Structure-Function Connections for Alzheimer's Disease

LDGAN: Longitudinal-Diagnostic Generative Adversarial Network for Disease Progression Prediction with Missing Structural MRI

A Deep Learning Framework Identifies Dimensional Representations of Alzheimer's Disease from Brain Structure.

A Novel Alzheimer’s Disease Detection Approach Using GAN-based Brain Slice Image Enhancement

SA-GAN: Structure-Aware GAN for Organ-Preserving Synthetic CT Generation

Smile-GANs: Semi-supervised clustering via GANs for dissecting brain disease heterogeneity from medical images

Disease-Image-Specific Learning for Diagnosis-Oriented Neuroimage Synthesis with Incomplete Multi-Modality Data

Alzheimer's Disease Prediction via Brain Structural-Functional Deep Fusing Network

Disentangling brain heterogeneity via semi-supervised deep-learning and MRI: dimensional representations of Alzheimer's Disease

Interpretation of Disease Evidence for Medical Images Using Adversarial Deformation Fields

Disease‐driven domain generalization for neuroimaging‐based assessment of Alzheimer's disease

Hypergraph Structural Information Aggregation Generative Adversarial Networks for Diagnosis and Pathogenetic Factors Identification of Alzheimer’s Disease With Imaging Genetic Data

Tetrahedral spectral feature-Based bayesian manifold learning for grey matter morphometry: Findings from the Alzheimer's disease neuroimaging initiative.

Multi-site MRI Harmonization Via Attention-Guided Deep Domain Adaptation for Brain Disorder Identification.

Predicting disease-related MRI patterns of multiple sclerosis through GAN-based image editing

Tetrahedral spectral feature-Based bayesian manifold learning for grey matter morphometry: Findings from the Alzheimer’s disease neuroimaging initiative

Collaborative Image Synthesis and Disease Diagnosis for Classification of Neurodegenerative Disorders with Incomplete Multi-modal Neuroimages