Title
Multi-Task Reinforcement Learning in Reproducing Kernel Hilbert Spaces via Cross-Learning
Abstract
Reinforcement learning is a framework to optimize an agent's policy using rewards that are revealed by the system as a response to an action. In its standard form, reinforcement learning involves a single agent that uses its policy to accomplish a specific task. These methods require large amounts of reward samples to achieve good performance, and may not generalize well when the task is modified, even if the new task is related. In this paper we are interested in a collaborative scheme in which multiple policies are optimized jointly. To this end, we we introduce cross-learning, in which policies are trained for related tasks in separate environments, and they are constrained to be close to one another. Two properties make our new approach attractive: (i) it produces a multi-task central policy that can be used as a starting point to adapt quickly to one of the tasks trained for, and (ii) as in meta-learning, it adapts to environments related but different to those seen during training. We focus on policies belonging to reproducing kernel Hilbert spaces for which we bound the distance between the task-specific policies and the cross-learned policy. To solve the resulting optimization problem, we resort to a projected policy gradient algorithm and prove that it converges to a near-optimal solution with high probability. We evaluate our methodology with a navigation example in which an agent moves through environments with obstacles of multiple shapes and avoids obstacles not trained for.
Year
DOI
Venue
2021
10.1109/TSP.2021.3122303
IEEE TRANSACTIONS ON SIGNAL PROCESSING
Keywords
DocType
Volume
Task analysis, Kernel, Training, Navigation, Convergence, Reinforcement learning, Optimization, Reinforcement learning, multi-task learning, meta-learning, optimization
Journal
69
ISSN
Citations 
PageRank 
1053-587X
0
0.34
References 
Authors
0
4
Name
Order
Citations
PageRank
Juan Cervino100.34
Juan Andres Bazerque200.34
Miguel Calvo-Fullana3135.70
Alejandro Ribeiro42817221.08