Agent-R: Het trainen van taalmodelagenten om te reflecteren via iteratieve zelftraining.Agent-R: Training Language Model Agents to Reflect via Iterative
Self-Training
Grote Taalmodellen (LLM's) zijn steeds belangrijker voor het aanpakken van complexe taken in interactieve omgevingen. Bestaand werk richt zich voornamelijk op het verbeteren van de prestaties door gedrag te klonen van sterkere experts, maar dergelijke benaderingen falen vaak in real-world toepassingen, voornamelijk vanwege het onvermogen om te herstellen van fouten. Echter, stapsgewijze kritiekdata is moeilijk en duur om te verzamelen. Het automatiseren en dynamisch construeren van zelfkritiekdatasets is daarom cruciaal om modellen te voorzien van intelligente agentcapaciteiten. In dit werk stellen we een iteratief zelftrainingskader voor, Agent-R, dat taalagenten in staat stelt om ter plekke te Reflecteren. In tegenstelling tot traditionele methoden die acties belonen of bestraffen op basis van juistheid, maakt Agent-R gebruik van MCTS om trainingsdata te construeren die correcte trajecten herstellen van foutieve trajecten. Een belangrijke uitdaging van agentreflectie ligt in de noodzaak voor tijdige herziening in plaats van te wachten tot het einde van een rollout. Om dit aan te pakken, introduceren we een modelgestuurd kritiekconstructiemechanisme: het actiemodel identificeert de eerste foutstap (binnen zijn huidige capaciteit) in een mislukt traject. Van daaruit splitsen we het met het aangrenzende juiste pad, dat dezelfde ouderknoop in de boom deelt. Deze strategie stelt het model in staat om reflectie te leren op basis van zijn huidige beleid, wat resulteert in een betere leerefficiëntie. Om verder de schaalbaarheid van dit zelfverbeteringsparadigma te verkennen, onderzoeken we iteratieve verfijning van zowel foutcorrectiecapaciteiten als datasetconstructie. Onze bevindingen tonen aan dat Agent-R continu de mogelijkheid van het model verbetert om te herstellen van fouten en tijdige foutcorrectie mogelijk maakt. Experimenten in drie interactieve omgevingen tonen aan dat Agent-R agenten effectief uitrust om foutieve acties te corrigeren en lussen te vermijden, met superieure prestaties in vergelijking met basismethoden (+5.59%).