CORTEX: Token-Level Hallucination Detection in RAG via Comparative Internal Representations
Het artikel stelt CORTEX voor, een methode voor hallucinatie-detectie op tokenniveau voor Retrieval-Augmented Generation die ongegronde inhoud identificeert door de interne representaties van een model met en zonder opgehaalde documenten te vergelijken, terwijl het gebruikmaakt van informatiepropagatie en span-consistente smoothing om fijnmazige lokalisatie van hallucinaties te bereiken.
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 zeer slimme, goed onderlegde assistent hebt (de AI) die probeert een vraag voor je te beantwoorden. Om ervoor te zorgen dat het antwoord nauwkeurig is, geef je de assistent een stapel referentieboeken (de opgehaalde documenten) om naar te kijken terwijl hij het antwoord schrijft.
Soms doet de assistent een geweldig werk en houdt hij zich strikt aan de boeken. Andere keren verzint hij per ongeluk een feit of verwisselt hij een detail, ook al liggen de boeken gewoon op het bureau. Dit wordt een "hallucinatie" genoemd.
Het probleem is dat in een lang, gedetailleerd antwoord de AI misschien 90% van de feiten uit de boeken goed heeft, maar slechts één of twee zinnen laat verslapen. De meeste huidige tools zijn als een bewaker die alleen het volledige rapport aan het einde controleert. Als het rapport grotendeels goed is, zegt de bewaker: "Ziet er goed uit!" en mist hij de kleine leugen die halverwege verborgen zit.
Dit artikel introduceert CORTEX, een nieuwe tool die is ontworpen als een "microscoop" die het antwoord woord voor woord (of token voor token) controleert om precies te ontdekken waar de AI dingen verzint.
Zo werkt CORTEX, met behulp van drie eenvoudige analogieën:
1. De "Met en Zonder" Test (Vergelijkbare Interne Representaties)
Stel je voor dat je probeert uit te zoeken of een student echt een tekstboek aan het lezen is of dat hij gewoon gokt.
- De Oude Manier: Je kijkt naar het antwoord van de student en probeert te raden of hij het boek heeft gelezen.
- De CORTEX-Manier: Je vraagt de student om dezelfde vraag twee keer te beantwoorden.
- Eerste keer: Je geeft hem het tekstboek om te lezen.
- Tweede keer: Je neemt het tekstboek weg en vraagt hem om vanuit zijn geheugen te antwoorden.
CORTEX vergelijkt de "hersenactiviteit" (interne gedachten) van de AI tijdens deze twee scenario's.
- Als de AI praat over een feit dat in het boek staat, verandert zijn "brein" aanzienlijk wanneer het boek aanwezig is. Het is alsof de ogen van de student oplichten omdat hij het antwoord in de tekst heeft gevonden.
- Als de AI iets verzint (hallucineert), ziet zijn "brein" er bijna hetzelfde uit of het boek er nu wel of niet is, omdat hij simpelweg put uit zijn eigen geheugen.
Door deze twee toestanden te vergelijken, kan CORTEX precies spotten welke woorden veranderden door het boek (goed) en welke woorden hetzelfde bleven omdat ze verzonnen zijn (slecht).
2. De "Chain of Thought" Detective (Contextuele Residu)
Soms is een AI slim genoeg om een feit uit het boek aan het begin van het antwoord te lezen, en dat feit vervolgens te gebruiken om de rest van de zin op te bouwen.
- Het Probleen: Als je alleen naar de latere zin kijkt, lijkt het alsof de AI het boek niet meer gebruikt, omdat het boek daar niet direct wordt vermeld. Dit zou een detector kunnen misleiden om te denken dat de latere zin een leugen is, terwijl deze wel gebaseerd is op een waar feit dat eerder werd gevonden.
- De CORTEX-Oplossing: CORTEX werkt als een detective die het spoor volgt. Het realiseert zich: "Hé, deze zin leunt op een feit dat de AI net twee zinnen geleden heeft gelezen." Het trekt die "indirecte invloed" af zodat het niet in de war raakt. Dit voorkomt dat de tool de AI onterecht beschuldigt van liegen wanneer de AI eigenlijk gewoon een waar verhaal voortzet.
3. De "Ruisfilter" (Label-Persistentie Smoothing)
Stel je voor dat je luistert naar een radiostation met veel statische ruis. Soms zorgt de statische ruis ervoor dat een enkel woord als een leugen klinkt, zelfs als de hele zin waar is.
- Het Probleem: Ruwe detectietools kunnen schokkerig zijn. Ze kunnen één woord markeren als een leugen, het volgende woord als waarheid, en het volgende weer als een leugen, wat een rommelig, verspreid patroon creëert. Maar in werkelijkheid, als een AI liegt, liegt hij meestal voor een hele frase of zin, en niet slechts voor één geïsoleerd woord.
- De CORTEX-Oplossing: CORTEX past een "smoothing" filter toe. Het gaat ervan uit dat als één woord een leugen is, de woorden direct ernaast waarschijnlijk deel uitmaken van dezelfde leugen. Het verbindt de punten en verandert een verspreide chaos van "misschien-leugens" in een duidelijke, solide blok van "duidelijke-leugens". Dit maakt het eindresultaat veel gemakkelijker voor een mens om te lezen en te begrijpen.
De Resultaten
De auteurs hebben CORTEX getest op twee verschillende sets vragen en drie verschillende AI-modellen. Ze kwamen tot de volgende bevindingen:
- CORTEX is veel beter in het vinden van de exacte woorden die verzonnen zijn vergeleken met andere methoden.
- Het werkt zelfs wanneer de AI een "API"-model is (een black box waar je niet in kunt kijken) door een ander, open AI-model te gebruiken om de detectie op de achtergrond uit te voeren.
- Elk onderdeel van het systeem (de vergelijking, de chain-of-thought check en de smoothing) heeft bijgedragen aan een nauwkeurigere detectie.
Kortom, CORTEX is een tool die ons helpt de antwoorden van AI meer te vertrouwen door precies aan te wijzen waar de AI dingen verzint, in plaats van alleen maar te gokken of het hele antwoord goed of slecht is.
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.