Morphological-Symmetry-Equivariant Heterogeneous Graph Neural Network for Robotic Dynamics Learning

Xie, Fengze, Wei, Sizhe, Song, Yue, Yue, Yisong, Gan, Lu

arXiv.org Artificial Intelligence 

These structural priors are embedded into the learning architecture as constraints, ensuring high generalizability, sample and model efficiency. The proposed MS-HGNN is a versatile and general architecture that is applicable to various multi-body dynamic systems and a wide range of dynamics learning problems.