RoboKoop: Efficient Control Conditioned Representations from Visual Input in Robotics using Koopman Operator
Fuente:
arXiv
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866912015380905984 |
|---|---|
| author | Kumawat, Hemant Chakraborty, Biswadeep Mukhopadhyay, Saibal |
| author_facet | Kumawat, Hemant Chakraborty, Biswadeep Mukhopadhyay, Saibal |
| contents | Developing agents that can perform complex control tasks from high-dimensional observations is a core ability of autonomous agents that requires underlying robust task control policies and adapting the underlying visual representations to the task. Most existing policies need a lot of training samples and treat this problem from the lens of two-stage learning with a controller learned on top of pre-trained vision models. We approach this problem from the lens of Koopman theory and learn visual representations from robotic agents conditioned on specific downstream tasks in the context of learning stabilizing control for the agent. We introduce a Contrastive Spectral Koopman Embedding network that allows us to learn efficient linearized visual representations from the agent's visual data in a high dimensional latent space and utilizes reinforcement learning to perform off-policy control on top of the extracted representations with a linear controller. Our method enhances stability and control in gradient dynamics over time, significantly outperforming existing approaches by improving efficiency and accuracy in learning task policies over extended horizons. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2409_03107 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | RoboKoop: Efficient Control Conditioned Representations from Visual Input in Robotics using Koopman Operator Kumawat, Hemant Chakraborty, Biswadeep Mukhopadhyay, Saibal Robotics Machine Learning Systems and Control Developing agents that can perform complex control tasks from high-dimensional observations is a core ability of autonomous agents that requires underlying robust task control policies and adapting the underlying visual representations to the task. Most existing policies need a lot of training samples and treat this problem from the lens of two-stage learning with a controller learned on top of pre-trained vision models. We approach this problem from the lens of Koopman theory and learn visual representations from robotic agents conditioned on specific downstream tasks in the context of learning stabilizing control for the agent. We introduce a Contrastive Spectral Koopman Embedding network that allows us to learn efficient linearized visual representations from the agent's visual data in a high dimensional latent space and utilizes reinforcement learning to perform off-policy control on top of the extracted representations with a linear controller. Our method enhances stability and control in gradient dynamics over time, significantly outperforming existing approaches by improving efficiency and accuracy in learning task policies over extended horizons. |
| title | RoboKoop: Efficient Control Conditioned Representations from Visual Input in Robotics using Koopman Operator |
| topic | Robotics Machine Learning Systems and Control |
| url | https://arxiv.org/abs/2409.03107 |