UCL Discovery
UCL home » Library Services » Electronic resources » UCL Discovery

ICAM: Interpretable Classification via Disentangled Representations and Feature Attribution Mapping

Bass, C; Silva, MD; Sudre, CH; Tudosiu, P-D; Smith, SM; Robinson, EC; (2020) ICAM: Interpretable Classification via Disentangled Representations and Feature Attribution Mapping. In: Larochelle, H and Ranzato, M and Hadsell, R and Balcan, M.F. and Lin, H, (eds.) Advances in Neural Information Processing Systems 33 (NeurIPS 2020). NeurIPS: Virtual conference (Vancouver, Canada.). Green open access

[thumbnail of 2006.08287.pdf]
Preview
Text
2006.08287.pdf - Accepted Version

Download (8MB) | Preview

Abstract

Feature attribution (FA), or the assignment of class-relevance to different locations in an image, is important for many classification problems but is particularly crucial within the neuroscience domain, where accurate mechanistic models of behaviours, or disease, require knowledge of all features discriminative of a trait. At the same time, predicting class relevance from brain images is challenging as phenotypes are typically heterogeneous, and changes occur against a background of significant natural variation. Here, we present a novel framework for creating class specific FA maps through image-to-image translation. We propose the use of a VAE-GAN to explicitly disentangle class relevance from background features for improved interpretability properties, which results in meaningful FA maps. We validate our method on 2D and 3D brain image datasets of dementia (ADNI dataset), ageing (UK Biobank), and (simulated) lesion detection. We show that FA maps generated by our method outperform baseline FA methods when validated against ground truth. More significantly, our approach is the first to use latent space sampling to support exploration of phenotype variation.

Type: Proceedings paper
Title: ICAM: Interpretable Classification via Disentangled Representations and Feature Attribution Mapping
Event: Advances in Neural Information Processing Systems 33 (NeurIPS 2020)
Open access status: An open access version is available from UCL Discovery
Publisher version: https://proceedings.neurips.cc/paper/2020/hash/56f...
Language: English
Additional information: This version is the author accepted manuscript. For information on re-use, please refer to the publisher’s terms and conditions.
UCL classification: UCL
UCL > Provost and Vice Provost Offices > School of Life and Medical Sciences
UCL > Provost and Vice Provost Offices > School of Life and Medical Sciences > Faculty of Population Health Sciences > Institute of Cardiovascular Science
UCL > Provost and Vice Provost Offices > School of Life and Medical Sciences > Faculty of Population Health Sciences > Institute of Cardiovascular Science > Population Science and Experimental Medicine
UCL > Provost and Vice Provost Offices > School of Life and Medical Sciences > Faculty of Population Health Sciences > Institute of Cardiovascular Science > Population Science and Experimental Medicine > MRC Unit for Lifelong Hlth and Ageing
URI: https://discovery.ucl.ac.uk/id/eprint/10108466
Downloads since deposit
7Downloads
Download activity - last month
Download activity - last 12 months
Downloads by country - last 12 months

Archive Staff Only

View Item View Item