Neuro-Cognitive Reward Modeling for Human-Centered Autonomous Vehicle Control
Organizations: University of Technology Sydney
Abstract
Recent advancements in computer vision have accelerated the development of autonomous driving. Despite these advancements, training machines to drive in a way that aligns with human expectations remains a significant challenge. Human factors are still essential, as humans possess a sophisticated cognitive system capable of rapidly interpreting scene information and making accurate decisions. Aligning machine with human intent has been explored with Reinforcement Learning with Human Feedback (RLHF). Conventional RLHF methods rely on collecting human preference data by manually ranking generated outputs, which is time-consuming and indirect. In this work, we propose an electroencephalography (EEG)-guided decision-making framework to incorporate human cognitive insights without behaviour response interruption into reinforcement learning (RL) for autonomous driving. We collected EEG signals from 20 participants in a realistic driving simulator and analyzed event-related potentials (ERP) in response to sudden environmental changes. Our proposed framework employs a neural network to predict the strength of ERP based on the cognitive information from visual scene information. Moreover, we explore the integration of such cognitive information into the reward signal of the RL algorithm. Experimental results show that our framework can improve the collision avoidance ability of the RL algorithm, highlighting the potential of neuro-cognitive feedback in enhancing autonomous driving systems. Our project page is: https://alex95gogo.github.io/Cognitive-Reward/.
Figures & tables
| Dataset | Active ctlr. | Vehicle data | Camera | Hazards | View | #subjects | # Frames | EEG | Gaze |
|---|---|---|---|---|---|---|---|---|---|
| Ours | + | + | + | VR | 20 | 720K | + | + | |
| Collision Threat [ 38 ] | + | - | - | - | screen | 25 | - | + | - |
| Sustained-attention [ 6 ] | + | + | - | + | screen | 27 | - | + | - |
| MPDB [ 51 ] | + | - | - | + | screen | 35 | - | + | + |
| DrFixD(night) [ 12 ] | - | - | - | screen | 31 | 67K* | - | + | |
| LBW [ 31 ] | - | - | - | screen | 28 | 123K* | - | + |
| Mean | FPS | ||||||
|---|---|---|---|---|---|---|---|
| Resnet-18 | 82 | 89 | 79 | 79 | 76 | 81 | 73 |
| Swin-ViT | 80 | 85 | 75 | 80 | 77 | 79 | 62 |
| ConvNeXt | 82 | 89 | 75 | 81 | 76 | 80 | 74 |
| Ours | 80 | 85 | 77 | 86 | 81 | 82 | 204 |
| Emergency Braking | Left Turn | |||||
|---|---|---|---|---|---|---|
| Route Completion | Driving Score | Infraction | Route Completion | Driving Score | Infraction | |
| Vanilla | ||||||
| BC | ||||||
| PHIL | ||||||
| RLHF | ||||||
| TD3-lag | ||||||
RL for Autonomous Driving
Reinforcement learning for autonomous driving trains vehicle policies through interaction to plan, control, and coordinate safely in traffic.
See all 120 papers