Scalable Infomin Learning
Not on the map. Why?
Abstract (arXiv)
The task of infomin learning aims to learn a representation with high utility while being uninformative about a specified target, with the latter achieved by minimising the mutual information between the representation and the target. It has broad applications, ranging from training fair prediction models against protected attributes, to unsupervised learning with disentangled representations. Recent works on infomin learning mainly use adversarial training, which involves training a neural network to estimate mutual information or its proxy and thus is slow and difficult to optimise. Drawing on recent advances in slicing techniques, we propose a new infomin learning approach, which uses a novel proxy metric to mutual information. We further derive an accurate and analytically computable approximation to this proxy metric, thereby removing the need of constructing neural network-based mutual information estimators. Experiments on algorithmic fairness, disentangled representation learning and domain adaptation verify that our method can effectively remove unwanted information with limited time budget.
Key facts. Cited by 8 papers in the map (data through 14 Oct 2026). Builds on 48 of its 72 references that are in the map. First posted on arXiv 21 Feb 2023.
Builds on
48 of 72 references.
References · 48
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2015Domain-Adversarial Training of Neural Networkscited by 3,044
- 2016Improved Techniques for Training GANscited by 2,970
- 2015Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networkscited by 2,054
- 2018Learning deep representations by mutual information estimation and maximizationcited by 1,273
- 2018Challenging Common Assumptions in the Unsupervised Learning of Disentangled Representationscited by 941
- 2016Deep Variational Information Bottleneckcited by 930
- 2015Deep Learning and the Information Bottleneck Principlecited by 802
- 2018MINE: Mutual Information Neural Estimationcited by 683
- 2018Disentangling by Factorisingcited by 616
36 more
- 2019On Variational Bounds of Mutual Informationcited by 559
- 2019Sliced Score Matching: A Scalable Approach to Density and Score Estimationcited by 384
- 2018Learning Adversarially Fair and Transferable Representationscited by 353
- 2018Understanding disentangling in β-VAEcited by 335
- 2018Towards a Definition of Disentangled Representationscited by 296
- 2017Variational Inference of Disentangled Latent Concepts from Unlabeled Observationscited by 241
- 2020CLUB: A Contrastive Log-ratio Upper Bound of Mutual Informationcited by 235
- 2018A DIRT-T Approach to Unsupervised Domain Adaptationcited by 215
- 2019Flexibly Fair Representation Learning by Disentanglementcited by 196
- 2015Censoring Representations with an Adversarycited by 192
- 2018Adversarial Removal of Demographic Attributes from Text Datacited by 170
- 2019Understanding the Limitations of Variational Mutual Information Estimatorscited by 140
- 2019Wasserstein Fair Classificationcited by 131
- 2017Controllable Invariance through Adversarial Feature Learningcited by 127
- 2018Invariant Representations without Adversarial Trainingcited by 112
- 2018Generative Modeling using the Sliced Wasserstein Distancecited by 108
- 2020Fair Regression with Wasserstein Barycenterscited by 100
- 2020InfoBERT: Improving Robustness of Language Models from An Information Theoretic Perspectivecited by 92
- 2020Statistical and Topological Properties of Sliced Probability Divergencescited by 85
- 2018Learning Controllable Fair Representationscited by 83
- 2019Overlearning Reveals Sensitive Attributescited by 76
- 2020Neural Approximate Sufficient Statistics for Implicit Modelscited by 74
- 2019Mitigating Information Leakage in Image Representations: A Maximum Entropy Approachcited by 48
- 2021Fair Normalizing Flowscited by 38
- 2019Orthogonal Estimation of Wasserstein Distancescited by 32
- 2019Learning Fair Representations via an Adversarial Frameworkcited by 29
- 2021Commutative Lie Group VAE for Disentanglement Learningcited by 27
- 2013The Randomized Dependence Coefficientcited by 26
- 2018CR-GAN: Learning Complete Representations for Multi-view Generationcited by 23
- 2020Sliced Kernelized Stein Discrepancycited by 23
- 2014Discovering Hidden Factors of Variation in Deep Networkscited by 21
- 2013Partial Distance Correlation with Methods for Dissimilaritiescited by 20
- 2019Invariant Representations through Adversarial Forgettingcited by 20
- 2019Fairness-Aware Neural R'eyni Minimization for Continuous Featurescited by 15
- 2022Controlling Directions Orthogonal to a Classifiercited by 13
- 2022Disentangled Representation Learning for RF Fingerprint Extraction under Unknown Channel Statisticscited by 5
Built on by
Citing papers per year: 2023: 2; 2024: 4; 2025: 1; 2026: 1.
8 papers.
- 2024Approximating mutual information of high-dimensional variables using learned representationscited by 21
- 2023Max-Sliced Mutual Informationcited by 10
- 2025Neural Mutual Information Estimation with Vector Copulascited by 4
- 2023Towards Poisoning Fair Representationscited by 3
- 2026InfoAtlas: A Foundation Model for Zero-Shot Statistical Dependence Estimatecited by 1
- 2024From Discrete to Continuous: Deep Fair Clustering With Transferable Representationscited by 0
- 2024Generalizing Orthogonalization for Models with Non-Linearitiescited by 0
- 2024Mutual Information Multinomial Estimationcited by 0
- 2026InfoAtlas: A Foundation Model for Zero-Shot Statistical Dependence Estimate29 May 2026
- 2025Neural Mutual Information Estimation with Vector Copulas23 Oct 2025
- 2024Approximating mutual information of high-dimensional variables using learned representations3 Sep 2024
- 2024Mutual Information Multinomial Estimation18 Aug 2024
- 2024Generalizing Orthogonalization for Models with Non-Linearities3 May 2024
- 2024From Discrete to Continuous: Deep Fair Clustering With Transferable Representations24 Mar 2024
- 2023Towards Poisoning Fair Representations28 Sep 2023
- 2023Max-Sliced Mutual Information28 Sep 2023