← 返回论文检索
ICLR 2024PosterAccept (poster)

Course Correcting Koopman Representations

Mahan Fathi, Clement Gehring, Jonathan Pilault, David Kanaa, Pierre-Luc Bacon, Ross Goroshin

Google DeepMind, Mila · Massachusetts Institute of Technology · MILA · Université de Montréal · Université de Montréal, Mila · Google DeepMind

PDF 由论文原始站点提供,PaperCompass 不保存论文文件。

摘要

Koopman representations aim to learn features of nonlinear dynamical systems (NLDS) which lead to linear dynamics in the latent space. Theoretically, such features can be used to simplify many problems in modeling and control of NLDS. In this work we study autoencoder formulations of this problem, and different ways they can be used to model dynamics, specifically for future state prediction over long horizons. We discover several limitations of predicting future states in the latent space and propose an inference-time mechanism, which we refer to as Periodic Reencoding, for faithfully capturing long term dynamics. We justify this method both analytically and empirically via experiments in low and high dimensional NLDS.