Efficient Reinforcement Learning using Linear Koopman Dynamics for Nonlinear Robotic Systems

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Hao, Wenjian, Fang, Yuxuan, Lu, Zehui, Mou, Shaoshuai
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918461067755520
author Hao, Wenjian
Fang, Yuxuan
Lu, Zehui
Mou, Shaoshuai
author_facet Hao, Wenjian
Fang, Yuxuan
Lu, Zehui
Mou, Shaoshuai
contents This paper presents a model-based reinforcement learning (RL) framework for optimal closed-loop control of nonlinear robotic systems. The proposed approach learns linear lifted dynamics through Koopman operator theory and integrates the resulting model into an actor-critic architecture for policy optimization, where the policy represents a parameterized closed-loop controller. To reduce computational cost and mitigate model rollout errors, policy gradients are estimated using one-step predictions of the learned dynamics rather than multi-step propagation. This leads to an online mini-batch policy gradient framework that enables policy improvement from streamed interaction data. The proposed framework is evaluated on several simulated nonlinear control benchmarks and two real-world hardware platforms, including a Kinova Gen3 robotic arm and a Unitree Go1 quadruped. Experimental results demonstrate improved sample efficiency over model-free RL baselines, superior control performance relative to model-based RL baselines, and control performance comparable to classical model-based methods that rely on exact system dynamics.
format Preprint
id arxiv_https___arxiv_org_abs_2604_19980
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Efficient Reinforcement Learning using Linear Koopman Dynamics for Nonlinear Robotic Systems
Hao, Wenjian
Fang, Yuxuan
Lu, Zehui
Mou, Shaoshuai
Robotics
Systems and Control
This paper presents a model-based reinforcement learning (RL) framework for optimal closed-loop control of nonlinear robotic systems. The proposed approach learns linear lifted dynamics through Koopman operator theory and integrates the resulting model into an actor-critic architecture for policy optimization, where the policy represents a parameterized closed-loop controller. To reduce computational cost and mitigate model rollout errors, policy gradients are estimated using one-step predictions of the learned dynamics rather than multi-step propagation. This leads to an online mini-batch policy gradient framework that enables policy improvement from streamed interaction data. The proposed framework is evaluated on several simulated nonlinear control benchmarks and two real-world hardware platforms, including a Kinova Gen3 robotic arm and a Unitree Go1 quadruped. Experimental results demonstrate improved sample efficiency over model-free RL baselines, superior control performance relative to model-based RL baselines, and control performance comparable to classical model-based methods that rely on exact system dynamics.
title Efficient Reinforcement Learning using Linear Koopman Dynamics for Nonlinear Robotic Systems
topic Robotics
Systems and Control
url https://arxiv.org/abs/2604.19980