| Title: | ICAM-Reg: Interpretable Classification and Regression With Feature Attribution for Mapping Neurological Phenotypes in Individual Scans |
| Journal: | IEEE Transactions on Medical Imaging |
| Published: | 3 Apr 2023 |
| Pubmed: | https://pubmed.ncbi.nlm.nih.gov/36374873/ |
| DOI: | https://doi.org/10.1109/tmi.2022.3221890 |
| URL: | https://ieeexplore.ieee.org/ielx7/42/10091712/09947339.pdf |
Publication 7882
WARNING: the interactive features of this website use CSS3, which your browser does not support. To use the full features of this website, please update your browser.
Abstract
An important goal of medical imaging is to be able to precisely detect patterns of disease specific to individual scans; however, this is challenged in brain imaging by the degree of heterogeneity of shape and appearance. Traditional methods, based on image registration, historically fail to detect variable features of disease, as they utilise population-based analyses, suited primarily to studying group-average effects. In this paper we therefore take advantage of recent developments in generative deep learning to develop a method for simultaneous classification, or regression, and feature attribution (FA). Specifically, we explore the use of a VAE-GAN (variational autoencoder - general adversarial network) for translation called ICAM, to explicitly disentangle class relevant features, from background confounds, for improved interpretability and regression of neurological phenotypes. We validate our method on the tasks of Mini-Mental State Examination (MMSE) cognitive test score prediction for the Alzheimer's Disease Neuroimaging Initiative (ADNI) cohort, as well as brain age prediction, for both neurodevelopment and neurodegeneration, using the developing Human Connectome Project (dHCP) and UK Biobank datasets. We show that the generated FA maps can be used to explain outlier predictions and demonstrate that the inclusion of a regression module improves the disentanglement of the latent space. Our code is freely available on GitHub https://github.com/CherBass/ICAM.12 Authors
- Cher Bass
- Mariana da Silva
- Carole Sudre
- Logan Z. J. Williams
- Helena S. Sousa
- Petru-Daniel Tudosiu
- Fidel Alfaro-Almagro
- Sean P. Fitzgibbon
- Matthew F. Glasser
- Stephen M. Smith
- Emma C. Robinson
- Alzheimer's Disease Neuroimaging Initiative
1 Application
| Application ID | Title |
|---|---|
| 8107 | Analysis of Biobank Neuro Imaging Data |
Enabling scientific discoveries that improve human health