Reward Design with Language Models
Auteurs : Minae Kwon, Sang Michael Xie, Kalesha Bullard, Dorsa Sadigh
Points clés générés par l'IA
- La conception de récompenses dans l'apprentissage par renforcement est difficile car spécifier les comportements souhaités peut être compliqué ou nécessiter beaucoup de démonstrations d'experts.
- L'utilisation d'un grand modèle langagier (LLM) tel que GPT-3 comme fonction de récompense proxy permet de simplifier la conception des récompenses à moindre coût en utilisant une interface de langage naturel.
- Les utilisateurs fournissent une prompt textuelle contenant quelques exemples (few-shot) ou une description (zero-shot) du comportement souhaité, et le LLM évalue le comportement d'un agent RL par rapport au comportement souhaité décrit par la prompt et produit un signal de récompense correspondant.
- Les agents RL formés avec cette méthode sont bien alignés sur les objectifs des utilisateurs dans différentes tâches, surpassant même les agents RL formés avec des fonctions de récompense apprises via l'apprentissage supervisé.
- Cette méthode est plus efficace en termes d'utilisation des données que les méthodes classiques d'apprentissage supervisé et peut produire différents styles d'agents alignés sur divers objectifs simplement en modifiant les exemples inclus dans la prompt.
- Cependant, il y a encore certaines limites à cette approche qui nécessitent plus d'études futures.
Résumé : Reward design in reinforcement learning (RL) is challenging since specifying human notions of desired behavior may be difficult via reward functions or require many expert demonstrations. Can we instead cheaply design rewards using a natural language interface? This paper explores how to simplify reward design by prompting a large language model (LLM) such as GPT-3 as a proxy reward function, where the user provides a textual prompt containing a few examples (few-shot) or a description (zero-shot) of the desired behavior. Our approach leverages this proxy reward function in an RL framework. Specifically, users specify a prompt once at the beginning of training. During training, the LLM evaluates an RL agent's behavior against the desired behavior described by the prompt and outputs a corresponding reward signal. The RL agent then uses this reward to update its behavior. We evaluate whether our approach can train agents aligned with user objectives in the Ultimatum Game, matrix games, and the DealOrNoDeal negotiation task. In all three tasks, we show that RL agents trained with our framework are well-aligned with the user's objectives and outperform RL agents trained with reward functions learned via supervised learning
Outils pour cet article
Posez des questions sur cet article à notre assistant IA
Vous pouvez aussi discutez avec plusieurs papiers à la fois ici.
Le résumé précédent a été créé il y a plus d'un an et peut être réexécuté (si nécessaire) en cliquant sur le bouton Exécuter ci-dessous.
Articles similaires résumés avec nos outils d'IA
Naviguez à travers encore plus d'articles similaires en utilisant une
représentation arborescenteRecherchez des articles similaires (en version bêta)
En cliquant sur le bouton ci-dessus, notre algorithme compare le titre et le résumé de cet article à tous les articles de la même catégorie arXiv que nous avons résumés, et trouve les plus proches. Vous pouvez le relancer de temps en temps : les résultats s'améliorent à mesure que notre base grandit.
Avertissement : Notre outil de synthèse basé sur l'IA et l'assistant virtuel fournis sur ce site Web peuvent ne pas toujours fournir des résumés complets ou des réponses exactes. Nous vous encourageons à examiner attentivement et à évaluer le contenu généré pour vous assurer de sa qualité et de sa pertinence par rapport à vos besoins.