Learning discrete distributions with infinite support
Doron Cohen, Aryeh Kontorovich, Geoffrey Wolfer
Abstract
We present a novel approach to estimating discrete distributions with (potentially) infinite support in the total variation metric. In a departure from the established paradigm, we make no structural assumptions whatsoever on the sampling distribution. In such a setting, distribution-free risk bounds are impossible, and the best one could hope for is a fully empirical data-dependent bound. We derive precisely such bounds, and demonstrate that these are, in a well-defined sense, the best possible. Our main discovery is that the half-norm of the empirical distribution provides tight upper and lower estimates on the empirical risk. Furthermore, this quantity decays at a nearly optimal rate as a function of the true distribution. The optimality follows from a minimax result, of possible independent interest. Additional structural results are provided, including an exact Rademacher complexity calculation and apparently a first connection between the total variation risk and the missing mass.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 245e5c52-862a-4684-994b-e621cd7c7c00Cited by top-tier papers3
- Dimension-free empirical entropy estimationDoron Cohen, Aryeh Kontorovich, Aaron Koolyk, Geoffrey WolferNeurIPS 2021 · 6 citations
- Nonparametric Density Estimation under Distribution DriftAlessio Mazzetto, Eli UpfalICML 2023 · 5 citations
- TURF: Two-Factor, Universal, Robust, Fast Distribution Learning AlgorithmYi Hao, Ayush Jain, Alon Orlitsky, Vaishakh RavindrakumarICML 2022
Related papers
- Improved Distribution Estimation in Doron Cohen, Aryeh Kontorovich, Yonatan LivshitzICML 2026
- Relative Deviation Margin BoundsCorinna Cortes, Mehryar Mohri, Ananda Theertha SureshICML 2021 · 16 citations
- How Much is Unseen Depends Chiefly on Information About the SeenSeongmin Lee, Marcel BöhmeICLR 2025
- Beyond the Best: Distribution Functional Estimation in Infinite-Armed BanditsYifei Wang, Tavor Z. Baharav, Yanjun Han, Jiantao Jiao et al.NeurIPS 2022 · 2 citations
- Instance-Optimality for Private KL Distribution EstimationJiayuan Ye, Vitaly Feldman, Kunal TalwarNeurIPS 2025
