Learning Nonholonomic Dynamics with Constraint Discovery
Baiyue Wang, Anthony Bloch
TL;DR
The paper addresses learning nonholonomic dynamics with constraint discovery by leveraging Hamel's formalism and symmetry reduction to a Lie algebra. It introduces a four-step learning procedure that parameterizes the constraint via a neural network, constructs a moving-basis representation, and trains a Neural ODE to recover the dynamics from trajectory data on the tangent bundle $TQ$. The Rolling Disk on $\mathbb{R}\times SE(2)$ serves as a case study, where the approach recovers the constraint map $\tilde{\mathcal{A}}(r,g)$ and matches ground-truth momenta, albeit with some limitations due to numerical integration and data diversity. The work demonstrates constraint discovery from data without explicit constraint specification and discusses practical implications, potential improvements (more diverse trajectories, symmetry-preserving integrators), and broader applicability to real-world nonholonomic systems.
Abstract
We consider learning nonholonomic dynamical systems while discovering the constraints, and describe in detail the case of the rolling disk. A nonholonomic system is a system subject to nonholonomic constraints. Unlike holonomic constraints, nonholonomic constraints do not define a sub-manifold on the configuration space. Therefore, the inverse problem of finding the constraints has to involve the tangent bundle. This paper discusses a general procedure to learn the dynamics of a nonholonomic system through Hamel's formalism, while discovering the system constraint by parameterizing it, given the data set of discrete trajectories on the tangent bundle $TQ$. We prove that there is a local minimum for convergence of the network. We also preserve symmetry of the system by reducing the Lagrangian to the Lie algebra of the selected group.
