SPARSE CHOLESKY FACTORIZATION BY KULLBACK-LEIBLER MINIMIZATION

被引:51
作者
Schaefer, Florian [1 ]
Katzfuss, Matthias [2 ]
Owhadi, Houman [1 ]
机构
[1] CALTECH, Comp & Math Sci, Pasadena, CA 91125 USA
[2] Texas A&M Univ, Dept Stat, College Stn, TX 77843 USA
基金
美国国家科学基金会;
关键词
Cholesky factorization; screening effect; Vecchia approximation; factorized approximate inverse; Gaussian process regression; integral equation; FAST MULTIPOLE METHOD; EFFICIENT; PRECONDITIONER; APPROXIMATION; INTERPOLATION; MODELS;
D O I
10.1137/20M1336254
中图分类号
O29 [应用数学];
学科分类号
070104 ;
摘要
We propose to compute a sparse approximate inverse Cholesky factor L of a dense covariance matrix circle minus by minimizing the Kullback-Leibler divergence between the Gaussian distributions N (0, circle minus) and N (0, L-inverted perpendicular L-1), subject to a sparsity constraint. Surprisingly, this problem has a closed-form solution that can be computed efficiently, recovering the popular Vecchia approximation in spatial statistics. Based on recent results on the approximate sparsity of inverse Cholesky factors of circle minus obtained from pairwise evaluation of Green's functions of elliptic boundary-value problems at points {x(i)} 1 <= i <= N subset of R-d, we propose an elimination ordering and sparsity pattern that allows us to compute epsilon -approximate inverse Cholesky factors of such circle minus in computational complexity O (N log(N/epsilon)(d)) in space and O (N log(N/epsilon)(2d)) in time. To the best of our knowledge, this is the best asymptotic complexity for this class of problems. Furthermore, our method is embarrassingly parallel, automatically exploits low-dimensional structure in the data, and can perform Gaussian-process regression in linear (in N) space complexity. Motivated by its optimality properties, we propose applying our method to the joint covariance of training and prediction points in Gaussian-process regression, greatly improving stability and computational cost. Finally, we show how to apply our method to the important setting of Gaussian processes with additive noise, compromising neither accuracy nor computational complexity.
引用
收藏
页码:A2019 / A2046
页数:28
相关论文
共 66 条
[11]   THE INVERSE FAST MULTIPOLE METHOD: USING A FAST APPROXIMATE DIRECT SOLVER AS A PRECONDITIONER FOR DENSE LINEAR SYSTEMS [J].
Coulier, Pieter ;
Pouransari, Hadi ;
Darve, Eric .
SIAM JOURNAL ON SCIENTIFIC COMPUTING, 2017, 39 (03) :A761-A796
[12]   Efficient mesh deformation based on radial basis function interpolation by means of the inverse fast multipole method [J].
Coulier, Pieter ;
Darve, Eric .
COMPUTER METHODS IN APPLIED MECHANICS AND ENGINEERING, 2016, 308 :286-309
[13]   Hierarchical Nearest-Neighbor Gaussian Process Models for Large Geostatistical Datasets [J].
Datta, Abhirup ;
Banerjee, Sudipto ;
Finley, Andrew O. ;
Gelfand, Alan E. .
JOURNAL OF THE AMERICAN STATISTICAL ASSOCIATION, 2016, 111 (514) :800-812
[14]   Interpreting Kullback-Leibler divergence with the Neyman-Pearson lemma [J].
Eguchi, Shinto ;
Copas, John .
JOURNAL OF MULTIVARIATE ANALYSIS, 2006, 97 (09) :2034-2040
[15]  
Eremin A. Y., 1998, ZAP NAUCHN SEM S PET, V248, P17, DOI [DOI 10.1007/BF02672769, 10.1007/BF02672769.]
[16]   A novel Factorized Sparse Approximate Inverse preconditioner with supernodes [J].
Ferronato, Massimiliano ;
Janna, Carlo ;
Gambolati, Giuseppe .
INTERNATIONAL CONFERENCE ON COMPUTATIONAL SCIENCE, ICCS 2015 COMPUTATIONAL SCIENCE AT THE GATES OF NATURE, 2015, 51 :266-275
[17]   Efficient SVM training using low-rank kernel representations [J].
Fine, S ;
Scheinberg, K .
JOURNAL OF MACHINE LEARNING RESEARCH, 2002, 2 (02) :243-264
[18]   Spectral grouping using the Nystrom method [J].
Fowlkes, C ;
Belongie, S ;
Chung, F ;
Malik, J .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2004, 26 (02) :214-225
[19]   Covariance tapering for interpolation of large spatial datasets [J].
Furrer, Reinhard ;
Genton, Marc G. ;
Nychka, Douglas .
JOURNAL OF COMPUTATIONAL AND GRAPHICAL STATISTICS, 2006, 15 (03) :502-523
[20]   Stochastic models that separate fractal dimension and the Hurst effect [J].
Gneiting, T ;
Schlather, M .
SIAM REVIEW, 2004, 46 (02) :269-282