cs.LGMar 24, 2026

A One-Inclusion Graph Approach to Multi-Group Learning

Authors: Noah Bergam, Samuel Deng, Daniel Hsu

Organizations: Columbia University

Abstract

We prove the tightest-known upper bounds on the sample complexity of multi-group learning. Our algorithm extends the one-inclusion graph prediction strategy using a generalization of bipartite bb-matching. In the group-realizable setting, we provide a lower bound confirming that our algorithm's log⁡n/n\log n / n convergence rate is optimal in general. If one relaxes the learning objective such that the group on which we are evaluated is chosen obliviously of the sample, then our algorithm achieves the optimal 1/n1/n convergence rate under group-realizability.

Explore similar work

CardsList
  1. ShakyPrepend: A Multi-Group Learner with Improved Sample Complexity

    Mar 7, 2026Lujing Zhang, Daniel Hsu, Sivaraman BalakrishnanStatistical Learning Theory

  2. The price of multi-group transductive learning

    Jun 3, 2026Noah Bergam, Samuel Deng, Daniel HsuTransductive LearningStatistical Learning Theory