Derandomizing Multi-Distribution Learning
Kasper Green Larsen, Omar Montasser, Nikita Zhivotovskiy
Abstract
Multi-distribution or collaborative learning involves learning a single predictor that works well across multiple data distributions, using samples from each during training. Recent research on multi-distribution learning, focusing on binary loss and finite VC dimension classes, has shown near-optimal sample complexity that is achieved with oracle efficient algorithms. That is, these algorithms are computationally efficient given an efficient ERM for the class. Unlike in classical PAC learning, where the optimal sample complexity is achieved with deterministic predictors, current multi-distribution learning algorithms output randomized predictors. This raises the question: can these algorithms be derandomized to produce a deterministic predictor for multiple distributions? Through a reduction to discrepancy minimization, we show that derandomizing multi-distribution learning is computationally hard, even when ERM is computationally efficient. On the positive side, we identify a structural condition enabling an efficient black-box reduction, converting existing randomized multi-distribution predictors into deterministic ones.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers2
- Mean-Field Sampling for Cooperative Multi-Agent Reinforcement LearningEmile Anand, Ishani Karmarkar, Guannan QuNeurIPS 2025 · 10 citations
- How Many Domains Suffice for Domain Generalization? A Tight Characterization via the Domain Shattering DimensionCynthia Dwork, Lunjia Hu, Han ShaoNeurIPS 2025 · 3 citations
Builds on5
- Distributionally Robust Neural NetworksShiori Sagawa, Pang Wei Koh, Tatsunori B. Hashimoto, Percy LiangICLR 2020 · 1,578 citations
- On-Demand Sampling: Learning Optimally from Multiple DistributionsNika Haghtalab, Michael I. Jordan, Eric ZhaoNeurIPS 2022 · 57 citations
- Multi-group Agnostic PAC LearnabilityGuy N. Rothblum, Gal YonaICML 2021 · 48 citations
- Adaptive Sampling for Minimax Fair ClassificationShubhanshu Shekhar, Greg Fields, Mohammad Ghavamzadeh, Tara JavidiNeurIPS 2021 · 46 citations
- Simple and near-optimal algorithms for hidden stratification and multi-group learningChristopher J. Tosh, Daniel HsuICML 2022 · 28 citations
Related papers
- Collaborative Learning with Different Labeling FunctionsYuyang Deng, Mingda QiaoICML 2024 · 2 citations
- Fast Rates for Bandit PAC Multiclass ClassificationLiad Erez, Alon Peled-Cohen, Tomer Koren, Yishay Mansour et al.NeurIPS 2024 · 7 citations
- Agnostic Multi-Group Active LearningNicholas Rittler, Kamalika ChaudhuriNeurIPS 2023 · 7 citations
- Transformation-Invariant Learning and Theoretical Guarantees for OOD GeneralizationOmar Montasser, Han Shao, Emmanuel AbbeNeurIPS 2024 · 7 citations
- Revisiting Agnostic PAC LearningSteve Hanneke, Kasper Green Larsen, Nikita ZhivotovskiyFOCS 2024 · 1 citation
