RTSNet: Learning to Smooth in Partially Known State-Space Models

Guy Revach, Xiaoyong Ni, Nir Shlezinger, Ruud J.G. Van Sloun, Yonina C. Eldar

Research output: Contribution to journalArticlepeer-review

Abstract

The smoothing task is core to many signal-processing applications. A widely popular smoother is the RTS algorithm, which achieves minimal mean-squared error recovery with low complexity for linear Gaussianstate-space (SS) models, yet is limited in systems that are only partially known, as well as nonlinear and non-Gaussian. In this work, we propose RTSNet, a highly efficient model-based and data-driven smoothing algorithm suitable for partially known SS models. RTSNet integrates dedicated trainable models into the flow of the classical RTS smoother, while iteratively refining its sequence estimate via deep unfolding methodology. As a result, RTSNet learns from data to reliably smooth when operating under model mismatch and nonlinearities while retaining the efficiency and interpretability of the traditional RTS smoothing algorithm. Our empirical study demonstrates that RTSNet overcomes nonlinearities and model mismatch, outperforming classic smoothers operating with both mismatched and accurate domain knowledge. Moreover, while RTSNet is based on compact neural networks, which leads to faster training and inference times, it demonstrates improved performance over previously proposed deep smoothers in nonlinear settings.

Original languageEnglish
Pages (from-to)4441-4456
Number of pages16
JournalIEEE Transactions on Signal Processing
Volume71
DOIs
StatePublished - 1 Jan 2023

Keywords

  • Time series analysis
  • deep learning
  • kalman filters

ASJC Scopus subject areas

  • Signal Processing
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'RTSNet: Learning to Smooth in Partially Known State-Space Models'. Together they form a unique fingerprint.

Cite this