2023/05/06 by Tai, Wai Ming, Aragam, Bryon · 1 citation
#FOS: Computer and information sciences #Machine Learning (cs.LG) #Machine Learning (stat.ML)
paper · doi:10.48550/arxiv.2305.04127
We study the problem of learning mixtures of Gaussians with censored data. Statistical learning with censored data is a classical problem, with numerous practical applications, however, finite-sample guarantees for even simple latent variable models such as Gaussian mixtures are missing. Formally, we are given censored data from a mixture of univariate Gaussians ∑i=1k wi N(μi,σ2), i.e. the sample is observed only if it lies inside a set S. The goal is to learn the weights wi and the means μi. We propose an algorithm that takes only \frac1εO(k) samples to estimate the weights wi and the means μi within ε error.