Dynamics Harmonic Analysis of Robotic Systems: Application in Data-Driven Koopman Modelling
Daniel Ordoñez-Apraez, Vladimir Kostic, Giulio Turrisi, Pietro Novelli, Carlos Mastalli, Claudio Semini, Massimiliano Pontil
TL;DR
This paper tackles the challenge of learning globally linear models for robotic dynamics by exploiting discrete morphological symmetries. It introduces Dynamics Harmonic Analysis (DHA), which decomposes the state space into isotypic subspaces corresponding to irreducible representations of a symmetry group, enabling independent linear dynamics on each subspace. Building on this, the authors propose the equivariant Dynamics Autoencoder (eDAE), a neural architecture that learns a Koopman-operator-based model while enforcing $\mathbb{G}$-equivariance, leading to improved generalization, sample efficiency, and interpretability. Experiments on synthetic symmetric systems and a quadruped robot (mini-cheetah) demonstrate the method’s ability to capture both transient and steady locomotion dynamics with fewer parameters and lower computational cost, aided by an open-access library for reproducibility. The approach offers a principled, symmetry-aware framework for data-driven robotics with broad applicability to systems possessing discrete symmetry groups.
Abstract
We introduce the use of harmonic analysis to decompose the state space of symmetric robotic systems into orthogonal isotypic subspaces. These are lower-dimensional spaces that capture distinct, symmetric, and synergistic motions. For linear dynamics, we characterize how this decomposition leads to a subdivision of the dynamics into independent linear systems on each subspace, a property we term dynamics harmonic analysis (DHA). To exploit this property, we use Koopman operator theory to propose an equivariant deep-learning architecture that leverages the properties of DHA to learn a global linear model of the system dynamics. Our architecture, validated on synthetic systems and the dynamics of locomotion of a quadrupedal robot, exhibits enhanced generalization, sample efficiency, and interpretability, with fewer trainable parameters and computational costs.
