[2505.12506] Unsupervised Representation Learning -- an Invariant Risk Minimization Perspective

Computer Science > Machine Learning arXiv:2505.12506 (cs) [Submitted on 18 May 2025 (v1), last revised 2 Mar 2026 (this version, v3)] Title:Unsupervised Representation Learning -- an Invariant Risk Minimization Perspective Authors:Yotam Norman, Ron Meir View a PDF of the paper titled Unsupervised Representation Learning -- an Invariant Risk Minimization Perspective, by Yotam Norman and Ron Meir View PDF HTML (experimental) Abstract:We propose a novel unsupervised framework for \emph{Invariant Risk Minimization} (IRM), extending the concept of invariance to settings where labels are unavailable. Traditional IRM methods rely on labeled data to learn representations that are robust to distributional shifts across environments. In contrast, our approach redefines invariance through feature distribution alignment, enabling robust representation learning from unlabeled data. We introduce two methods within this framework: Principal Invariant Component Analysis (PICA), a linear method that extracts invariant directions under Gaussian assumptions, and Variational Invariant Autoencoder (VIAE), a deep generative model that separates environment-invariant and environment-dependent latent factors. Our approach is based on a novel ``unsupervised'' structural causal model and supports environment-conditioned sample-generation and intervention. Empirical evaluations on synthetic dataset, modified versions of MNIST, and CelebA demonstrate the effectiveness of our methods in capturing inva...