Highway lane change decision-making via attention-based deep reinforcement learning

Code for "Highway lane change decision-making via attention-based deep reinforcement learning".

By Junjie Wang, Qichao Zhang, and Dongbin Zhao.

Citation

@article{wang2021highway,
  title={Highway lane change decision-making via attention-based deep reinforcement learning},
  author={Wang, Junjie and Zhang, Qichao and Zhao, Dongbin},
  journal={IEEE/CAA Journal of Automatica Sinica},
  volume={9},
  number={3},
  pages={567--569},
  year={2022},
  publisher={IEEE}
}

Acknowledgements

highway-env

ego-attention

non-local-block