Learning from Demonstrations with Koopman Operators
Example of learnt vector fields (gray) from training demonstrations (red), with matching trajectories (black).Project Description
We use a Koopman-based approach for learning dynamical systems from demonstrations. Our sample-efficient method captures nonlinearities in demonstrations, allows us to learn attractors that bring the system towards the desired demonstration when away, mimics the demonstration motion, and stops when the system reaches the end of the demonstration.
Beyond this, since we adopt a Koopman based approach, we model the nonlinear dynamics by lifting the system into a higher dimensional yet linear space, where the dynamics evolve linearly. This enables linear analaysis as well as physically interpretable feature extraction, where the state space can be partitioned into regions exihibiting similar dynamics, revealing system attractors.
Relevant Publication: Li, Alice K., Thales C. Silva, Victoria Edwards, Vijay Kumar, and M. Ani Hsieh. “KoopMotion: Learning Almost Divergence Free Koopman Flow Fields for Motion Planning.” CoRL 2025. Link to paper.