Scientific article
OA Policy
English

A robust estimator of mutual information for deep learning interpretability

Published inMachine learning: science and technology, vol. 4, no. 2, p. 1-16; 025006
Publication date2023
First online date2023-04-11
Abstract

We develop the use of mutual information (MI), a well-established metric in information theory, to interpret the inner workings of deep learning (DL) models. To accurately estimate MI from a finite number of samples, we present GMM-MI (pronounced ‘Jimmie’), an algorithm based on Gaussian mixture models that can be applied to both discrete and continuous settings. GMM-MI is computationally efficient, robust to the choice of hyperparameters and provides the uncertainty on the MI estimate due to the finite sample size. We extensively validate GMM-MI on toy data for which the ground truth MI is known, comparing its performance against established MI estimators. We then demonstrate the use of our MI estimator in the context of representation learning, working with synthetic data and physical datasets describing highly non-linear processes. We train DL models to encode high-dimensional data within a meaningful compressed (latent) representation, and use GMM-MI to quantify both the level of disentanglement between the latent variables, and their association with relevant physical quantities, thus unlocking the interpretability of the latent representation. We make GMM-MI publicly available in this GitHub repository.

Keywords
  • Deep learning
  • Mutual information
  • Interpretability
  • Representation learning
Funding
Citation (ISO format)
PIRAS, Davide et al. A robust estimator of mutual information for deep learning interpretability. In: Machine learning: science and technology, 2023, vol. 4, n° 2, p. 1–16. doi: 10.1088/2632-2153/acc444
Main files (1)
Article (Published version)
Identifiers
Journal ISSN2632-2153
74views
529downloads

Technical informations

Creation19/11/2024 08:04:38
First validation22/11/2024 11:21:55
Update22/11/2024 11:21:55
Status update22/11/2024 11:21:55
Last indexation22/11/2024 11:21:56
All rights reserved by Archive ouverte UNIGE and the University of GenevaunigeBlack