Bridging Compute- and Data-Optimal Pretraining
Dieses Paper führt die Compute-Data (CD) Scaling Laws ein, ein einheitliches Framework, das die abnehmenden Erträge abgeleiteter Daten durch eine Token-Effektivitätsfunktion modelliert, um die compute- und datenoptimalen Pretraining-Regime zu überbrücken, wobei aufgezeigt wird, dass die klassische compute-optimale Allokation für die meisten praktischen Szenarien suboptimal ist.
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, einem riesigen, superintelligenten Roboter beizubringen, die menschliche Sprache zu sprechen. Jahrelang glaubten Wissenschaftler, das Geheimrezept sei simpel: Füttern Sie den Roboter einfach immer mehr mit neuen Geschichten, Fakten und Büchern. Je größer das Gehirn des Roboters (seine „Modellgröße“) und je mehr frische Bücher Sie ihm gaben, desto klüger würde er werden. Diese Idee, bekannt als „Scaling Laws“ (Skalierungsgesetze), legte nahe, dass man mit genügend Rechenleistung einfach mehr Daten kaufen konnte und der Roboter dadurch immer besser werden würde.
Aber es gibt einen Haken. Wir gehen die frischen, hochwertigen Bücher aus. Das Internet ist endlich, und das Schreiben neuer, perfekter Geschichten kostet Zeit und Geld. Währenddessen werden unsere Computer jedoch immer schneller und billiger. Dies erzeugt ein seltsames Problem: Wir haben einen superstarken Motor (Rechenleistung/Compute), aber einen fast leeren Tank (Daten). Die große Frage für Wissenschaftler lautete: Wenn wir keine neuen frischen Bücher mehr bekommen können, können wir den Roboter dann einfach dieselben Bücher immer wieder lesen lassen oder sie auf verschiedene Arten umschreiben, um ihn klüger zu machen? Hilft es dem Roboter genauso viel, eine Geschichte ein zweites Mal zu lesen, wie eine brandneue zu lesen?
Dies ist das Rätsel, das die Forscher Tian Qin, Kimia Hamidieh und David Alvarez-Melis in ihrem neuen Paper angegangen sind. Sie wollten das perfekte Rezept für das Training dieser KI-Modelle finden, wenn frische Daten knapp werden. Sie entwickelten einen neuen Satz von Regeln namens „Compute-Data (CD) Scaling Laws“. Denken Sie an eine neue Karte für eine Reise, auf der die alte Karte sagte: „Fahren Sie einfach geradeaus“, während die neue Karte sagt: „Hey, die Straße endet! Hier ist, wie man die Umwege navigiert.“
Das Team fand heraus, dass das bloße wiederholte Lesen alter Daten kein magischer Fix ist. Sie entdeckten, dass „abgeleitete Token“ – was so etwas wie das mehrmalige Lesen eines Buches oder das Umschreiben eines Satzes in den eigenen Worten ist – weniger wert sind als frischer, origineller Text. Je mehr man wiederholt oder paraphrasiert, desto weniger Wert fügt jedes neue Wort hinzu. Es ist wie beim Essen eines köstlichen Kuchens: Das erste Stück ist fantastisch, das zweite ist gut, aber beim zehnten Stück ist man einfach satt und genießt es nicht mehr wirklich.
Um dies zu beweisen, trainierten sie Dutzende von KI-Modellen, die von winzigen Modellen mit 14 Millionen „Neuronen“ bis hin zu mittelgroßen mit 600 Millionen reichen, unter Verwendung einer massiven Textbibliothek namens Dolma-3. Sie testeten zwei Hauptstrategien: Multi-Epoch-Repetition (das mehrfache Lesen desselben Buches) und Paraphrasierung (das Umschreiben des Buches in verschiedene Stile, wie etwa einen Nachrichtenartikel in ein mathematisches Problem oder eine FAQ zu verwandeln).
Ihre Ergebnisse waren überraschend und sehr spezifisch. Sie fanden heraus, dass der „Wert“ des wiederholten Lesens oder Umschreibens stark davon abhängt, wie groß das Gehirn des Roboters ist und wie viele frische Daten er bereits gesehen hat.
- Für kleine Roboter (unter 600 Millionen Parametern) mit begrenzten frischen Daten ist Paraphrasierung ein großartiger Trick. Es ist, als würde man dem Roboter eine neue Perspektive auf dieselbe Geschichte geben, was ihm hilft zu lernen.
- Für große Roboter (über 7 Milliarden Parameter) oder wenn man bereits sehr viele frische Daten hat, funktioniert Paraphrasierung nicht mehr. Es ist, als würde man versuchen, einem Doktoranden ein Kinderbuch in verschiedenen Versionen vorzulesen; er hat die Grundlagen bereits gelernt, und die neue Version fügt nichts Neues hinzu.
- Repetition (das wiederholte Lesen desselben Buches) funktioniert besser für größere Modelle, aber selbst dann gibt es eine Grenze. Man kann nicht einfach dasselbe Buch immer und immer wieder lesen und erwarten, dass der Roboter unendlich klüger wird.
Das Paper führt eine spezielle Zahl ein, die sie (eta) nennen, um genau zu messen, wie viel ein „abgeleitetes“ Wort im Vergleich zu einem „frischen“ Wort wert ist. Wenn gleich 1 ist, ist das neue Wort so gut wie ein frisches. Wenn es 0 ist, ist es nutzlos. Sie fanden heraus, dass kein fester Wert ist; es sinkt, wenn das Modell größer wird und wenn man mehr abgeleitete Daten verwendet.
Dies führt zu drei unterschiedlichen „Zonen“ des Trainings, die Ingenieuren helfen sollen zu entscheiden, was als Nächstes zu tun ist:
- Compute-Bound (Rechenleistung begrenzt): Sie haben reichlich frische Daten, aber nicht genug Rechenleistung. Hier sollten Sie einfach mit dem Training auf frischen Daten fortfahren.
- Data-Bound (Daten begrenzt): Sie haben keine frischen Daten mehr. Hier können Sie mehr Rechenleistung für Repetition oder Paraphrasierung aufwenden, aber Sie müssen vorsichtig sein, da die Erträge schnell sinken.
- Model-Bound (Modell begrenzt): Sie haben alle verfügbaren Daten genutzt und die Rechenleistung ist am Limit. Der einzige Weg, jetzt klüger zu werden, ist, ein größeres Gehirn (ein größeres Modell) zu bauen.
Eines der praktischsten Ergebnisse betrifft die Frage, wie oft ein Buch gelesen werden sollte. Eine populäre Faustregel in der Vergangenheit war, einen Datensatz 4 Mal zu lesen (4 Epochen). Die Autoren fanden heraus, dass dies nur für mittelgroße Modelle mit einer spezifischen Menge an Daten eine gute Idee ist. Wenn Sie ein riesiges Modell oder einen massiven Datensatz haben, ist das 4-malige Lesen eine Verschwendung von Zeit und Geld; Sie sollten viel früher damit aufhören. Umgekehrt könnten Sie für sehr kleine Modelle mit sehr wenig Daten vielleicht viel öfter lesen müssen.
Sie verglichen auch die beiden Strategien direkt miteinander. Sie fanden einen „Wendepunkt“. Wenn Ihr Modell klein ist (unter 600 Millionen) und Sie nicht viel Daten haben, ist Paraphrasierung der Gewinner. Aber sobald Ihr Modell größer wird (über 7 Milliarden) oder Sie einen riesigen Datensatz haben, wird Repetition zur besseren Wahl, und Paraphrasierung wird tatsächlich ineffektiv.
Kurz gesagt: Dieses Paper sagt uns, dass die Ära von „kauf einfach mehr Daten“ vorbei ist. Wir treten in eine neue Ära ein, in der wir klug mit den Daten umgehen müssen, die wir haben. Wir können nicht einfach mehr Rechenleistung auf das Problem werfen und erwarten, dass es funktioniert; wir müssen genau wissen, wann wir mit dem Wiederholen aufhören müssen, wann wir das Umschreiben versuchen sollten und wann wir zugeben müssen, dass wir ein größeres Gehirn brauchen. Die Autoren legen nahe, dass für die meisten praktischen Situationen die alte „Chinchilla“-Regel (die davon ausging, dass Daten unendlich vorhanden sind) nicht länger der beste Leitfaden ist. Stattdessen müssen wir unsere Rechenleistung, unsere Modellgröße und unser Datenbudget sorgfältig abwägen und diese neuen Regeln nutzen, um Zeit mit Tricks zu vermeiden, die nicht funktionieren.
Die Forscher sind sehr zuversichtlich in ihre Ergebnisse, da sie diese über viele verschiedene Modellgrößen und Datenmengen hinweg getestet haben, und die Mathematik hielt stand. Sie prüften sogar, ob die mit diesen Tricks trainierten Modelle bei realen Aufgaben (wie dem Beantworten von Fragen oder dem Lösen von Matheaufgaben) tatsächlich besser wurden, und ja, das taten sie. Ein niedrigerer „Loss“ (ein Maß dafür, wie verwirrt der Roboter ist) korrelierte mit einer besseren Leistung des Roboters.
Wenn Sie also das nächste Mal von einem neuen KI-Modell hören, denken Sie daran: Es geht nicht nur darum, wie viel es gelesen hat, sondern wie es gelesen hat. Und wenn Sie derjenige sind, der die KI trainiert, drücken Sie nicht einfach immer wieder auf „Repeat“ bei derselben Playlist; manchmal brauchen Sie einen Remix, und manchmal müssen Sie einfach aufhören und sich ein größeres Gehirn holen.
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.