Where Knowledge Collides: A Mechanistic Study of Intra-Memory Knowledge Conflict in Language Models
Dit artikel presenteert een mechanistische studie naar intra-geheugen kennisconflicten in taalmodellen, waarbij wordt onthuld dat dergelijke conflicten doorgaans in de laatste lagen worden opgelost via afzonderlijke circuits in plaats van via een universeel mechanisme, en dat interventies aanzienlijk minder effectief zijn bij real-world conflicten vergeleken met synthetische conflicten vanwege verschillen in de distributie van gespecialiseerde attention heads.
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 een gigantisch, digitaal brein voor dat bijna alles op het internet heeft gelezen. Dit brein, bekend als een Large Language Model (LLM), is als een superintelligente bibliothecaris die vragen kan beantwoorden, verhalen kan schrijven en problemen kan oplossen. Maar hier is de crux: soms raakt deze bibliothecaris in de war. Als je vraagt: "Waar ging Niels naar school?", kan de bibliothecaris tegelijkertijd twee verschillende antwoorden onthouden. Misschien zegt een deel van zijn geheugen "Universiteit van A", terwijl een ander deel volhoudt "Universiteit van B". Dit komt niet doordat de bibliothecaris naar een nieuw boek kijkt (externe informatie); het komt doordat de interne aantekeningen van de bibliothecaris zelf een rommeltje zijn. Wetenschappers noemen dit "intra-memory knowledge conflict" (kennisconflict binnen het geheugen). Het is een beetje alsof je twee verschillende versies van hetzelfde verhaal hebt geschreven in hetzelfde schrift, en de bibliothecaris niet weet welke hij hardop moet voorlezen. Dit is een groot probleem, want als deze AI-systemen vertrouwd moeten worden met taken in de echte wereld, moeten ze stoppen met hallucineren of tegenstrijdige antwoorden geven. Onderzoekers willen weten: waar in het digitale brein vindt deze verwarring precies plaats, en kunnen we het specifieke deel dat de verwarring veroorzaakt repareren?
Dit artikel, getiteld "Where Knowledge Collides", duikt diep in de binnenwereld van deze AI-breinen om de bron van de verwarring te vinden. De auteurs bouwden een speciale testomgeving met behulp van vier verschillende AI-modellen. Ze creëerden een dataset van fictieve biografieën — denk aan nep-Wikipedia-pagina's over mensen die niet bestaan — waar ze doelbewust tegenstrijdige feiten in hebben geplant. Bijvoorbeeld, ze vertelden de AI dat een fictieve persoon genaamd "Niels Cavalli" afstudeerde aan de "Universiteit van Ukopnwm", maar vertelden de AI daarnaast dat dezelfde Niels afstudeerde aan de "Universiteit van Ohfgrgr". Door de AI te trainen op deze door elkaar gehusselde verhalen, creëerden ze een gecontroleerde omgeving waarin het model moest twee tegenstrijdige feiten tegelijkertijd in zijn hoofd te houden.
Om te begrijpen wat er aan de hand was, gebruikten de onderzoekers een techniek genaand "mechanistic interpretability" (mechanistische interpreteerbaarheid). Stel je de AI voor als een fabriek met meerdere verdiepingen waar informatie van de onderste naar de bovenste verdieping stroomt. Elke verdieping heeft werkers (genaamd "layers" of lagen) en specifieke teams binnen die verdiepingen (genaamd "attention heads" of aandachtskoppen) die verschillende delen van de klus afhandelen. De onderzoekers gebruikten een tool genaamd een "logit lens" om in elke verdieping van de fabriek in te kijken en te zien hoe het vertrouwen van de AI veranderde terwijl het de vraag verwerkte. Ze ontdekten iets zeer consistente: de verwarring gebeurde niet aan het begin of in het midden van de fabriek. In plaats daarvan barstte het conflict altijd los in de laatste paar lagen. Het is also en de AI het grootste deel van de tijd doorbrengt met het verzamelen van feiten, maar vlak aan het einde, net voordat hij spreekt, botsen de twee tegenstrijdige herinneringen op elkaar, waardoor het model wankelt en onzeker wordt.
Het team probeerde het probleem vervolgens op te lossen door de AI te "patchen". Dit is alsof je in de fabriek reikt en het werk van een specifiek team vervangt door het werk van een team dat het juiste antwoord weet. Ze probeerden twee hoofdbenaderingen: het repareren van een hele verdieping (layer-wise intervention) of het repareren van slechts één specifiek team (head-level intervention). De resultaten waren duidelijk: het repareren van de hele verdieping was oké, maar het repareren van het specifieke team was veel, veel beter. Sterker nog, door specifiek de "attention heads" aan te pakken die verantwoordelijk waren voor het foute feit, konden ze de AI met grote successen richting het juiste antwoord sturen. Ze ontdekten zelfs dat er in hun fictieve, gecontroleerde wereld vaak "specialistische" heads waren die alleen om één specifiek feit gaven. Echter, toen ze dit op echte gegevens probeerden (met een dataset genaamd DynamicQA met werkelijke, rommelige menselijke kennis), waren de reparaties minder effectief. Dit suggereert dat conflicten in de echte wereld verstrengelder en moeilijker te ontwarren zijn dan de schone, synthetische conflicten die ze in het lab creëerden.
Een van de meest interessante zaken waar het artikel juist tegenin gaat, is het idee van een "universeel circuit". Je zou kunnen denken dat alle AI-modellen één specifieke, toegewezen schakelaar of pad hebben dat alle kennisconflicten afhandelt, zoals een "conflictresolutieknop". De auteurs suggereren dat dit waarschijnlijk niet waar is. In plaats daarvan wijzen hun bevindingen op het idee dat verschillende stukjes kennis in verschillende, aparte circuits zijn opgeslagen. Het uiteindelijke antwoord dat de AI geeft, hangt af van welk van deze concurrerende circuits het sterkste signaal krijgt van jouw vraag. Er is niet één enkele "conflictmanager"; het is meer een touwtrekken tussen verschillende groepen werkers, en degene die het hardst trekt, wint.
De onderzoekers merkten ook op dat een van de modellen, GPT-2 XL, moeilijker te repareren was dan de andere. Ze concludeerden dat dit niet kwam omdat het model kapot was, maar simpelweg omdat het groter was. Het had veel meer "werkers" (attention heads), waardoor ze een groter aantal van hen moesten repareren om dezelfde resultaten te zien als bij de kleinere modellen. Het is alsof je een verkeersopstopping probeert op te lossen in een klein dorpje versus een enorme stad; je moet meer auto's in de stad wegwerken om dezelfde doorstroming te krijgen.
Uiteindelijk geeft deze studie ons de eerste echte kaart van waar deze interne conflicten plaatsvinden. Het vertelt ons dat de strijd tussen conflicterende feiten plaatsvindt vlak bij de finishlijn van het denkproces van de AI. Hoewel we deze conflicten in de rommelige echte wereld nog niet perfect kunnen oplossen, suggereert het artikel dat we, door de specifieke, kleine teams aan te pakken die verantwoordelijk zijn voor de foute feiten, deze AI-systemen betrouwbaarder kunnen maken. Het is een cruciale stap naar het bouwen van AI die niet alleen slim klinkt, maar ook echt weet waar het over praat.
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.