mlsanityData to 7 Oct 2026

Bayes-Sufficient Representations in Supervised Learning

arXiv 2606.04045 ↗First posted 2 Jun 2026cs.LG cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

Representation learning is often described as preserving the information in an input that is relevant for prediction. This work asks what relevance means for a fixed supervised decision problem. A representation is defined to be Bayes-sufficient for a joint distribution and loss if some prediction head can use it to implement a Bayes-optimal action rule. This makes the target information loss-dependent. In the almost-surely unique Bayes-action case, the relevant object is a Bayes quotient, which identifies inputs that require the same Bayes-optimal action. A representation is sufficient when it refines this quotient, and Bayes-minimal when it is informationally equivalent to it. The framework connects naturally to property elicitation: zero-one loss requires the Bayes class, squared loss the conditional mean, Brier loss the conditional probability in binary prediction, and log loss or strictly proper scoring rules the predictive distribution. Controlled finite experiments, learned neural bottleneck experiments, and a real-data iNaturalist taxonomic refinement experiment illustrate the distinction between sufficiency, minimality, and retained non-required information. For a fixed supervised problem, the distribution and the loss determine the Bayes action, the Bayes action determines the quotient, and the quotient determines the minimal information required for Bayes-optimal prediction.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 7 of its 25 references that are in the map. First posted on arXiv 2 Jun 2026.

Builds on

7 of 25 references.

References · 7

  1. 2016Understanding intermediate layers using linear classifier probescited by 1,279
  2. 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
  3. 2015Deep Learning and the Information Bottleneck Principlecited by 793
  4. 2019Designing and Interpreting Probes with Control Taskscited by 608
  5. 2017Emergence of Invariance and Disentanglement in Deep Representationscited by 213
  6. 2016Information Dropout: Learning Optimal Representations Through Noisy Computationcited by 131
  7. 2026Gauge-invariant representation holonomycited by 7

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.