A novel trajectory learning method for robotic arms based on Gaussian Mixture Model and k-value selection algorithm

作者：Jingnan Yan, Yue Wu, Kexin Ji, Cheng Cheng, Yili Zheng · 发表于：PLoS ONE · 年份：2025 · DOI：10.1371/journal.pone.0318403 · 被引用次数：8 · 研究领域：Robot Manipulation and Learning、Prosthetics and Rehabilitation Robotics、Robotic Mechanisms and Dynamics

In the field of robotic arm trajectory imitation learning, Gaussian Mixture Models are widely used for their ability to capture the characteristics of complex trajectories. However, one major challenge in utilizing these models lies in the initialization process, particularly in determining the number of Gaussian kernels, or the k-value. The choice of the k-value significantly impacts the model's performance, and traditional methods, such as random selection or selection based on empirical knowledge, often lead to suboptimal outcomes. To address this challenge, this paper proposes a novel trajectory learning method for robotic arms that combines Gaussian Mixture Model with a k-value selection algorithm. The proposed approach leverages the principles of the elbow method along with the properties of exponential functions, correction terms, and weight adjustments to accurately determine the optimal k-value. Next, k-means clustering is applied with the optimal k-value to initialize the parameters of the Gaussian Mixture Model, which are then refined and trained through the Expectation-Maximization algorithm. The resulting model parameters are then employed in Gaussian Mixture Regression to generate the robotic arm trajectories. The effectiveness of the proposed method is validated through both simulation experiments with two-dimensional theoretical nonlinear dynamic systems and physical experiments with actual robotic arm data. Experimental results demonstrate that, compared to t...