NAS-X: Neural Adaptive Smoothing via Twisting

Dieterich Lawson,Michael Li,Scott Linderman
DOI: https://doi.org/10.48550/arXiv.2308.14864
IF: 5.414
2023-08-28
Machine Learning
Abstract:We present Neural Adaptive Smoothing via Twisting (NAS-X), a method for learning and inference in sequential latent variable models based on reweighted wake-sleep (RWS). NAS-X works with both discrete and continuous latent variables, and leverages smoothing SMC to fit a broader range of models than traditional RWS methods. We test NAS-X on discrete and continuous tasks and find that it substantially outperforms previous variational and RWS-based methods in inference and parameter recovery.
What problem does this paper attempt to address?