Learning Exponential Families from Truncated Samples
Jane H. Lee, Andre Wibisono, Emmanouil Zampetakis
Abstract
Missing data problems have many manifestations across many scientific fields. A fundamental type of missing data problem arises when samples are truncated, i.e., samples that lie in a subset of the support are not observed. Statistical estimation from truncated samples is a classical problem in statistics which dates back to Galton, Pearson, and Fisher. A recent line of work provides the first efficient estimation algorithms for the parameters of a Gaussian distribution [10] and for linear regression with Gaussian noise [11, 14, 37] . In this paper we generalize these results to log-concave exponential families. We provide an estimation algorithm that shows that extrapolation is possible for a much larger class of distributions while it maintains a polynomial sample and time complexity on average. Our algorithm is based on Projected Stochastic Gradient Descent and is not only applicable in a more general setting but is also simpler than the recent algorithms of [10, 26, 11, 14, 37] . Our work also has interesting implications for learning general log-concave distributions and sampling given only access to truncated data.
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 papers5
- Smoothed Analysis of Learning from Positive SamplesJane H. Lee, Anay Mehrotra, Manolis ZampetakisSTOC 2026 · 2 citations
- Linear Regression with Unknown Truncation Beyond Gaussian FeaturesAlexandros Kouridakis, Anay Mehrotra, Alkis Kalavasis, Constantine CaramanisICML 2026 · 2 citations
- Private Statistical Estimation via TruncationManolis Zampetakis, Felix ZhouNeurIPS 2025 · 1 citation
- Efficient Statistics With Unknown Truncation, Polynomial Time Algorithms, Beyond GaussiansJane H. Lee, Anay Mehrotra, Manolis ZampetakisFOCS 2024 · 1 citation
- Oracle efficient truncated statisticsKonstantinos Karatapanis, Vasilis Kontonis, Christos TzamosICLR 2025
Builds on5
- Truncated Linear Regression in High DimensionsConstantinos Daskalakis, Dhruv Rohatgi, Emmanouil ZampetakisNeurIPS 2020 · 19 citations
- Efficient Truncated Linear Regression with Unknown Noise VarianceConstantinos Daskalakis, Patroklos Stefanou, Rui Yao, Emmanouil ZampetakisNeurIPS 2021 · 15 citations
- A Computationally Efficient Method for Learning Exponential Family DistributionsAbhin Shah, Devavrat Shah, Gregory W. WornellNeurIPS 2021 · 15 citations
- Sampling from Log-Concave Distributions with Infinity-Distance GuaranteesOren Mangoubi, Nisheeth K. VishnoiNeurIPS 2022 · 15 citations
- What Makes a Good Fisherman? Linear Regression under Self-Selection BiasYeshwanth Cherapanamjeri, Constantinos Daskalakis, Andrew Ilyas, Manolis ZampetakisSTOC 2023 · 4 citations
Related papers
- Algorithms for heavy-tailed statistics: regression, covariance estimation, and beyondYeshwanth Cherapanamjeri, Samuel B. Hopkins, Tarun Kathuria, Prasad Raghavendra et al.STOC 2020 · 2 citations
- Sampling and Integration of Logconcave Functions by Algorithmic DiffusionYunbum Kook, Santosh S. VempalaSTOC 2025 · 2 citations
- List-Decodable Sparse Mean EstimationShiwei Zeng, Jie ShenNeurIPS 2022 · 13 citations
- Mean Estimation from Coarse Data: Characterizations and Efficient AlgorithmsAlkis Kalavasis, Anay Mehrotra, Manolis Zampetakis, Felix Zhou et al.ICLR 2026
- Learning Mixtures of Gaussians with Censored DataWai Ming Tai, Bryon AragamICML 2023 · 1 citation
