SPformer: A Transformer Based DRL Decision Making Method for Connected Automated Vehicles

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Han, Ye, Zhang, Lijun, Meng, Dejian, Hu, Xingyu, Lu, Yixia
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916859714994176
author Han, Ye
Zhang, Lijun
Meng, Dejian
Hu, Xingyu
Lu, Yixia
author_facet Han, Ye
Zhang, Lijun
Meng, Dejian
Hu, Xingyu
Lu, Yixia
contents In mixed autonomy traffic environment, every decision made by an autonomous-driving car may have a great impact on the transportation system. Because of the complex interaction between vehicles, it is challenging to make decisions that can ensure both high traffic efficiency and safety now and futher. Connected automated vehicles (CAVs) have great potential to improve the quality of decision-making in this continuous, highly dynamic and interactive environment because of their stronger sensing and communicating ability. For multi-vehicle collaborative decision-making algorithms based on deep reinforcement learning (DRL), we need to represent the interactions between vehicles to obtain interactive features. The representation in this aspect directly affects the learning efficiency and the quality of the learned policy. To this end, we propose a CAV decision-making architecture based on transformer and reinforcement learning algorithms. A learnable policy token is used as the learning medium of the multi-vehicle joint policy, the states of all vehicles in the area of interest can be adaptively noticed in order to extract interactive features among agents. We also design an intuitive physical positional encodings, the redundant location information of which optimizes the performance of the network. Simulations show that our model can make good use of all the state information of vehicles in traffic scenario, so as to obtain high-quality driving decisions that meet efficiency and safety objectives. The comparison shows that our method significantly improves existing DRL-based multi-vehicle cooperative decision-making algorithms.
format Preprint
id arxiv_https___arxiv_org_abs_2409_15105
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle SPformer: A Transformer Based DRL Decision Making Method for Connected Automated Vehicles
Han, Ye
Zhang, Lijun
Meng, Dejian
Hu, Xingyu
Lu, Yixia
Artificial Intelligence
Multiagent Systems
Systems and Control
In mixed autonomy traffic environment, every decision made by an autonomous-driving car may have a great impact on the transportation system. Because of the complex interaction between vehicles, it is challenging to make decisions that can ensure both high traffic efficiency and safety now and futher. Connected automated vehicles (CAVs) have great potential to improve the quality of decision-making in this continuous, highly dynamic and interactive environment because of their stronger sensing and communicating ability. For multi-vehicle collaborative decision-making algorithms based on deep reinforcement learning (DRL), we need to represent the interactions between vehicles to obtain interactive features. The representation in this aspect directly affects the learning efficiency and the quality of the learned policy. To this end, we propose a CAV decision-making architecture based on transformer and reinforcement learning algorithms. A learnable policy token is used as the learning medium of the multi-vehicle joint policy, the states of all vehicles in the area of interest can be adaptively noticed in order to extract interactive features among agents. We also design an intuitive physical positional encodings, the redundant location information of which optimizes the performance of the network. Simulations show that our model can make good use of all the state information of vehicles in traffic scenario, so as to obtain high-quality driving decisions that meet efficiency and safety objectives. The comparison shows that our method significantly improves existing DRL-based multi-vehicle cooperative decision-making algorithms.
title SPformer: A Transformer Based DRL Decision Making Method for Connected Automated Vehicles
topic Artificial Intelligence
Multiagent Systems
Systems and Control
url https://arxiv.org/abs/2409.15105