The Optimal Approximation Factor in Density Estimation
This work addresses a fundamental problem in density estimation for machine learning and statistics, providing optimal approximation bounds that clarify the advantage of improper learning, though it is incremental as it builds on prior results like Yatracos' theorem.
The paper tackles the problem of approximating an unknown target density from a finite class of densities, showing that the optimal approximation factor is 2 when allowing improper learning (outputting arbitrary densities), which improves upon the previously known factor of 3 for proper learning (outputting from the class). It proves that factor 3 is optimal for proper learning and achieves factor 2 with sample complexity bounds using adaptive data analysis techniques.
Consider the following problem: given two arbitrary densities $q_1,q_2$ and a sample-access to an unknown target density $p$, find which of the $q_i$'s is closer to $p$ in total variation. A remarkable result due to Yatracos shows that this problem is tractable in the following sense: there exists an algorithm that uses $O(ε^{-2})$ samples from $p$ and outputs~$q_i$ such that with high probability, $TV(q_i,p) \leq 3\cdot\mathsf{opt} + ε$, where $\mathsf{opt}= \min\{TV(q_1,p),TV(q_2,p)\}$. Moreover, this result extends to any finite class of densities $\mathcal{Q}$: there exists an algorithm that outputs the best density in $\mathcal{Q}$ up to a multiplicative approximation factor of 3. We complement and extend this result by showing that: (i) the factor 3 can not be improved if one restricts the algorithm to output a density from $\mathcal{Q}$, and (ii) if one allows the algorithm to output arbitrary densities (e.g.\ a mixture of densities from $\mathcal{Q}$), then the approximation factor can be reduced to 2, which is optimal. In particular this demonstrates an advantage of improper learning over proper in this setup. We develop two approaches to achieve the optimal approximation factor of 2: an adaptive one and a static one. Both approaches are based on a geometric point of view of the problem and rely on estimating surrogate metrics to the total variation. Our sample complexity bounds exploit techniques from {\it Adaptive Data Analysis}.