Authors:
Manuel Bazzi
1
;
Asad Shahid
2
;
Christopher Agia
3
;
John Alora
3
;
Marco Forgione
2
;
Dario Piga
2
;
Francesco Braghin
1
;
Marco Pavone
3
and
Loris Roveda
2
;
3
Affiliations:
1
Department of Mechanical Engineering, Politecnico di Milano, Italy
;
2
Istituto Dalle Molle di Studi Sull’Intelligenza Artificiale (IDSIA USI-SUPSI), Scuola Universitaria Professionale della Svizzera Italiana, DTI , Italy
;
3
Stanford University, U.S.A.
Keyword(s):
Transformers, In-Context Learning, Meta Learning, Transfer Learning, Deep Learning, Isaac Gym, Robot Dynamics Modeling.
Abstract:
The landscape of Deep Learning has experienced a major shift with the pervasive adoption of Transformer-based architectures, particularly in Natural Language Processing (NLP). Novel avenues for physical applications, such as solving Partial Differential Equations and Image Vision, have been explored. However, in challenging domains like robotics, where high non-linearity poses significant challenges, Transformer-based applications are scarce. While Transformers have been used to provide robots with knowledge about high-level tasks, few efforts have been made to perform system identification. This paper proposes a novel methodology to learn a meta-dynamical model of a high-dimensional physical system, such as the Franka robotic arm, using a Transformer-based architecture without prior knowledge of the system’s physical parameters. The objective is to predict quantities of interest (end-effector pose and joint positions) given the torque signals for each joint. This prediction can be u
seful as a component for Deep Model Predictive Control frameworks in robotics. The meta-model establishes the correlation between torques and positions and predicts the output for the complete trajectory. This work provides empirical evidence of the efficacy of the in-context learning paradigm, suggesting future improvements in learning the dynamics of robotic systems without explicit knowledge of physical parameters. Code, videos, and supplementary materials can be found at project website.
(More)