mlsanityData to 14 Oct 2026

Accurate Estimation of Mutual Information in High Dimensional Data

arXiv 2506.00330 ↗First posted 31 May 2025physics.data-an cs.IT math.ITcited by 2

Not on the map. Why?

Abstract (arXiv)

Mutual information (MI) quantifies statistical dependence between variables and is widely used across scientific disciplines, yet accurate estimation from finite data remains notoriously difficult. Common approaches fail in high-dimensional, undersampled regimes (N lesssim K) typical of modern experiments, and no accepted tests exist to detect when neural network-based estimators fail, making them effectively unusable as scientific instruments. We show that neural MI estimators can be made reliable when the statistical dependencies admit a low-dimensional latent representation. Sample complexity is then governed by the latent dimensionality KZ ll K rather than the ambient dimension -- a regime shift we confirm empirically and ground theoretically via random matrix theory. Building on this insight, we develop a practical protocol that provides neural estimators with explicit statistical consistency checks, bias correction, and confidence intervals. We additionally introduce a new class of probabilistic critics (the VSIB family) that substantially reduce bias and variance at higher MI values where standard estimators break down. We validate the protocol on synthetic benchmarks (K=500, N as low as 256), on the standard 40-dataset benchmark suite of Czyz et al. (2023), on noisy MNIST (K=784), and on CIFAR-10/100 (K=3072) with a ResNet-20 backbone. Our protocol consistently matches or exceeds existing methods while being the only approach to report confidence intervals and flag unreliable estimates, achieving reliable MI detection well below the ambient pixel dimension on real images.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 25 of its 75 references that are in the map. First posted on arXiv 31 May 2025.

Builds on

25 of 75 references.

References · 25

  1. 2017Attention Is All You Needcited by 60,361
  2. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  3. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  4. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
  5. 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,838
  6. 2018MINE: Mutual Information Neural Estimationcited by 683
  7. 2019On Variational Bounds of Mutual Informationcited by 559
  8. 2016On Deep Multi-View Representation Learning: Objectives and Optimizationcited by 162
  9. 2019Understanding the Limitations of Variational Mutual Information Estimatorscited by 140
  10. 2014Efficient Estimation of Mutual Information for Strongly Dependent Variablescited by 46
  11. 2016Deep Variational Canonical Correlation Analysiscited by 44
  12. 2023Beyond Normal: On the Evaluation of Mutual Information Estimatorscited by 37
13 more
  1. 2023MINDE: Mutual Information Neural Diffusion Estimationcited by 31
  2. 2013Predictive information in a sensory populationcited by 22
  3. 2024Mutual Information Estimation via Normalizing Flowscited by 22
  4. 2024Approximating mutual information of high-dimensional variables using learned representationscited by 21
  5. 2021Tight Mutual Information Estimation With Contrastive Fenchel-Legendre Optimizationcited by 18
  6. 2013Multivariate Information Bottleneckcited by 17
  7. 2023Mutual Information Estimation via f-Divergence and Data Derangementscited by 17
  8. 2022Generative power of a protein language model trained on multiple sequence alignmentscited by 9
  9. 2019Data-Efficient Mutual Information Neural Estimatorcited by 6
  10. 2023Deep Variational Multivariate Information Bottleneck -- A Framework for Variational Lossescited by 6
  11. 2023Simultaneous Dimensionality Reduction: A Data Efficient Approach for Multimodal Representations Learningcited by 6
  12. 2017Quantifying how much sensory information in a neural code is relevant for behaviorcited by 3
  13. 2014Millisecond-scale motor encoding in a cortical vocal areacited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.