In this paper, a new adversarial imitation learning algorithm is proposed based on Weighted Wasserstein distance (WWDAIL). This algorithm employs an enhanced proximal optimization strategy to improve the efficiency and stability of parameter updates. Furthermore, the weighted Wasserstein distance is utilized as a metric for assessing policy differences, thereby increasing the flexibility and dynamism of the measurement. Furthermore, by refining the reward function derived from the discriminator’s output, the reward collection capability and efficiency are significantly enhanced. Experimental results in the Mujoco simulation environment have demonstrated that the proposed algorithm exhibits substantial stability in continuous control tasks and markedly improves the agent’s reward collection ability and efficiency.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Adversarial Imitation Learning Based on Weighted Wasserstein Distance

  • Zhengzuo Qin,
  • Yuejiao Wang,
  • Dongdong Zhao,
  • Shi Yan

摘要

In this paper, a new adversarial imitation learning algorithm is proposed based on Weighted Wasserstein distance (WWDAIL). This algorithm employs an enhanced proximal optimization strategy to improve the efficiency and stability of parameter updates. Furthermore, the weighted Wasserstein distance is utilized as a metric for assessing policy differences, thereby increasing the flexibility and dynamism of the measurement. Furthermore, by refining the reward function derived from the discriminator’s output, the reward collection capability and efficiency are significantly enhanced. Experimental results in the Mujoco simulation environment have demonstrated that the proposed algorithm exhibits substantial stability in continuous control tasks and markedly improves the agent’s reward collection ability and efficiency.