Towards Differentiable Resampling

Abstract

Resampling is a key component of sample-based recursive state estimation in particle filters. Recent work explores differentiable particle filters for end-to-end learning. However, resampling remains a challenge in these works, as it is inherently non-differentiable. We address this challenge by replacing traditional resampling with a learned neural network resampler. We present a novel network architecture, the particle transformer, and train it for particle resampling using a likelihood-based loss function over sets of particles. Incorporated into a differentiable particle filter, our model can be end-to-end optimized jointly with the other particle filter components via gradient descent. Our results show that our learned resampler outperforms traditional resampling techniques on synthetic data and in a simulated robot localization task.

References in corpus (1)

• Particle Filter Recurrent Neural Networks

Cited by in corpus (10)

• Conditional Set Generation with Transformers
• Efficient Learning of the Parameters of Non-Linear Models using Differentiable Resampling in Particle Filters
• NeRF in detail: Learning to sample for view synthesis
• Differentiable Particle Filtering via Entropy-Regularized Optimal Transport
• Differentiable Particle Filtering without Modifying the Forward Pass
• Regime Learning for Differentiable Particle Filters
• Differentiable Particle Filtering using Optimal Placement Resampling
• End-To-End Semi-supervised Learning for Differentiable Particle Filters
• Variational Marginal Particle Filters
• Differentiable Nonparametric Belief Propagation