Randomized Learning of the Second-Moment Matrix of a Smooth Function

Abstract

Consider an open set $\mathbb{D}\subseteq\mathbb{R}^n$, equipped with a probability measure $μ$. An important characteristic of a smooth function $f:\mathbb{D}\rightarrow\mathbb{R}$ is its \emph{second-moment matrix} $Σ_μ:=\int \nabla f(x) \nabla f(x)^* μ(dx) \in\mathbb{R}^{n\times n}$, where $\nabla f(x)\in\mathbb{R}^n$ is the gradient of $f(\cdot)$ at $x\in\mathbb{D}$ and $*$ stands for transpose. For instance, the span of the leading $r$ eigenvectors of $Σ_μ$ forms an \emph{active subspace} of $f(\cdot)$, which contains the directions along which $f(\cdot)$ changes the most and is of particular interest in \emph{ridge approximation}. In this work, we propose a simple algorithm for estimating $Σ_μ$ from random point evaluations of $f(\cdot)$ \emph{without} imposing any structural assumptions on $Σ_μ$. Theoretical guarantees for this algorithm are established with the aid of the same technical tools that have proved valuable in the context of covariance matrix estimation from partial measurements.