Generating and Evolving Reward Functions for Highway Driving with Large Language Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Han, Xu, Yang, Qiannan, Chen, Xianda, Chu, Xiaowen, Zhu, Meixin
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929387213946880
author Han, Xu
Yang, Qiannan
Chen, Xianda
Chu, Xiaowen
Zhu, Meixin
author_facet Han, Xu
Yang, Qiannan
Chen, Xianda
Chu, Xiaowen
Zhu, Meixin
contents Reinforcement Learning (RL) plays a crucial role in advancing autonomous driving technologies by maximizing reward functions to achieve the optimal policy. However, crafting these reward functions has been a complex, manual process in many practices. To reduce this complexity, we introduce a novel framework that integrates Large Language Models (LLMs) with RL to improve reward function design in autonomous driving. This framework utilizes the coding capabilities of LLMs, proven in other areas, to generate and evolve reward functions for highway scenarios. The framework starts with instructing LLMs to create an initial reward function code based on the driving environment and task descriptions. This code is then refined through iterative cycles involving RL training and LLMs' reflection, which benefits from their ability to review and improve the output. We have also developed a specific prompt template to improve LLMs' understanding of complex driving simulations, ensuring the generation of effective and error-free code. Our experiments in a highway driving simulator across three traffic configurations show that our method surpasses expert handcrafted reward functions, achieving a 22% higher average success rate. This not only indicates safer driving but also suggests significant gains in development productivity.
format Preprint
id arxiv_https___arxiv_org_abs_2406_10540
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Generating and Evolving Reward Functions for Highway Driving with Large Language Models
Han, Xu
Yang, Qiannan
Chen, Xianda
Chu, Xiaowen
Zhu, Meixin
Artificial Intelligence
Neural and Evolutionary Computing
Robotics
Reinforcement Learning (RL) plays a crucial role in advancing autonomous driving technologies by maximizing reward functions to achieve the optimal policy. However, crafting these reward functions has been a complex, manual process in many practices. To reduce this complexity, we introduce a novel framework that integrates Large Language Models (LLMs) with RL to improve reward function design in autonomous driving. This framework utilizes the coding capabilities of LLMs, proven in other areas, to generate and evolve reward functions for highway scenarios. The framework starts with instructing LLMs to create an initial reward function code based on the driving environment and task descriptions. This code is then refined through iterative cycles involving RL training and LLMs' reflection, which benefits from their ability to review and improve the output. We have also developed a specific prompt template to improve LLMs' understanding of complex driving simulations, ensuring the generation of effective and error-free code. Our experiments in a highway driving simulator across three traffic configurations show that our method surpasses expert handcrafted reward functions, achieving a 22% higher average success rate. This not only indicates safer driving but also suggests significant gains in development productivity.
title Generating and Evolving Reward Functions for Highway Driving with Large Language Models
topic Artificial Intelligence
Neural and Evolutionary Computing
Robotics
url https://arxiv.org/abs/2406.10540