References & Citations
Computer Science > Machine Learning
Title: Exploration and Anti-Exploration with Distributional Random Network Distillation
(Submitted on 18 Jan 2024 (v1), last revised 20 May 2024 (this version, v4))
Abstract: Exploration remains a critical issue in deep reinforcement learning for an agent to attain high returns in unknown environments. Although the prevailing exploration Random Network Distillation (RND) algorithm has been demonstrated to be effective in numerous environments, it often needs more discriminative power in bonus allocation. This paper highlights the "bonus inconsistency" issue within RND, pinpointing its primary limitation. To address this issue, we introduce the Distributional RND (DRND), a derivative of the RND. DRND enhances the exploration process by distilling a distribution of random networks and implicitly incorporating pseudo counts to improve the precision of bonus allocation. This refinement encourages agents to engage in more extensive exploration. Our method effectively mitigates the inconsistency issue without introducing significant computational overhead. Both theoretical analysis and experimental results demonstrate the superiority of our approach over the original RND algorithm. Our method excels in challenging online exploration scenarios and effectively serves as an anti-exploration mechanism in D4RL offline tasks. Our code is publicly available at this https URL
Submission history
From: Kai Yang [view email][v1] Thu, 18 Jan 2024 06:32:53 GMT (11350kb,D)
[v2] Thu, 25 Jan 2024 11:36:00 GMT (11350kb,D)
[v3] Fri, 17 May 2024 14:41:31 GMT (11352kb,D)
[v4] Mon, 20 May 2024 02:12:21 GMT (11352kb,D)
Link back to: arXiv, form interface, contact.