Grassmann stein variational gradient descent
File(s)2202.03297v2.pdf (9.32 MB)
Accepted version
Author(s)
Liu, Xing
Zhu, Harrison
Ton, Jean-Francois
Wynne, George
Duncan, Andrew
Type
Conference Paper
Abstract
Stein variational gradient descent (SVGD) is a deterministic particle inference algorithm that provides an efficient alternative to Markov chain Monte Carlo. However, SVGD has been found to suffer from variance underestimation when the dimensionality of the target distribution is high. Recent developments have advocated projecting both the score function and the data onto real lines to sidestep this issue, although this can severely overestimate the epistemic (model) uncertainty. In this work, we propose Grassmann Stein variational gradient descent (GSVGD) as an alternative approach, which permits projections onto arbitrary dimensional subspaces. Compared with other variants of SVGD that rely on dimensionality reduction, GSVGD updates the projectors simultaneously for the score function and the data, and the optimal projectors are determined through a coupled Grassmann-valued diffusion process which explores favourable subspaces. Both our theoretical and experimental results suggest that GSVGD enjoys efficient state-space exploration in high-dimensional problems that have an intrinsic low-dimensional structure.
Editor(s)
Camps-Valls, G
Ruiz, FJR
Valera, I
Date Issued
2022-01-01
Date Acceptance
2022-03-01
Citation
INTERNATIONAL CONFERENCE ON ARTIFICIAL INTELLIGENCE AND STATISTICS, VOL 151, 2022, 151, pp.1-20
ISSN
2640-3498
Publisher
JMLR-JOURNAL MACHINE LEARNING RESEARCH
Start Page
1
End Page
20
Journal / Book Title
INTERNATIONAL CONFERENCE ON ARTIFICIAL INTELLIGENCE AND STATISTICS, VOL 151
Volume
151
Identifier
https://www.webofscience.com/api/gateway?GWVersion=2&SrcApp=PARTNER_APP&SrcAuth=LinksAMR&KeyUT=WOS:000828072702003&DestLinkType=FullRecord&DestApp=ALL_WOS&UsrCustomerID=1ba7043ffcc86c417c072aa74d649202
Source
International Conference on Artificial Intelligence and Statistics
Subjects
Computer Science
Computer Science, Artificial Intelligence
KERNELS
Mathematics
Physical Sciences
Science & Technology
Statistics & Probability
Technology
Publication Status
Published
Start Date
2022-03-28
Finish Date
2022-03-30
Coverage Spatial
ELECTR NETWORK
Date Publish Online
2022-03-28