Organizations: Department of Data Sciences and Operations, University of Southern California · Department of Statistics and Data Science, Carnegie Mellon University · Department of Computer Science, Stanford University
We consider computationally-efficient estimation of population parameters when observations are subject to missing data. In particular, we consider estimation under the realizable contamination model of missing data in which an ε fraction of the observations are subject to an arbitrary (and unknown) missing not at random (MNAR) mechanism. When the true data is Gaussian, we provide evidence towards statistical-computational gaps in several problems. For mean estimation in ℓ2 norm, we show that in order to obtain error at most ρ, for any constant contamination ε∈(0,1), (roughly) n≳de1/ρ2 samples are necessary and that there is a computationally-inefficient algorithm which achieves this error. On the other hand, we show that any computationally-efficient method within certain popular families of algorithms requires a much larger sample complexity of (roughly) n≳d1/ρ2 and that there exists a polynomial time algorithm based on sum-of-squares which (nearly) achieves this lower bound. For covariance estimation in relative operator norm, we show that a parallel development holds. Finally, we turn to linear regression with missing observations and show that such a gap does not persist. Indeed, in this setting we show that minimizing a simple, strongly convex empirical risk nearly achieves the information-theoretic lower bound in polynomial time.
Figures & tables
Figure 1: Types of missing data patterns. Each row indicates a single sample, where the entries in gray indicate an observed value and the ⋆ entries indicate missingness. In order to simplify the results in the main text, we focus on the all-or-nothing patterns described in panel (a), deferring the extension to the more general setting to Appendix F .
Figure 2: Sample complexity phase diagram for mean estimation with ϵ a fixed constant. In order to achieve ℓ2 norm error ρ , it is information-theoretically necessary and sufficient to take n≍de1/ρ2 many samples. On the other hand, any statistical query algorithm must take (roughly) d1/ρ2 many samples and a polynomial time algorithm (nearly) saturates this lower bound.
Learning with missing data develops methods to estimate, predict, and make decisions when variables, labels, modalities, or time points are unobserved.