Trainings-FLOPs Rechner
Eingaben
| Durchlauftyp | Trainingsschritt (6ND) |
|---|---|
| Parameter | 70 |
| Trainings-Tokens | 1.400 |
Trainings-FLOPs Rechner
Schätzt den Rechenaufwand für das Training eines Transformers mit der 6ND-Regel aus Parameterzahl und Trainings-Tokens, ausgegeben in Gleitkommaoperationen (FLOPs) und PetaFLOP/s-Tagen.
Eingaben
Modell und Daten
Ergebnisse
Geben Sie einen Wert ein, um die Ergebnisse zu sehen.
Details
Trainings-FLOPs
Der Rechenaufwand für das Training eines großen Sprachmodells wird von einer einzigen Größe bestimmt: der Zahl der Gleitkommaoperationen, die über den gesamten Trainingslauf ausgeführt werden. Für dichte Transformer lässt sich diese Zahl gut durch die 6ND-Regel annähern, nach der das Training etwa sechs Operationen je Parameter und je gesehenem Token kostet. Dieser Rechner wendet die Regel an: Aus einer Parameterzahl und einer Token-Zahl ergeben sich die reine Operationszahl und der entsprechende Wert in PetaFLOP/s-Tagen.
Die 6ND-Regel
Ein Vorwärtsdurchlauf eines Transformers führt grob zwei Gleitkommaoperationen je Parameter und Token aus — eine Multiplikation und eine Addition in den Matrixmultiplikationen, die den Aufwand dominieren. Das Training fügt einen Rückwärtsdurchlauf hinzu, der die Gradienten sowohl zu den Aktivierungen als auch zu den Gewichten zurückführt und etwa das Doppelte des Vorwärtsdurchlaufs kostet. Vorwärts- und Rückwärtsdurchlauf zusammen ergeben rund sechs Operationen je Parameter und Token. Mit Parametern und Trainings-Tokens liegt die Summe daher nahe bei . Die Schätzung erfasst die dominierenden linearen Schichten und lässt kleinere Beiträge von Attention, Embeddings und Normalisierung außer Acht, die für große Modelle vernachlässigbar sind.
Die Formel
Mit dem Durchlauffaktor (sechs für einen Trainingsschritt, zwei für einen reinen Vorwärtsdurchlauf), der Parameterzahl in Milliarden und der Token-Zahl in Milliarden ergeben sich die Gesamtoperationszahl und die Rechenleistung in PetaFLOP/s-Tagen als
CP=k⋅N⋅D⋅1018=8.64×1019Cwobei die FLOPs gesamt und die Rechenleistung in PetaFLOP/s-Tagen ist. Der Faktor setzt die Milliarden in und wieder in absolute Zahlen um. Ein PetaFLOP/s-Tag ist ein PetaFLOP/s über einen Tag gehalten, gleich Operationen, sodass die Division durch diesen Wert eine reine FLOP-Zahl in die Einheit umrechnet, in der große Läufe üblicherweise angegeben werden.
Rechenbeispiel
Betrachtet sei ein Modell mit 70 Milliarden Parametern, trainiert auf 1.400 Milliarden Tokens — ein Verhältnis von zwanzig Tokens je Parameter, im Einklang mit der rechenoptimalen Praxis:
CP=6×70×1400×1018=5.88×1023 FLOPs=8.64×10195.88×1023≈6805.6 PFLOP/s-TageDer Lauf benötigt etwa Operationen, also rund 6.806 PetaFLOP/s-Tage. Setzt man den Durchlauffaktor auf zwei, ergibt sich allein der Aufwand des Vorwärtsdurchlaufs, ein Drittel des Trainingswerts.
Grenzen
Die 6ND-Regel ist eine idealisierte Zählung von Arithmetik, keine Vorhersage von Laufzeit oder Kosten. Reale Hardware hält nur einen Bruchteil ihres Spitzendurchsatzes, sodass sich die tatsächliche Zeit aus der Division dieser Rechenleistung durch den erreichten und nicht durch den angegebenen Durchsatz ergibt — eine Lücke, die die Rechner für die Modell-FLOPs-Auslastung erfasst. Die Schätzung lässt zudem die Neuberechnung von Aktivierungen, Optimierer-Schritte, Evaluierungsdurchläufe sowie fehlgeschlagene und neu gestartete Läufe außer Acht, die alle den praktischen Aufwand erhöhen. Die ergänzende Frage, wie viele Tokens ein Modell gegebener Größe sehen sollte, behandelt der Chinchilla-optimale Tokens Rechner, und die Umrechnung dieser Rechenleistung in ein Mietbudget der GPU-Cloud-Kosten-Rechner.
Häufig gestellte Fragen (FAQ)
Was besagt die 6ND-Regel?
Die 6ND-Regel schätzt, dass das Training eines dichten Transformers etwa sechs Gleitkommaoperationen je Parameter und Trainings-Token kostet, wobei N die Parameterzahl und D die Token-Zahl ist. Der Faktor sechs setzt sich aus rund zwei Operationen je Parameter für den Vorwärtsdurchlauf und etwa vier für den Rückwärtsdurchlauf zusammen, der die Gradienten berechnet.
Es handelt sich um eine Näherung erster Ordnung aus der Forschung zu Skalierungsgesetzen; sie erfasst die dominierenden Matrixmultiplikationen und lässt kleinere Terme wie Attention und Embeddings außer Acht.
Wie hängen FLOPs gesamt und PetaFLOP/s-Tage zusammen?
Ein PetaFLOP/s-Tag ist die Rechenleistung, die entsteht, wenn ein PetaFLOP/s einen vollen Tag lang gehalten wird, also 8,64 × 10¹⁹ Gleitkommaoperationen (10¹⁵ Operationen pro Sekunde mal 86.400 Sekunden). Die Division einer Gesamt-FLOP-Zahl durch diesen Wert ergibt PetaFLOP/s-Tage, die Einheit, in der große Trainingsläufe meist angegeben werden, weil reine FLOP-Zahlen unhandliche Größenordnungen erreichen.
Was schließt diese Schätzung ein und was nicht?
Die Schätzung zählt die Gleitkommaoperationen der dominierenden linearen Schichten über Vorwärts- und Rückwärtsdurchlauf für die angegebene Token-Zahl. Sie berücksichtigt nicht die Hardwareauslastung, sodass die tatsächliche Laufzeit und die Kosten eines realen Laufs höher liegen als die ideale Rechenleistung nahelegt.
Sie lässt außerdem die Neuberechnung von Aktivierungen, den Optimierer-Overhead, die Evaluierung sowie fehlgeschlagene oder neu gestartete Läufe außer Acht, die alle den praktischen Aufwand erhöhen.
Haftungsausschluss
Dies verwendet die 6ND-Näherung erster Ordnung für dichte Transformer und zählt nur die ideale Rechenleistung. Die tatsächliche Trainingszeit und die Kosten hängen von der Hardwareauslastung ab, die deutlich unter dem Spitzenwert liegt, sowie von Overhead, den die Formel auslässt. Das Ergebnis ist als Planungsschätzung zu verstehen, nicht als exakte Abrechnung.