KEEC: Koopman Embedded Equivariant Control

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cheng, Xiaoyuan, Yang, Yiming, Tang, Xiaohang, Jiang, Wei, Hu, Yukun
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916634288979968
author Cheng, Xiaoyuan
Yang, Yiming
Tang, Xiaohang
Jiang, Wei
Hu, Yukun
author_facet Cheng, Xiaoyuan
Yang, Yiming
Tang, Xiaohang
Jiang, Wei
Hu, Yukun
contents An efficient way to control systems with unknown nonlinear dynamics is to find an appropriate embedding or representation for simplified approximation (e.g. linearization), which facilitates system identification and control synthesis. Nevertheless, there has been a lack of embedding methods that can guarantee (i) embedding the dynamical system comprehensively, including the vector fields (ODE form) of the dynamics, and (ii) preserving the consistency of control effect between the original and latent space. To address these challenges, we propose Koopman Embedded Equivariant Control (KEEC) to learn an embedding of the states and vector fields such that a Koopman operator is approximated as the latent dynamics. Due to the Koopman operator's linearity, learning the latent vector fields of the dynamics becomes simply solving linear equations. Thus in KEEC, the analytical form of the greedy control policy, which is dependent on the learned differential information of the dynamics and value function, is also simplified. Meanwhile, KEEC preserves the effectiveness of the control policy in the latent space by preserving the metric in two spaces. Our algorithm achieves superior performances in the experiments conducted on various control domains, including the image-based Pendulum, Lorenz-63 and the wave equation. The code is available at https://github.com/yyimingucl/Koopman-Embedded-Equivariant-Control.
format Preprint
id arxiv_https___arxiv_org_abs_2312_01544
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle KEEC: Koopman Embedded Equivariant Control
Cheng, Xiaoyuan
Yang, Yiming
Tang, Xiaohang
Jiang, Wei
Hu, Yukun
Machine Learning
Artificial Intelligence
Systems and Control
An efficient way to control systems with unknown nonlinear dynamics is to find an appropriate embedding or representation for simplified approximation (e.g. linearization), which facilitates system identification and control synthesis. Nevertheless, there has been a lack of embedding methods that can guarantee (i) embedding the dynamical system comprehensively, including the vector fields (ODE form) of the dynamics, and (ii) preserving the consistency of control effect between the original and latent space. To address these challenges, we propose Koopman Embedded Equivariant Control (KEEC) to learn an embedding of the states and vector fields such that a Koopman operator is approximated as the latent dynamics. Due to the Koopman operator's linearity, learning the latent vector fields of the dynamics becomes simply solving linear equations. Thus in KEEC, the analytical form of the greedy control policy, which is dependent on the learned differential information of the dynamics and value function, is also simplified. Meanwhile, KEEC preserves the effectiveness of the control policy in the latent space by preserving the metric in two spaces. Our algorithm achieves superior performances in the experiments conducted on various control domains, including the image-based Pendulum, Lorenz-63 and the wave equation. The code is available at https://github.com/yyimingucl/Koopman-Embedded-Equivariant-Control.
title KEEC: Koopman Embedded Equivariant Control
topic Machine Learning
Artificial Intelligence
Systems and Control
url https://arxiv.org/abs/2312.01544