基于局部时序建模与Transformer的机器人运动技能学习

Robot Motor Skill Learning Based on Local Temporal Modeling and Transformer

  • 摘要: 为了提高机器人运动技能学习的效率和精度,提出一种基于序列特征处理的动作决策Transformer模型,命名为门控机制Transformer(gated mechanism Transformer,GMT). 模型以GPT-2为核心,结合门控机制提取隐藏状态特征,通过自回归建模捕捉时间依赖关系,解决机器人运动数据中深层特征难以提取的问题. 同时,利用参数共享策略细化预测特征完成动作推理. GMT在MuJoCo平台的三个机器人运动技能任务中进行了验证. 实验结果表明,GMT在学习效率和精度方面较Decision Transformer最高提升28.5%. 研究表明,GMT能够高效建模机器人运动序列特征,为机器人动作决策提供新的技术方案.

     

    Abstract: To improve the efficiency and accuracy of robot motion skill learning, in this paper, a novel action decision Transformer model based on sequential feature processing, named GMT (gated mechanism Transformer), was proposed. Using GPT-2 as its core architecture and incorporating a gating mechanism to extract hidden state features, the model addressed the challenge of extracting deep features from robot motion data by capturing temporal dependencies through autoregressive modeling. Meanwhile, the model employed parameter sharing strategies to refine predictive features for action inference. GMT was evaluated on three robot motion skill tasks on the MuJoCo platform. Experimental results demonstrate that, compared with Decision Transformer, GMT achieved up to 28.5% improvement in learning efficiency and accuracy. The research indicates that GMT can effectively model robot motion sequence features, providing a new technical approach for robot action decision-making.

     

/

返回文章
返回
Baidu
map