Going with the Flow: Koopman Behavioral Models as Implicit Planners for Visuo-Motor Dexterity

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Han, Yunhai, Bai, Linhao, Xiao, Ziyu, Yang, Zhaodong, Choudhary, Yogita, Jha, Krishna, Kong, Chuizheng, Kousik, Shreyas, Ravichandar, Harish
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912895000903680
author Han, Yunhai
Bai, Linhao
Xiao, Ziyu
Yang, Zhaodong
Choudhary, Yogita
Jha, Krishna
Kong, Chuizheng
Kousik, Shreyas
Ravichandar, Harish
author_facet Han, Yunhai
Bai, Linhao
Xiao, Ziyu
Yang, Zhaodong
Choudhary, Yogita
Jha, Krishna
Kong, Chuizheng
Kousik, Shreyas
Ravichandar, Harish
contents There has been rapid and dramatic progress in learning complex visuo-motor manipulation skills from demonstrations, thanks in part to expressive policy classes that employ diffusion- and transformer-based backbones. However, these design choices require significant data and computational resources and remain far from reliable, particularly within the context of multi-fingered dexterous manipulation. Fundamentally, they model skills as reactive mappings and rely on fixed-horizon action chunking to mitigate jitter, creating a rigid trade-off between temporal coherence and reactivity. In this work, we introduce Unified Behavioral Models (UBMs), a framework that learns to represent dexterous skills as coupled dynamical systems that capture how visual features of the environment (visual flow) and proprioceptive states of the robot (action flow) co-evolve. By capturing such behavioral dynamics, UBMs can ensure temporal coherence by construction rather than by heuristic averaging. To operationalize these models, we propose Koopman-UBM, a first instantiation of UBMs that leverages Koopman Operator theory to effectively learn a unified representation in which the joint flow of latent visual and proprioceptive features is governed by a structured linear system. We demonstrate that Koopman-UBM can be viewed as an implicit planner: given an initial condition, it computes the desired robot behavior with the resulting flow of visual features over the entire skill horizon. To enable reactivity, we introduce an online replanning strategy in which the model acts as its own runtime monitor that automatically triggers replanning when predicted and observed visual flow diverge. Across seven simulated and two real-world tasks, we demonstrate that K-UBM matches or exceeds the performance of SOTA baselines, while offering faster inference, smooth execution, robustness to occlusions, and flexible replanning.
format Preprint
id arxiv_https___arxiv_org_abs_2602_07413
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Going with the Flow: Koopman Behavioral Models as Implicit Planners for Visuo-Motor Dexterity
Han, Yunhai
Bai, Linhao
Xiao, Ziyu
Yang, Zhaodong
Choudhary, Yogita
Jha, Krishna
Kong, Chuizheng
Kousik, Shreyas
Ravichandar, Harish
Robotics
There has been rapid and dramatic progress in learning complex visuo-motor manipulation skills from demonstrations, thanks in part to expressive policy classes that employ diffusion- and transformer-based backbones. However, these design choices require significant data and computational resources and remain far from reliable, particularly within the context of multi-fingered dexterous manipulation. Fundamentally, they model skills as reactive mappings and rely on fixed-horizon action chunking to mitigate jitter, creating a rigid trade-off between temporal coherence and reactivity. In this work, we introduce Unified Behavioral Models (UBMs), a framework that learns to represent dexterous skills as coupled dynamical systems that capture how visual features of the environment (visual flow) and proprioceptive states of the robot (action flow) co-evolve. By capturing such behavioral dynamics, UBMs can ensure temporal coherence by construction rather than by heuristic averaging. To operationalize these models, we propose Koopman-UBM, a first instantiation of UBMs that leverages Koopman Operator theory to effectively learn a unified representation in which the joint flow of latent visual and proprioceptive features is governed by a structured linear system. We demonstrate that Koopman-UBM can be viewed as an implicit planner: given an initial condition, it computes the desired robot behavior with the resulting flow of visual features over the entire skill horizon. To enable reactivity, we introduce an online replanning strategy in which the model acts as its own runtime monitor that automatically triggers replanning when predicted and observed visual flow diverge. Across seven simulated and two real-world tasks, we demonstrate that K-UBM matches or exceeds the performance of SOTA baselines, while offering faster inference, smooth execution, robustness to occlusions, and flexible replanning.
title Going with the Flow: Koopman Behavioral Models as Implicit Planners for Visuo-Motor Dexterity
topic Robotics
url https://arxiv.org/abs/2602.07413