CORTEX: Token-Level Hallucination Detection in RAG via Comparative Internal Representations
Das Papier schlägt CORTEX vor, eine Methode zur Halluzinationserkennung auf Token-Ebene für Retrieval-Augmented Generation, die unbegründete Inhalte identifiziert, indem sie die internen Repräsentationen eines Modells mit und ohne abgerufene Dokumente vergleicht, während sie Informationspropagation und span-konsistente Glättung nutzt, um eine feingliedrige Lokalisierung von Halluzinationen zu erreichen.
Originalarbeit lizenziert unter CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). Dies ist eine KI-generierte Erklärung des untenstehenden Papers. Sie wurde nicht von den Autoren verfasst oder gebilligt. Für technische Genauigkeit konsultieren Sie das Originalpaper. Vollständigen Haftungsausschluss lesen
Stellen Sie sich vor, Sie haben einen sehr klugen, belesenen Assistenten (die KI), der versucht, eine Frage für Sie zu beantworten. Um sicherzustellen, dass die Antwort korrekt ist, geben Sie dem Assistenten einen Stapel Referenzbücher (die abgerufenen Dokumente), an denen er sich orientieren kann, während er schreibt.
Manchmal macht der Assistent einen großartigen Job und hält sich strikt an die Bücher. Ein andermal erfindet er versehentlich ein Faktum oder vermischt ein Detail, obwohl die Bücher direkt vor ihm auf dem Tisch liegen. Das nennt man eine „Halluzination“.
Das Problem ist: In einer langen, detaillierten Antwort schreibt die KI vielleicht 90 % der Fakten korrekt aus den Büchern, unterläuft ihr aber bei nur ein oder zwei Sätzen ein Fehler. Die meisten aktuellen Werkzeuge sind wie ein Sicherheitsmann, der erst am Ende den gesamten Bericht prüft. Wenn der Bericht größtenteils gut ist, sagt er vielleicht: „Sieht gut aus!“ und übersieht die winzige Lüge, die mitten im Text versteckt ist.
Dieses Paper stellt CORTEX vor, ein neues Werkzeug, das wie ein „Mikroskop“ konzipiert ist, um die Antwort Wort für Wort (oder Token für Token) zu prüfen, um genau zu erkennen, an welcher Stelle die KI Dinge erfindet.
So funktioniert CORTX, erklärt anhand von drei einfachen Analogien:
1. Der „Mit und Ohne“-Test (Vergleichende interne Repräsentationen)
Stellen Sie sich vor, Sie versuchen herauszufinden, ob ein Schüler tatsächlich ein Lehrbuch liest oder nur rät.
- Der alte Weg: Sie schauen sich die Antwort des Schülers an und versuchen zu erraten, ob er das Buch gelesen hat.
- Der CORTEX-Weg: Sie stellen dem Schüler dieselbe Frage zweimal.
- Das erste Mal: Sie geben ihm das Lehrbuch zum Lesen.
- Das zweite Mal: Sie nehmen das Lehrbuch weg und lassen ihn aus dem Gedächtnis antworten.
CORTEX vergleicht die „Gehirnaktivität“ (die internen Gedanken) der KI in diesen beiden Szenarien.
- Wenn die KI über einen Fakt spricht, der im Buch steht, verändert sich ihr „Gehirn“ signifikant, wenn das Buch präsent ist. Es ist, als würden die Augen des Schülers aufleuchten, weil er die Antwort im Text gefunden hat.
- Wenn die KI etwas erfindet (halluziniert), sieht ihr „Gehirn“ fast genauso aus, egal ob das Buch da ist oder nicht, weil sie einfach aus ihrem eigenen Gedächtnis schöpft.
Durch den Vergleich dieser beiden Zustände kann CORTEX genau erkennen, welche Wörter sich aufgrund des Buches geändert haben (gut) und welche Wörter gleich geblieben sind, weil sie erfunden wurden (schlecht).
2. Der „Chain of Thought“-Detektiv (Kontextueller Residualwert)
Manchmal ist eine KI klug genug, einen Fakt aus dem Buch am Anfang der Antwort zu lesen und diesen Fakt dann zu nutzen, um den Rest des Satzes später aufzubauen.
- Das Problem: Wenn man nur den späteren Satz betrachtet, könnte es so aussehen, als würde die KI das Buch nicht mehr verwenden, weil das Buch dort nicht direkt erwähnt wird. Dies könnte einen Detektor täuschen, der fälschlicherweise denkt, der spätere Satz sei eine Lüge, obwohl er auf einem wahren Fakt basiert, der zuvor gefunden wurde.
- Die CORTEX-Lösung: CORTEX fungiert wie ein Detektiv, der der Spur folgt. Es erkennt: „Hey, dieser Satz stützt sich auf einen Fakt, den die KI vor zwei Sätzen gelesen hat.“ Es subtrahiert diesen „indirekten Einfluss“, damit es nicht verwirrt wird. Dies verhindert, dass das Tool die KI fälschlicherweise der Lüge bezichtigt, wenn sie eigentlich nur eine wahre Geschichte fortführt.
3. Der „Rauschfilter“ (Label-Persistence Smoothing)
Stellen Sie sich vor, Sie hören einen Radiosender, der viel Rauschen hat. Manchmal lässt das Rauschen ein einzelnes Wort wie eine Lüge klingen, obwohl der ganze Satz wahr ist.
- Das Problem: Rohe Detektionswerkzeuge können „zittrig“ sein. Sie markieren vielleicht ein Wort als Lüge, das nächste als Wahrheit und das darauffolgende wieder als Lüge, was ein unordentliches, verstreutes Muster erzeugt. In der Realität ist es jedoch meist so, dass eine KI, wenn sie lügt, eine ganze Phrase oder einen ganzen Satz erfindet und nicht nur ein isoliertes Wort.
- Die CORTEX-Lösung: CORTEX wendet einen „Glättungsfilter“ an. Es geht davon aus, dass, wenn ein Wort eine Lüge ist, die unmittelbar benachbarten Wörter wahrscheinlich Teil derselben Lüge sind. Es verbindet die Punkte und verwandelt ein verstreutes Chaos aus „Vielleicht-Lügen“ in einen klaren, soliden Block von „definitiven Lügen“. Dies macht das Endergebnis für einen Menschen viel einfacher lesbar und verständlich.
Die Ergebnisse
Die Autoren haben CORTEX an zwei verschiedenen Fragen sets und drei verschiedenen KI-Modellen getestet. Sie fanden heraus, dass:
- CORTEX viel besser darin ist, die exakten Wörter zu finden, die erfunden wurden, im Vergleich zu anderen Methoden.
- Es selbst dann funktioniert, wenn die KI ein „API“-Modell ist (eine Black Box, in die man nicht hineinsehen kann), indem es ein anderes, offenes KI-Modell nutzt, um die Detektivarbeit im Hintergrund zu leisten.
- Jeder Teil des Systems (der Vergleich, die Chain-of-Thought-Prüfung und die Glättung) hat dazu beigetragen, die Detektion genauer zu machen.
Kurz gesagt: CORTEX ist ein Werkzeug, das uns hilft, KI-Antworten mehr zu vertrauen, indem es genau punktgenau aufzeigt, wo die KI Dinge erfindet, anstatt nur zu raten, ob die gesamte Antwort gut oder schlecht ist.
Ertrinken Sie in Arbeiten in Ihrem Fachgebiet?
Erhalten Sie tägliche Digests der neuesten Arbeiten passend zu Ihren Forschungsbegriffen — mit technischen Zusammenfassungen, in Ihrer Sprache.