要約
我々は、学習ベースの接触知覚のための形態情報異種グラフニューラルネットワーク(MI-HGNN)を提案する。MI-HGNNのアーキテクチャと接続性はロボットの形態から構築され、ノードとエッジはそれぞれロボットの関節とリンクである。形態情報に基づく制約をニューラルネットワークに組み込むことで、モデルベースの知識を用いた学習ベースのアプローチを改善する。提案するMI-HGNNを2つの接触知覚問題に適用し、2つの四足歩行ロボットを用いて収集した実データとシミュレーションデータの両方を用いた広範な実験を行う。実験の結果、有効性、汎化能力、モデル効率、サンプル効率の観点から、本手法の優位性が実証された。我々のMI-HGNNは、ロボットの形態学的対称性を利用した最先端のモデルの性能を、わずか0.21%のパラメータで8.4%向上させた。本研究では、MI-HGNNを脚式ロボットの接触知覚問題に適用したが、他のタイプのマルチボディ力学系にもシームレスに適用でき、他のロボット学習フレームワークを改善する可能性がある。我々のコードはhttps://github.com/lunarlab-gatech/Morphology-Informed-HGNN。
要約(オリジナル)
We present a Morphology-Informed Heterogeneous Graph Neural Network (MI-HGNN) for learning-based contact perception. The architecture and connectivity of the MI-HGNN are constructed from the robot morphology, in which nodes and edges are robot joints and links, respectively. By incorporating the morphology-informed constraints into a neural network, we improve a learning-based approach using model-based knowledge. We apply the proposed MI-HGNN to two contact perception problems, and conduct extensive experiments using both real-world and simulated data collected using two quadruped robots. Our experiments demonstrate the superiority of our method in terms of effectiveness, generalization ability, model efficiency, and sample efficiency. Our MI-HGNN improved the performance of a state-of-the-art model that leverages robot morphological symmetry by 8.4% with only 0.21% of its parameters. Although MI-HGNN is applied to contact perception problems for legged robots in this work, it can be seamlessly applied to other types of multi-body dynamical systems and has the potential to improve other robot learning frameworks. Our code is made publicly available at https://github.com/lunarlab-gatech/Morphology-Informed-HGNN.
arxiv情報
著者 | Daniel Butterfield,Sandilya Sai Garimella,Nai-Jen Cheng,Lu Gan |
発行日 | 2025-04-03 16:23:01+00:00 |
arxivサイト | arxiv_id(pdf) |