Prediction-Loss-Guided Long-Tail Augmentation and Uncertainty-Aware Reinforcement Learning for Car-Following Control

View at publisher
Authors: Hongrui Cao, Lei Cai, Ming Yue, Haiyun Bao, Hsin Guan
Cited by: 1
References: 25

Abstract

Reinforcement learning (RL) has demonstrated great potential for autonomous car-following control by directly optimizing key performance metrics, including safety, efficiency, and ride comfort. This study proposes an integrated RL-based car-following framework that combines trajectory prediction, reshaping of long-tail distributions, and uncertainty-aware decision-making. The trajectory prediction loss is redefined as a quantitative metric for scenario rarity and risk, facilitating a distribution-level data augmentation strategy that enriches safety-critical scenarios while preserving the naturalness of driving behavior. A probabilistic Gaussian LSTM predictor is further employed to model future vehicle motion uncertainty, which is explicitly incorporated into multi-step action evaluation through variance-weighted virtual rewards. Experiments on the highD dataset demonstrate that the proposed approach significantly improves robustness and safety, achieving substantial reductions in collision rates compared with baseline RL methods. Cross-dataset evaluation on NGSIM further confirms its generalization capability.

Fields of Study & Subjects

Vehicle Dynamics and Control Systems Reinforcement Learning in Robotics Traffic control and management

Publication details

Journal / source
IEEE Internet of Things Journal
Publisher
Institute of Electrical and Electronics Engineers (IEEE)
Volume
13
Issue
16
Pages
36930–36946
Publication date
8 June 2026