Human as AI Mentor: Enhanced Human-in-the-loop Reinforcement Learning for Safe and Efficient Autonomous Driving

University of Wisconsin-Madison
*Corresponding Author

After almost 65 minutes (30K steps) of training, the AV agent learns how to drive safely and efficiency.

Abstract

Despite significant progress in autonomous vehicles (AVs), the development of driving policies that ensure both the safety of AVs and traffic flow efficiency has not yet been fully explored. In this paper, we propose an enhanced human-in-the-loop reinforcement learning method, termed the Human as AI mentor-based deep reinforcement learning (HAIM-DRL) framework, which facilitates safe and efficient autonomous driving in mixed traffic platoon. Drawing inspiration from the human learning process, we first introduce an innovative learning paradigm that effectively injects human intelligence into AI, termed Human as AI mentor (HAIM). In this paradigm, the human expert serves as a mentor to the AI agent. While allowing the agent to sufficiently explore uncertain environments, the human expert can take control in dangerous situations and demonstrate correct actions to avoid potential accidents. On the other hand, the agent could be guided to minimize traffic flow disturbance, thereby optimizing traffic flow efficiency. In detail, HAIM-DRL leverages data collected from free exploration and partial human demonstrations as its two training sources. Remarkably, we circumvent the intricate process of manually designing reward functions; instead, we directly derive proxy state-action values from partial human demonstrations to guide the agents' policy learning. Additionally, we employ a minimal intervention technique to reduce the human mentor's cognitive load. Comparative results show that HAIM-DRL outperforms traditional methods in driving safety, sampling efficiency, mitigation of traffic flow disturbance, and generalizability to unseen traffic scenarios.

Overview


Descriptive Alt Text

Human as AI Mentor Paradigm


Descriptive Alt Text

Explicit and Implicit Intervention Mechanism


Descriptive Alt Text

HAIM-DRL Framework


Descriptive Alt Text

Experiment Results


Descriptive Alt Text

Visualization of Comparison Results


Demonstration Video for Each Model

BibTeX

@article{huang2024human,
  title={Human as AI Mentor: Enhanced Human-in-the-loop Reinforcement Learning for Safe and Efficient Autonomous Driving},
  author={Huang, Zilin and Sheng, Zihao and Ma, Chengyuan and Chen, Sikai},
  journal={arXiv preprint arXiv:2401.03160},
  year={2024}
}