DisGNet: A Distance Graph Neural Network for Forward Kinematics Learning of Gough-Stewart Platform

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhu, Huizhi, Xu, Wenxia, Huang, Jian, Li, Jiaxin
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866913234396643328
author Zhu, Huizhi
Xu, Wenxia
Huang, Jian
Li, Jiaxin
author_facet Zhu, Huizhi
Xu, Wenxia
Huang, Jian
Li, Jiaxin
contents In this paper, we propose a graph neural network, DisGNet, for learning the graph distance matrix to address the forward kinematics problem of the Gough-Stewart platform. DisGNet employs the k-FWL algorithm for message-passing, providing high expressiveness with a small parameter count, making it suitable for practical deployment. Additionally, we introduce the GPU-friendly Newton-Raphson method, an efficient parallelized optimization method executed on the GPU to refine DisGNet's output poses, achieving ultra-high-precision pose. This novel two-stage approach delivers ultra-high precision output while meeting real-time requirements. Our results indicate that on our dataset, DisGNet can achieves error accuracys below 1mm and 1deg at 79.8\% and 98.2\%, respectively. As executed on a GPU, our two-stage method can ensure the requirement for real-time computation. Codes are released at https://github.com/FLAMEZZ5201/DisGNet.
format Preprint
id arxiv_https___arxiv_org_abs_2402_09077
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle DisGNet: A Distance Graph Neural Network for Forward Kinematics Learning of Gough-Stewart Platform
Zhu, Huizhi
Xu, Wenxia
Huang, Jian
Li, Jiaxin
Robotics
Machine Learning
In this paper, we propose a graph neural network, DisGNet, for learning the graph distance matrix to address the forward kinematics problem of the Gough-Stewart platform. DisGNet employs the k-FWL algorithm for message-passing, providing high expressiveness with a small parameter count, making it suitable for practical deployment. Additionally, we introduce the GPU-friendly Newton-Raphson method, an efficient parallelized optimization method executed on the GPU to refine DisGNet's output poses, achieving ultra-high-precision pose. This novel two-stage approach delivers ultra-high precision output while meeting real-time requirements. Our results indicate that on our dataset, DisGNet can achieves error accuracys below 1mm and 1deg at 79.8\% and 98.2\%, respectively. As executed on a GPU, our two-stage method can ensure the requirement for real-time computation. Codes are released at https://github.com/FLAMEZZ5201/DisGNet.
title DisGNet: A Distance Graph Neural Network for Forward Kinematics Learning of Gough-Stewart Platform
topic Robotics
Machine Learning
url https://arxiv.org/abs/2402.09077