Training-Free Halving of Activated Experts in Fine-Grained Mixture-of-Experts Models
Dieses Paper schlägt eine trainingsfreie Methode vor, um die Rechenkosten von feingranularen Mixture-of-Experts-Modellen bei der Inferenz zu halbieren, indem es die Expertenauswahl von der Wahrscheinlichkeitsrenormierung entkoppelt und dadurch die Leistung bewahrt, indem es gegenüber einem größeren Referenzsatz an Experten statt gegenüber dem reduzierten aktiven Satz normalisiert.
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. Für technische Genauigkeit konsultieren Sie das Originalpaper. Vollständigen Haftungsausschluss lesen
Moderne Systeme der künstlichen Intelligenz, die Texte generieren, verlassen sich oft auf ein Design, das die Gesamtmenge des Wissens, das sie besitzen, von der Menge des Denkens trennt, das sie für jedes Wort aufwenden. Stellen Sie sich eine riesige Bibliothek mit tausenden spezialisierten Büchern vor, aber für jeden geschriebenen Satz öffnet das System nur eine Handvoll davon, um die richtigen Informationen zu finden. Dieser Ansatz, bekannt als Mixture-of-Experts-Modell, ermöglicht es Computern, enorme Mengen an Daten zu speichern, ohne jedes Mal langsamer zu werden, wenn sie sprechen. Das System nutzt einen kleinen internen Wegweiser, um zu entscheiden, welche spezifischen Bücher für jedes Wort geöffnet werden sollen, wodurch sichergestellt wird, dass nur ein winziger Bruchteil der gesamten Bibliothek in jedem Moment konsultiert wird. Dies schafft eine einzigartige Situation: Der Computer muss alle Bücher in seinen Regalen aufbewahren, um für jedes Thema bereit zu sein, aber er liest immer nur wenige Seiten zur Zeit.
Da der Computer all diese spezialisierten Abschnitte in seinem Speicher behalten muss, suchen Forscher schon lange nach Wegen, das System schneller zu machen, indem sie es dazu zwingen, noch weniger Seiten zu lesen. Die Standardmethode hierfür ist simpel: Sagen Sie dem System, es solle pro Satz weniger Bücher öffnen. Dieser direkte Ansatz hat jedoch historisch gesehen dazu geführt, dass das System verwirrt wurde und mehr Fehler machte. Die Verwirrung entsteht nicht deshalb, weil dem System die richtigen Bücher fehlen, sondern weil sich die Art und Weise, wie es die Informationen aus diesen Büchern gewichtet, unerwartet ändert, wenn die Anzahl der Bücher reduziert wird. Eine neue Studie von Xing Chen und Hengshuai Yao zeigt, dass diese Verwirrung durch eine versteckte mathematische Anpassung verursacht wird, die das System automatisch vornimmt, und sie fanden einen Weg, diese Anpassung auszuschalten, ohne das System neu trainieren oder neue Komponenten hinzufügen zu müssen.
Die Forscher konzentrierten sich auf zwei sehr große Sprachmodelle, eines mit 35 Milliarden Parametern und ein anderes mit fast 400 Milliarden. Diese Modelle sind mit Hunderten von winzigen spezialisierten Abschnitten, oder Experten, für jede Verarbeitungsschicht ausgestattet. In ihrem Standardbetrieb wählt das System acht dieser Experten für jedes verarbeitete Wort aus. Als die Forscher versuchten, das System zu beschleunigen, indem sie diese Anzahl halbierten und es zwangen, nur vier Experten zu verwenden, sank die Leistung erheblich. Bei einem Standardtest des Allgemeinwissens fiel die Genauigkeit um fast fünf Punkte. Dieser Verlust war so schwerwiegend, dass viele angenommen hatten, die Reduzierung der aktiven Experten sei schlichtweg zu kostspielig, um nützlich zu sein. Die Forscher vermuteten, dass das Problem nicht der Verlust der Experten selbst war, sondern vielmehr die Art und Weise, wie das System die Bedeutung der verbleibenden Experten neu kalibrierte.
Im Standardsetup passt das System den Einfluss der ausgewählten Experten so an, dass ihre kombinierte Bedeutung immer eine feste Gesamtsumme ergibt. Dies ist ein Sicherheitsmechanismus, um das System stabil zu halten. In diesen feingliedrigen Modellen halten die obersten acht Experten jedoch meist nur einen kleinen Bruchteil der möglichen Gesamtbedeutung. Wenn das System gezwungen wird, nur vier Experten zu wählen, multipliziert die Standardanpassung deren Einfluss mit einem großen Faktor, um die Differenz auszugleichen. Dieser plötzliche Schub drängt das System in einen Zustand, für den es nie trainiert wurde, was dazu führt, dass es überreagiert und Fehler macht. Die Forscher erkannten, dass das System nicht wirklich an Informationen mangelte; es wurde lediglich angewiesen, lauter zu schreien, als es eigentlich sollte.
Um dies zu beheben, führte das Team eine einfache Änderung ein, die kein Training und keine zusätzliche Rechenleistung erfordert. Sie trennten die Entscheidung, welche Experten verwendet werden sollen, von der Entscheidung, wie sie gewichtet werden sollen. Sie erlaubten dem System, nur vier Experten für die Arbeit auszuwählen, sagten ihm aber, die Gewichtung der Bedeutung basierend auf den obersten sechzehn Experten zu berechnen anstatt nur auf den obersten vier. Durch dies bewirkte das System, dass es das gleiche Volumen an Informationen behielt, das es gewohnt war zu hören, obwohl es nur die Hälfte der aktiven Abschnitte nutzte. Diese kleine Anpassung, die die Änderung eines einzigen Ganzzahl-Wertes beinhaltet, veränderte das Ergebnis komplett. Beim 35-Milliarden-Parameter-Modell verursachte die Reduzierung der aktiven Experten von acht auf vier mit dieser neuen Einstellung einen Rückgang der Genauigkeit von nur 0,3,,5 Punkten – ein Unterschied, der so gering war, dass er statistisch nicht von dem ursprünglichen, langsameren System zu unterscheiden war.
Die Studie testete diese Methode auch am viel größeren 400-Milliarden-Parameter-Modell, wobei die Ergebnisse ähnlich dramatisch ausfielen. Das Halbieren der aktiven Experten mit der Standardmethode verursachte einen signifikanten Leistungsverlust, aber die Verwendung der neuen Gewichtungsmethode reduzierte den Verlust auf nur 0,55 Punkte. Die Forscher fanden heraus, dass die beste Einstellung für diese neue Gewichtung von der spezifischen Aufgabe abhing. Beispielsweise führte eine Einstellung, die das System beim Lesen von Texten am besten klingen ließ, nicht immer dazu, dass es bei Logikrätseln oder Wissenstests am besten abschnitt. Diese Erkenntnis deutet darauf darauf hin, dass es nicht ausreicht, lediglich die Fähigkeit eines komprimierten Modells beim Lesen von Texten zu messen, um zu garantieren, dass es auch bei anderen Aufgaben gut funktionieren wird. Die Forscher kamen zu dem Schluss, dass man, um die Rechenleistung dieser Modelle sicher zu reduzieren, diese Gewichtungseinstellung sorgfältig anhand tatsächlicher Leistungstests abstimmen muss, anstatt sich allein auf die Textflüssigkeit zu verlassen.
Das Paper untersuchte auch, warum es schwierig ist, die zusätzlichen Experten einfach dauerhaft zu entfernen, anstatt sie nur vorübergehend seltener zu nutzen. Sie fanden heraus, dass diese Modelle hochspezialisiert sind, wobei verschiedene Experten unterschiedliche Arten von Inhalten wie Code oder allgemeine Texte behandeln. Da die Experten so spezialisiert und das System so gut ausbalanciert ist, gibt es kaum Spielraum, sie dauerhaft zu löschen, ohne die Fähigkeit zu verlieren, vielfältige Themen zu bearbeiten. Der effektivste Weg nach vorn besteht daher nicht darin, Teile des Modells wegzuwerfen, sondern es schlichtweg anzuweisen, zu einer Zeit weniger Teile zu verwenden, während das Gesamtvolumen der verarbeiteten Informationen konsistent bleibt. Dieser Ansatz bietet einen Weg, diese leistungsstarken Systeme schneller und effizienter zu machen, ohne ihre Intelligenz zu opfern, indem man lediglich einen versteckten Kalibrierungsfehler korrigiert, der auftritt, wenn sich die Regeln des Engagements ändern.
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.