RDQ: Residual Distribution Quantization for Large Language Models
Dieses Paper führt RDQ ein, ein Post-Training-Quantisierungs-Framework, das die superlineare Akkumulation von Quantisierungsrauschen in großen Sprachmodellen durch den Einsatz von kaskadierter Fehlerkompensation mildert, um Per-Channel-Skalierungen an driftende Residuenverteilungen anzupassen, wodurch eine State-of-the-Art-Perplexität bei 3-Bit- und 4-Bit-Präzision ohne Inferenz-Overhead erreicht wird.
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 versuchen, eine geheime Nachricht durch einen langen, gewundenen Tunnel zu senden, der aus 32 miteinander verbundenen Räumen besteht. In der ursprünglichen, hochwertigen Version dieses Tunnels (FP16) reist Ihre Nachricht perfekt klar vom Anfang bis zum Ende. Aber wenn wir versuchen, die Nachricht zu verkleinern, um sie in ein winziges, komprimiertes Format (Quantisierung) zu pressen, um Platz auf Ihrem Telefon oder Edge-Gerät zu sparen, geht einiges schief.
Lange Zeit dachten Wissenschaftler, das Problem liege einfach darin, dass jeder Raum für sich genommen ein wenig unordentlich sei. Sie dachten: „Wenn wir nur Raum 1 säubern, dann Raum 2, dann Raum 3, dann wird alles gut sein.“ Aber diese neue Arbeit, RDQ (Residual Distribution Quantization), hat entdeckt, dass dieser „jeden Raum separat säubern“-Ansatz eigentlich der falsche Weg ist, über das Problem nachzudenken.
Der wahre Übeltäter: Der „Schneeballeffekt“
Die Autoren fanden heraus, dass das eigentliche Problem ein Schneeballeffekt ist. Jedes Mal, wenn eine Nachricht durch einen Raum gleitet, wird ein winziges bisschen „Rauschen“ oder statisches Rauschen hinzugefügt. In einem normalen Tunnel spielt das keine Rolle. Aber in einem komprimierten Tunnel verschwindet dieses winzige Rauschen nicht; es wird in den nächsten Raum getragen, wo es sich mit dem neuen Rauschen aus diesem Raum vermischt.
Bis die Nachricht den 32. Raum erreicht, hat sich dieses winzige statische Rauschen in ein massives Brüllen verwandelt. Die Autoren haben dieses „Rauschakkumulation“ (das sie Residual Stream Drift nennen) gemessen und festgestellt: Die Menge des Rauschens sagt perfekt voraus, wie schlecht die Nachricht klingt. Tatsächlich fanden sie eine so starke Korrelation (Pearson r=0,999), dass man, wenn man weiß, wie stark das Signal abgedriftet ist, genau vorhersagen kann, wie verwirrt der Computer sein wird.
Die große Entdeckung: Der Tunnel ist nicht glatt
Vor dieser Arbeit gingen die meisten Experten davon aus, dass das Rauschen in diesen Tunneln wie ein glatter, vorhersehbarer Hügel aussieht (eine „Gaußsche“ Verteilung). Sie bauten ihre Reinigungswerkzeuge basestehend auf diesem glatten Hügel.
Aber die Autoren sind hineingegangen und haben nachgesehen. Sie entdeckten, dass 84 % der Räume im LLaMA-3-8B-Modell überhaupt keine glatten Hügel haben! Stattdessen ist das Rauschen zackig, seltsam und manchmal zweigipfelig (bimodal). Es ist, als würde man einen ruhigen See erwarten und einen stürmischen Ozean mit plötzlichen Wellen vorfinden. Weil das Rauschen so seltsam ist, versagten die alten „glatte Hügel“-Reinigungswerkzeuge kläglich, besonders wenn sie versuchten, die Nachricht auf nur 3 Bit (eine sehr winzige Größe) zu schrumpfen.
Die Lösung: „Cascaded Error Compensation“ (CEC)
Wie also repariert man einen Tunnel, in dem sich das Rauschen verändert, während man tiefer hineingeht? Die Autoren schlagen eine neue Methode vor, die Cascaded Error Compensation (CEC) genannt wird.
Anstatt jeden Raum zu reinigen, während man so tut, als wäre die Nachricht noch frisch und sauber (wie es alte Methoden taten), sagt CEC: „Lasst uns zuerst durch den Tunnel gehen, sehen, wie die Nachricht in jedem Raum tatsächlich aussieht, und dann reinigen.“
So funktioniert es:
- Der Rundgang: Der Computer lässt eine Testnachricht durch den Tunnel laufen. Er lässt die ersten Räume einfach „schmutzig“ werden (quantisiert), genau wie sie es im echten Leben tun würden.
- Der reale Anblick: Wenn die Nachricht in Raum 10 ankommt, ist sie bereits durch die Räume 1 bis 9 etwas verrauscht. Das System erfasst dieses tatsächliche, unordentliche Signal.
- Die maßgeschneiderte Korrektur: Anstatt ein generisches Reinigungswerkzeug zu verwenden, baut das System ein maßgeschneidertes Reinigungswerkzeug speziell für dieses unordentliche Signal. Es passt die „Lautstärke“ (Skalierungen) der Nachricht genau dort im Raum an.
- Die magische Faltung: Sobald das Reinigungswerkzeug gebaut ist, wird es in den Türrahmen des Raums gefaltet (die RMSNorm-Gewichte). Das bedeutet, wenn die echte Nachricht später durchreist, geschieht die Reinigung automatisch, mit null zusätzlichem Zeit- oder Geschwindigkeitsverlust.
Die Ergebnisse: Eine massive Verbesserung
Das Team testete dies an drei berühmten KI-Modellen: LLaMA-3-8B, Qwen-2.5-7B und Mistral-7B.
- Der alte Weg: Als sie versuchten, LLaMA-3-8B mit der Standardmethode (RTN) auf 3 Bit zu schrumpfen, wurde die Nachricht fast unverständlich, wobei der Wert (Perplexity) von 5,83 auf 14,09 sprang. Das ist ein enormer Qualitätsverlust.
- Der RDQ-Weg: Mit ihrer neuen „Rundgang“-Methode brachten sie den Wert auf 7,55. Das ist eine 46,4 % Verbesserung gegenüber der alten Methode!
- Bei 4 Bit: Sie erreichten auch die bisher besten jemals aufgezeichneten Ergebnisse und übertrafen die bisherigen Champions deutlich (z. B. 5,62 vs. 6,92 bei LLaMA).
Was das bedeutet
Die Autoren sind sich sehr sicher: Dies ist nicht nur ein besseres Werkzeug, sondern eine neue Art, das Problem zu betrachten. Sie haben bewiesen, dass der „Drift“ der Hauptgrund dafür ist, dass diese Modelle bei niedrigen Bit-Breiten versagen. Sie haben auch gezeigt, dass die alte Idee, „jeden Raum unabhängig voneinander zu reinigen“, eine Sackgasse ist, weil sie den Schneeball des Rauschens aus den vorherigen Räumen ignoriert.
Obwohl sie das Problem für eine 2-Bit-Kompression noch nicht lösen können (das Rauschen ist dort einfach zu stark), haben sie gezeigt, dass wir bei einer 3-Bit- und 4-Bit-Kompression die Modelle intelligent und schnell halten können. Das Beste daran? Diese neue Methode funktioniert mit den Standardwerkzeugen, die Ingenieure bereits verwenden, sodass sie ohne spezielle, langsame Hardware sofort auf Telefonen und Computern eingesetzt werden kann.
Kurz gesagt: Der Tunnel wurde unordentlich, weil wir die Räume in der falschen Reihenfolge gereinigt haben. Indem wir sie in der Reihenfolge reinigen, in der die Nachricht tatsächlich reist, können wir das Signal bis zum Ende klar halten.
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.