Optimal Potential Shaping on SE(3) via Neural ODEs on Lie Groups

Abstract

This work presents a novel approach for the optimization of dynamic systems on finite-dimensional Lie groups. We rephrase dynamic systems as so-called neural ordinary differential equations (neural ODEs), and formulate the optimization problem on Lie groups. A gradient descent optimization algorithm is presented to tackle the optimization numerically. Our algorithm is scalable, and applicable to any finite dimensional Lie group, including matrix Lie groups. By representing the system at the Lie algebra level, we reduce the computational cost of the gradient computation. In an extensive example, optimal potential energy shaping for control of a rigid body is treated. The optimal control problem is phrased as an optimization of a neural ODE on the Lie group SE(3), and the controller is iteratively optimized. The final controller is validated on a state-regulation task.

arXiv comment21 pages, 11 figures

References in corpus (8)

• An overview of gradient descent optimization algorithms
• How to Train Your Robot with Deep Reinforcement Learning; Lessons We've Learned
• An introduction to Lie group integrators -- basics, new developments and applications
• Active Learning in Robotics: A Review of Control Principles
• A Diffusion Theory For Deep Learning Dynamics: Stochastic Gradient Descent Exponentially Favors Flat Minima
• Machine Learning Lie Structures & Applications to Physics
• Neural Ordinary Differential Equations on Manifolds
• Discovering Efficient Periodic Behaviours in Mechanical Systems via Neural Approximators