f-GAIL: Learning f-Divergence for Generative Adversarial Imitation Learning
f-GAIL: Learning f-Divergence for Generative Adversarial Imitation Learning
复制标题
DOI:
--
复制
发表时间:
2020-10
期刊:
影响因子:
--
通讯作者:
Xin Zhang;Yanhua Li;Ziming Zhang;Zhi-Li Zhang
中科院分区:
文献类型:
--
作者:
Xin Zhang;Yanhua Li;Ziming Zhang;Zhi-Li Zhang
Imitation learning (IL) aims to learn a policy from expert demonstrations that minimizes the discrepancy between the learner and expert behaviors. Various imitation learning algorithms have been proposed with different pre-determined divergences to quantify the discrepancy. This naturally gives rise to the following question: Given a set of expert demonstrations, which divergence can recover the expert policy more accurately with higher data efficiency? In this work, we propose $f$-GAIL, a new generative adversarial imitation learning (GAIL) model, that automatically learns a discrepancy measure from the $f$-divergence family as well as a policy capable of producing expert-like behaviors. Compared with IL baselines with various predefined divergence measures, $f$-GAIL learns better policies with higher data efficiency in six physics-based control tasks.