LLMs Are In-Context Bandit Reinforcement Learners

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Monea, Giovanni, Bosselut, Antoine, Brantley, Kianté, Artzi, Yoav
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918150130368512
author Monea, Giovanni
Bosselut, Antoine
Brantley, Kianté
Artzi, Yoav
author_facet Monea, Giovanni
Bosselut, Antoine
Brantley, Kianté
Artzi, Yoav
contents Large Language Models (LLMs) excel at in-context learning (ICL), a supervised learning technique that relies on adding annotated examples to the model context. We investigate a contextual bandit version of in-context reinforcement learning (ICRL), where models learn in-context, online, from external reward, instead of supervised data. We show that LLMs effectively demonstrate such learning, and provide a detailed study of the phenomena, experimenting with challenging classification tasks and models of sizes from 500M to 70B parameters. This includes identifying and addressing the instability of the process, demonstrating learning with both semantic and abstract labels, and showing scaling trends. Our findings highlight ICRL capabilities in LLMs, while also underscoring fundamental limitations in their implicit reasoning about errors.
format Preprint
id arxiv_https___arxiv_org_abs_2410_05362
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle LLMs Are In-Context Bandit Reinforcement Learners
Monea, Giovanni
Bosselut, Antoine
Brantley, Kianté
Artzi, Yoav
Computation and Language
Artificial Intelligence
Machine Learning
Large Language Models (LLMs) excel at in-context learning (ICL), a supervised learning technique that relies on adding annotated examples to the model context. We investigate a contextual bandit version of in-context reinforcement learning (ICRL), where models learn in-context, online, from external reward, instead of supervised data. We show that LLMs effectively demonstrate such learning, and provide a detailed study of the phenomena, experimenting with challenging classification tasks and models of sizes from 500M to 70B parameters. This includes identifying and addressing the instability of the process, demonstrating learning with both semantic and abstract labels, and showing scaling trends. Our findings highlight ICRL capabilities in LLMs, while also underscoring fundamental limitations in their implicit reasoning about errors.
title LLMs Are In-Context Bandit Reinforcement Learners
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2410.05362