Optimization over covariance matrices with a parameterized metric
Organizations: Nanyang Technological University, Singapore · Microsoft India · Indian Institute of Technology Bombay, India
Abstract
The choice of Riemannian metric can strongly influence the convergence of gradient-based optimization over covariance matrices. Euclidean, Bures-Wasserstein and affine-invariant metrics are common choices, but their relative effectiveness depends on the objective. We introduce a two-parameter family defined by , solved for at each tangent vector , that contains all three as exact members, at , and , and extends past them. We treat the choice of member as a particular way of preconditioning for a given problem. To this end, we analyze the conditioning of the Riemannian Hessian at the solution. We show that it obeys a lower bound that depends on only through the exponent . When the Euclidean Hessian is a pure power that mixes no eigendirections, the member attains that bound, and a closed-form criterion identifies the other members that do. We discuss ways to tune for a given problem. Experiments on real covariance data confirm the predicted conditioning and the benefit of tuning . A task covariance example shows a further gain from tuning the shape.