English

Robust Knowledge Transfer in Tiered Reinforcement Learning

Machine Learning 2024-06-14 v3 Artificial Intelligence Machine Learning

Abstract

In this paper, we study the Tiered Reinforcement Learning setting, a parallel transfer learning framework, where the goal is to transfer knowledge from the low-tier (source) task to the high-tier (target) task to reduce the exploration risk of the latter while solving the two tasks in parallel. Unlike previous work, we do not assume the low-tier and high-tier tasks share the same dynamics or reward functions, and focus on robust knowledge transfer without prior knowledge on the task similarity. We identify a natural and necessary condition called the ``Optimal Value Dominance'' for our objective. Under this condition, we propose novel online learning algorithms such that, for the high-tier task, it can achieve constant regret on partial states depending on the task similarity and retain near-optimal regret when the two tasks are dissimilar, while for the low-tier task, it can keep near-optimal without making sacrifice. Moreover, we further study the setting with multiple low-tier tasks, and propose a novel transfer source selection mechanism, which can ensemble the information from all low-tier tasks and allow provable benefits on a much larger state-action space.

Keywords

Cite

@article{arxiv.2302.05534,
  title  = {Robust Knowledge Transfer in Tiered Reinforcement Learning},
  author = {Jiawei Huang and Niao He},
  journal= {arXiv preprint arXiv:2302.05534},
  year   = {2024}
}

Comments

47 Pages; 1 Figure; NeurIPS 2023

R2 v1 2026-06-28T08:37:28.789Z