Nesterov Acceleration for Ensemble Kalman Inversion and Variants

Abstract

Ensemble Kalman inversion (EKI) is a derivative-free, particle-based optimization method for solving inverse problems. It can be shown that EKI approximates a gradient flow, which allows the application of methods for accelerating gradient descent. Here, we show that Nesterov acceleration is effective in speeding up the reduction of the EKI cost function on a variety of inverse problems. We also implement Nesterov acceleration for two EKI variants, unscented Kalman inversion and ensemble transform Kalman inversion. Our specific implementation takes the form of a particle-level nudge that is demonstrably simple to couple in a black-box fashion with any existing EKI variant algorithms, comes with no additional computational expense, and with no additional tuning hyperparameters. This work shows a pathway for future research to translate advances in gradient-based optimization into advances in gradient-free Kalman optimization.

References in corpus (8)

• The Ensemble Kalman Filter for Inverse Problems
• Calibrate, Emulate, Sample
• Ensemble Kalman Inversion: A Derivative-Free Technique For Machine Learning Tasks
• Acceleration Methods
• Ensemble Kalman Inversion: mean-field limit and convergence analysis
• Fast sampling of parameterised Gaussian random fields
• Localization in Ensemble Kalman inversion
• Continuous Time Analysis of Momentum Methods

Cited by in corpus (1)

• Forecast error growth: A dynamic-stochastic model