Hierarchical Imitation and Reinforcement Learning
About
We study how to effectively leverage expert feedback to learn sequential decision-making policies. We focus on problems with sparse rewards and long time horizons, which typically pose significant challenges in reinforcement learning. We propose an algorithmic framework, called hierarchical guidance, that leverages the hierarchical structure of the underlying problem to integrate different modes of expert interaction. Our framework can incorporate different combinations of imitation learning (IL) and reinforcement learning (RL) at different levels, leading to dramatic reductions in both expert effort and cost of exploration. Using long-horizon benchmarks, including Montezuma's Revenge, we demonstrate that our approach can learn significantly faster than hierarchical RL, and be significantly more label-efficient than standard IL. We also theoretically analyze labeling cost for certain instantiations of our framework.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multi-hop Question Answering | HotpotQA (test) | F121.16 | 198 | |
| Question Answering | PubMedQA (test) | Accuracy58.8 | 81 | |
| Medical Question Answering | PubMedQA Reasoning Required | Accuracy58.8 | 10 | |
| Code Generation | BigCodeBench instruction split (test) | Pass Rate27.78 | 6 | |
| Complex Retrieval | STARK-PRIME (test) | MRR36.95 | 6 | |
| RAG | HotpotQA (test) | F1 Score21.16 | 6 | |
| Semi-structured Retrieval and Query Optimization | STARK-PRIME (test) | MRR36.95 | 6 | |
| Verified Code Gen. | BigCodeBench (test) | Pass Rate27.78 | 6 |