Current browse context:
cs.AI
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Meta Reinforcement Learning with Autonomous Inference of Subtask Dependencies
(Submitted on 1 Jan 2020 (v1), last revised 14 Apr 2020 (this version, v2))
Abstract: We propose and address a novel few-shot RL problem, where a task is characterized by a subtask graph which describes a set of subtasks and their dependencies that are unknown to the agent. The agent needs to quickly adapt to the task over few episodes during adaptation phase to maximize the return in the test phase. Instead of directly learning a meta-policy, we develop a Meta-learner with Subtask Graph Inference(MSGI), which infers the latent parameter of the task by interacting with the environment and maximizes the return given the latent parameter. To facilitate learning, we adopt an intrinsic reward inspired by upper confidence bound (UCB) that encourages efficient exploration. Our experiment results on two grid-world domains and StarCraft II environments show that the proposed method is able to accurately infer the latent task parameter, and to adapt more efficiently than existing meta RL and hierarchical RL methods.
Submission history
From: Sungryull Sohn [view email][v1] Wed, 1 Jan 2020 17:34:00 GMT (9070kb,D)
[v2] Tue, 14 Apr 2020 01:44:03 GMT (9069kb,D)
Link back to: arXiv, form interface, contact.