On sample complexity for covariance estimation via the unadjusted Langevin algorithm
Shogo Nakakita
TL;DR
This work analyzes covariance estimation for strongly log-concave, smooth distributions using the unadjusted Langevin algorithm (ULA). It introduces a concentration bound for the dependent sample covariance via a log-Sobolev inequality on the joint ULA iterates, yielding a variance rate of order $O\left(\frac{1}{\alpha}\sqrt{\frac{d}{\alpha \eta n}}\right)$ and an explicit bias–variance decomposition. By combining this with existing discretization bias bounds, the authors derive non-asymptotic sample complexities for estimating Cov(π) with ULA, and compare against embarrassingly parallel ULA, showing a logarithmic advantage for the single-chain approach due to burn-in bias reduction. A key technical contribution is establishing a joint log-Sobolev inequality with constant $2/(\alpha^2 \eta)$ for the sequence $(X_{m+1},\ldots,X_{m+n})$, which underpins the concentration results. The findings quantify how burn-in and dependence affect sample complexity and provide practical guidance on the trade-offs between single-chain and parallel ULA in high-dimensional covariance estimation.
Abstract
We establish sample complexity guarantees for estimating the covariance matrix of strongly log-concave smooth distributions using the unadjusted Langevin algorithm (ULA). We quantitatively compare our complexity estimates on single-chain ULA with embarrassingly parallel ULA and derive that the sample complexity of the single-chain approach is smaller than that of embarrassingly parallel ULA by a logarithmic factor in the dimension and the reciprocal of the prescribed precision, with the difference arising from effective bias reduction through burn-in. The key technical contribution is a concentration bound for the sample covariance matrix around its expectation, derived via a log-Sobolev inequality for the joint distribution of ULA iterates.
