Maximum Entropy Gain Exploration for Long Horizon Multi-goal Reinforcement Learning

Silviu Pitis, Harris Chan, Stephen Zhao, Bradly Stadie, Jimmy Ba
Proceedings of the 37th International Conference on Machine Learning, PMLR 119:7750-7761, 2020.

Abstract

What goals should a multi-goal reinforcement learning agent pursue during training in long-horizon tasks? When the desired (test time) goal distribution is too distant to offer a useful learning signal, we argue that the agent should not pursue unobtainable goals. Instead, it should set its own intrinsic goals that maximize the entropy of the historical achieved goal distribution. We propose to optimize this objective by having the agent pursue past achieved goals in sparsely explored areas of the goal space, which focuses exploration on the frontier of the achievable goal set. We show that our strategy achieves an order of magnitude better sample efficiency than the prior state of the art on long-horizon multi-goal tasks including maze navigation and block stacking.

Cite this Paper


BibTeX
@InProceedings{pmlr-v119-pitis20a, title = {Maximum Entropy Gain Exploration for Long Horizon Multi-goal Reinforcement Learning}, author = {Pitis, Silviu and Chan, Harris and Zhao, Stephen and Stadie, Bradly and Ba, Jimmy}, booktitle = {Proceedings of the 37th International Conference on Machine Learning}, pages = {7750--7761}, year = {2020}, editor = {III, Hal Daumé and Singh, Aarti}, volume = {119}, series = {Proceedings of Machine Learning Research}, month = {13--18 Jul}, publisher = {PMLR}, pdf = {http://proceedings.mlr.press/v119/pitis20a/pitis20a.pdf}, url = {https://proceedings.mlr.press/v119/pitis20a.html}, abstract = {What goals should a multi-goal reinforcement learning agent pursue during training in long-horizon tasks? When the desired (test time) goal distribution is too distant to offer a useful learning signal, we argue that the agent should not pursue unobtainable goals. Instead, it should set its own intrinsic goals that maximize the entropy of the historical achieved goal distribution. We propose to optimize this objective by having the agent pursue past achieved goals in sparsely explored areas of the goal space, which focuses exploration on the frontier of the achievable goal set. We show that our strategy achieves an order of magnitude better sample efficiency than the prior state of the art on long-horizon multi-goal tasks including maze navigation and block stacking.} }
Endnote
%0 Conference Paper %T Maximum Entropy Gain Exploration for Long Horizon Multi-goal Reinforcement Learning %A Silviu Pitis %A Harris Chan %A Stephen Zhao %A Bradly Stadie %A Jimmy Ba %B Proceedings of the 37th International Conference on Machine Learning %C Proceedings of Machine Learning Research %D 2020 %E Hal Daumé III %E Aarti Singh %F pmlr-v119-pitis20a %I PMLR %P 7750--7761 %U https://proceedings.mlr.press/v119/pitis20a.html %V 119 %X What goals should a multi-goal reinforcement learning agent pursue during training in long-horizon tasks? When the desired (test time) goal distribution is too distant to offer a useful learning signal, we argue that the agent should not pursue unobtainable goals. Instead, it should set its own intrinsic goals that maximize the entropy of the historical achieved goal distribution. We propose to optimize this objective by having the agent pursue past achieved goals in sparsely explored areas of the goal space, which focuses exploration on the frontier of the achievable goal set. We show that our strategy achieves an order of magnitude better sample efficiency than the prior state of the art on long-horizon multi-goal tasks including maze navigation and block stacking.
APA
Pitis, S., Chan, H., Zhao, S., Stadie, B. & Ba, J.. (2020). Maximum Entropy Gain Exploration for Long Horizon Multi-goal Reinforcement Learning. Proceedings of the 37th International Conference on Machine Learning, in Proceedings of Machine Learning Research 119:7750-7761 Available from https://proceedings.mlr.press/v119/pitis20a.html.

Related Material