CoEvo-Mem: Co-Evolving Retrieval Policy and Memory Bank for LLM Agents
CoEvo-Mem is een closed-loop framework dat gelijktijdig ophaalbeleid en de evolutie van de geheugenbank optimaliseert door middel van alternerende updates en trajectgeconditioneerde feedback, waarmee het de state-of-the-art prestaties bereikt over diverse benchmarks door de fundamentele onderlinge afhankelijkheid tussen geheugentoegang en verfijning in langetermijn LLM-agenten aan te pakken.
Oorspronkelijk artikel gelicentieerd onder CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). Dit is een AI-gegenereerde uitleg van het onderstaande artikel. Het is niet geschreven of goedgekeurd door de auteurs. Raadpleeg het oorspronkelijke artikel voor technische nauwkeurigheid. Lees de volledige disclaimer
Stel je voor dat je een briljante maar vergeetachtige robot leert om een complexe videogame te spelen. De robot heeft een enorm brein (een Large Language Model) dat weet hoe het moet praten en denken, maar het kan tegelijkertijd slechts een piepkleine hoeveelheid informatie vasthouden in zijn "werkgeheugen"—alsof je een puzzel probeert op te lossen terwijl je slechts de laatste drie stukjes ziet. Om op de lange termijn goed te kunnen spelen, heeft de robot een bibliotheek met ervaringen uit het verleden nodig om in op te zoeken. Dit is waar Retrieval-Augmented Generation (RAG) om de hoek komt kijken: het is alsof de robot de bibliothecaris vraagt: "Hé, hebben we ooit een puzzel gezien zoals deze?"
Maar hier komt de lastige kant aan: de robot heeft twee dingen nodig om perfect te functioneren. Ten eerste heeft hij een Bibliothecaris (het retrieval policy) die echt goed is in het vinden van de juiste oude aantekeningen voor de huidige vraag. Ten tweede heeft hij een Notitieschrijver (de geheugenbank) die goed is in het organiseren van die aantekeningen, het beslissen welke echt nuttig zijn en het bijwerken van de bank naarmate het spel moeilijker wordt. Lange tijd probeerden wetenschappers de Bibliothecaris slimmer te maken terwijl ze de Notitieschrijver statisch hielden, of ze maakten de Notitieschrijver beter terwijl ze de Bibliothecaris hetzelfde lieten. Maar wat als de Bibliothecaris moet weten hoe de aantekeningen zijn georganiseerd om ze te kunnen vinden, en de Notitieschrijver moet weten waar de Bibliothecaris naar op zoek is om ze beter te kunnen organiseren? Ze moeten samen leren, in een lus.
Dit is precies waar het papier CoEvo-Mem onderzoek naar doet. De auteurs stellen een systeem voor waarin de "Librarian" en de "Note-Taker" samen evolueren, waarbij ze elkaar voortdurend leren hoe ze hun taken beter kunnen uitvoeren. Ze hebben dit idee getest op zeven verschillende soorten uitdagende taken, variërend van het schrijven van computercode en het oplossen van wetenschappelijke vragen tot het besturen van een besturingssysteem van een computer. De resultaten suggereren dat wanneer deze twee onderdelen synchroon leren, de robot aanzienlijk beter wordt in het oplossen van lange, complexe problemen dan wanneer ze afzonderlijk leren.
Het Probleem: De Gebroken Feedbackloop
Stel je voor dat je studeert voor een enorme geschiedenisexamens. Je hebt een stapel flashcards (je geheugen) en een studiegids (je retrieval policy). Als je studiegids slecht is, kies je misschien de verkeerde flashcards om te herhalen. Maar als je flashcards rommelig en ongeorganiseerd zijn, zal zelfs een goede studiegids moeite hebben om de juiste te vinden.
In de wereld van AI-agenten proberen bestaande methoden meestal één kant van het probleem op te lossen terwijl ze de andere negeren. Sommige methoden proberen de AI beter te maken in het vragen om herinneringen (retrieval), terwijl andere de AI beter proberen te maken in het opslaan en organiseren van herinneringen (evolutie). De auteurs van dit papier stellen dat deze scheiding een fout is. Ze wijzen op een "fundamentele feedbackloop" die werd genegeerd:
- Retrieval bepaalt wat gebruikt wordt: Als de AI de verkeerde herinneringen kiest, krijgen die herinneringen geen "credits" voor het helpen (of schaden) van de taak.
- Geheugenupdates veranderen de toekomst: Als de AI zijn geheugens bijwerkt op basis van wat er is gebeurd, verandert de manier waarop die geheugens zijn georganiseerd, wat het de volgende keer makkelijker of juist moeilijker maakt om ze te vinden.
Als je alleen de retrieval-methode verbetert zonder het geheugen te laten veranderen, of andersom, mis je de kans dat ze samen beter worden. Het is alsof je een gitaar probeert af te stemmen terwijl de snaren constant van dikte veranderen; je kunt niet een perfect geluid krijgen tenzij je beide tegelijkertijd aanpast.
De Oplossing: CoEvo-Mem
De auteurs hebben een framework gebouwd genaamd CoEvo-Mem (Co-Evolving Memory). Denk aan een dans tussen twee partners: de Router (de Bibliothecaris) en de Memory Bank (de Notitieschrijver).
Zo werkt de dans:
1. Het Bevroren Brein en de Slimme Router
Het hoofd-"brein" van de AI (het Large Language Model) blijft bevroren—het leert niets nieuws. In plaats daarvan gebruikt het systeem een kleine, lichtgewicht "Router" om te beslissen hoe er om hulp wordt gevraagd.
- Wanneer er een vraag binnenkomt, stelt het bevroren brein een paar manieren voor om de vraag te herschrijven (sommigen richten zich op de betekenis, andere op specifieke trefwoorden).
- De Router leert vervolgens deze suggesties aan te passen op basis van de vraag of de AI het antwoord goed of fout had. Het is als een coach die fluistert: "Probeer het de volgende keer op deze manier," gebaseerd op de uitkomst van het spel.
2. De Geheugenbank als een Levende Kaart
De herinneringen zijn niet zomaar een lijst met aantekeningen; ze zijn een Relationele Graaf. Stel je een kaart voor waar elke herinnering een stad is, en de wegen tussen hen laten zien hoe ze met elkaar verbonden zijn.
- Dichte wegen verbinden herinneringen die hetzelfde betekenen (semantisch).
- Dunne wegen verbinden herinneringen die specifieke woorden of namen delen (lexicaal).
- Tijd-wegen verbinden herinneringen die in een sequentie plaatsvonden (temporeel).
Wanneer de AI een taak oplost, slaat hij niet alleen het antwoord op. Hij werkt de "waarde" van de steden die hij heeft bezocht bij. Als een herinnering de AI hielp winnen, krijgt die stad een hogere score. Als het hielp, maar slechts een beetje, gaat de score een klein beetje omhoog. Cruciaal is dat deze credit niet alleen bij die ene herinnering blijft; het reist langs de wegen naar naburige steden, waardoor de hele buurt slimmer wordt.
3. De Afwisselende Dans (Het Geheime Ingrediënt)
Je zou kunnen denken: "Waarom updaten we de Router en het Geheugen niet op exact hetzelfde moment?" De auteurs ontdekten dat dit chaos veroorzaakt. Als beide tegelijk veranderen, is het alsof je leert fietsen terwijl de fiets ook nog eens van vorm verandert. Het systeem raakt in de war en wordt instabiel.
In plaats daarvan gebruikt CoEvo-Mem een afwisselend schema:
- Fase 1: De Memory Bank is bevroren (vastgezet). De Router oefent met het vinden van de beste herinneringen met behulp van deze vaste kaart.
- Fase 2: De Router is bevroren (vastgezet). De Memory Bank werkt zijn organisatie en waarden bij op basis van wat de Router zojuist heeft gevonden.
- Ze wisselen elkaar voortdurend af. Dit stelt elke partner in staat om te leren van de huidige staat van de ander zonder de verwarring dat beide tegelijkertijd veranderen.
Wat Ze Hebben Gevonden
Het team heeft CoEvo-Mem getest op zeven zeer verschillende uitdagingen:
- Interactie met Besturingssystemen: Een computer besturen om taken uit te voeren.
- Code Generatie: Werkende computerprogramma's schrijven.
- Multimodale Redenering: Puzzels oplossen die zowel tekst als afbeeldingen bevatten.
- Wetenschappelijke Vragen: Moeilijke vragen beantwoorden over fysica, chemie en biologie.
- Function Calling: De AI leren om tools en API's correct te gebruiken.
- Langdurige Conversatie: Details onthouden uit lange chats.
In al deze tests presteerde CoEvo-Mem beter dan de beste bestaande methoden.
- Op de GPQA Diamond (moeilijke wetenschappelijke vragen) verbeterde het de score met 7,50 procentpunt ten opzichte van de sterkste baseline.
- Op LiveCodeBench (coderen) verbeterde het met 3,81 procentpunt.
- Op Long-term Conversational Memory (LoCoMo) bereikte het een score van 81,71, waarmee het de op één na beste methode met bijna 5 punten versloeg.
De auteurs voerden ook "ablatie-studies" uit (experimenten waarbij ze onderdelen van het systeem verwijderden) om te bewijzen waarom het werkte.
- Wanneer ze de query rewriting verwijderden (het deel dat verandert hoe de vraag wordt gesteld), daalde de prestatie aanzienlijk.
- Wanneer ze de memory evolution verwijderden (het deel dat de graaf en waarden bijwerkt), daalde de prestatie ook, vooral bij complexe taken zoals coderen en wetenschap.
- Wanneer ze probeerden zowel de Router als het Geheugen tegelijkertijd bij te werken (in plaats van afwisselend), presteerde het systeem slechter dan de afwisselende methode.
De Kernboodschap
Het artikel suggereert dat voor AI-agenten om echt meesterschap te bereiken in langdurige taken, we niet alleen een betere zoekmachine of een betere database kunnen bouwen. We moeten een systeem bouwen waarin de zoekmachine en de database leren om zich op elkaar aan te passen. Door de "Librarian" en de "Note-Taker" om de beurt te laten leren terwijl de ander stilstaat, creëert CoEvo-Mem een stabiele, verbeterende lus die helpt AI-agenten beter te onthouden, helderder te denken en moeilijkere problemen op te lossen.
De auteurs merken er voorzichtig bij op dat hoewel de resultaten sterk zijn over deze zeven benchmarks, dit een specifiek framework is voor co-evolutie. Ze beweren niet dat dit alle problemen in AI oplost, maar ze laten wel zien dat deze specifieke manier van het koppelen van retrieval en geheugen een krachtige stap voorwaarts is bij het bouwen van agenten die kunnen leren en zich aanpassen over een langere periode.
Verdrinkt u in papers in uw vakgebied?
Ontvang dagelijkse digests van de nieuwste papers die bij uw onderzoekswoorden passen — met technische samenvattingen, in uw taal.