Reading the Math in AI Papers · Teil 3 von 3

Trainingsziele lesen: Distillation, DPO und GRPO

Automatische Übersetzung

Dieser Artikel wurde automatisch aus der englischen Originalversion übersetzt.

Distillation, Direct Preference Optimization (DPO) und Group Relative Policy Optimization (GRPO) verändern alle die Wahrscheinlichkeiten eines Sprachmodells, verwenden aber unterschiedliche Trainingsbeispiele und Rückmeldungen. Dieser Leitfaden richtet sich an Ingenieure, die diese Unterschiede verstehen möchten, bevor sie die vollständigen Zielfunktionen lesen. Die Diagramme zeigen, woher die Rückmeldungen stammen und wie sie Wahrscheinlichkeitsvergleiche verändern; die Gleichungen beschreiben die entsprechenden Rechenschritte.

Beim gewöhnlichen überwachten Fine-Tuning lernt das Model, das nächste Token in gespeicherten Beispielen vorherzusagen. Die folgenden Methoden verändern die Rückmeldung: Ein Teacher liefert Wahrscheinlichkeiten, ein Präferenzpaar gibt die bessere Antwort an, oder ein Reward bewertet Antworten, die während des Trainings gesampelt werden. Eine Policy ist einfach das Model, das Antworten erzeugt. Alle folgenden Logarithmen sind natürliche Logarithmen.

Teil 1 erklärt, wie man eine Gleichung liest, und enthält eine Notationsübersicht. Hier erklärt jeder Abschnitt die benötigten Begriffe und lässt sich unabhängig lesen.

Klassische Distillation: eine Temperatur für beide Softmax-Funktionen

Ein Trainingslabel bezeichnet das richtige Ergebnis: Bei der Vorhersage des nächsten Tokens ist es das nächste Token in einem gespeicherten Beispiel. Ein Teacher-Model liefert mehr Informationen: seine Wahrscheinlichkeiten für alle möglichen Tokens, einschließlich plausibler Alternativen. Distillation trainiert einen kleineren Student, diese Wahrscheinlichkeiten nachzubilden. Das Training verringert einen numerischen Loss, indem es die Parameter des Students verändert; die des Teachers bleiben unverändert.

Das Distillation-Paper von Hinton et al., §2, flacht die Wahrscheinlichkeitsverteilungen beider Models vor dem Vergleich ab. Dazu wird jedes Logit, der Rohwert vor Softmax, durch eine Temperatur TT geteilt:

qi=exp⁡(zi/T)∑jexp⁡(zj/T).q_i=\frac{\exp(z_i/T)}{\sum_j\exp(z_j/T)}.

Hier ist ziz_i der Score des Kandidaten ii und qiq_i seine Wahrscheinlichkeit. Angenommen, ein einfaches Beispielmodel hat nur zwei mögliche nächste Tokens, „cat“ und „dog“, und weist ihnen nach „The animal is a“ die Scores (2,0)(2,0) zu. Bei T=1T=1 betragen die Wahrscheinlichkeiten ungefähr (0.88,0.12)(0.88,0.12); bei T=2T=2 werden daraus (0.73,0.27)(0.73,0.27). Der kleinere Score trägt nun stärker zum Vergleich bei. Teacher und Student verwenden dieselbe Temperatur.

Die Balken zeigen, wie aus denselben beiden Scores mit steigender Temperatur eine weniger konzentrierte Verteilung entsteht.

Für die beispielhaften nächsten Tokens cat und dog ergeben die Logits 2 und 0 bei Temperatur 1 die Wahrscheinlichkeiten 0.88 und 0.12, bei Temperatur 2 dagegen 0.73 und 0.27. Beide Diagramme verwenden dieselbe Skala von null bis eins. Die Wahrscheinlichkeit von dog steigt.Für die beispielhaften nächsten Tokens cat und dog ergeben die Logits 2 und 0 bei Temperatur 1 die Wahrscheinlichkeiten 0.88 und 0.12, bei Temperatur 2 dagegen 0.73 und 0.27. Beide Diagramme verwenden dieselbe Skala von null bis eins. Die Wahrscheinlichkeit von dog steigt.

Der Trainingsloss kombiniert den gewöhnlichen Label-Loss mit einem Loss zum Abgleich mit der abgeflachten Teacher-Verteilung. p(T)p^{(T)} bezeichnet die Wahrscheinlichkeiten des Teachers und q(T)q^{(T)} die des Students bei Temperatur TT. Für eine Eingabe mit Label gilt:

L=(1−λ)(−log⁡qy(1))+λT2DKL(p(T)∥q(T)).\begin{aligned} \mathcal L&=(1-\lambda)\bigl(-\log q_y^{(1)}\bigr)\\ &\quad+\lambda T^2 D_{\mathrm{KL}}\bigl(p^{(T)}\Vert q^{(T)}\bigr). \end{aligned}

Hier ist L\mathcal L der Loss, yy das richtige nächste Token, und λ∈[0,1]\lambda\in[0,1] bestimmt die Mischung. Hochgestellte Ausdrücke in Klammern bezeichnen Temperaturen, keine Potenzen. Der erste Term bedeutet: „Eine niedrige Wahrscheinlichkeit für das richtige Token bei T=1T=1 bestrafen“. Der zweite bedeutet: „Die abgeflachten Wahrscheinlichkeiten des Teachers nachbilden“. Das Training mittelt diese Losses über die gespeicherten Eingaben und verändert nur den Student.

DKL(p∥q)D_{\mathrm{KL}}(p\Vert q) ist die KL-Divergenz vom Teacher pp zum Student qq. Sie misst, wie sich zwei Wahrscheinlichkeitsverteilungen unterscheiden: Bei Übereinstimmung ist sie null. Die erste Verteilung liefert die Gewichtungsfaktoren für den Vergleich; beim Teacher wahrscheinliche Ergebnisse erhalten somit mehr Gewicht. Hinton et al. beschreiben einen Abgleichterm mit Kreuzentropie; diese KL-Form zieht eine nur vom Teacher abhängige Konstante ab und ergibt dieselben Gradienten für den Student.

Der Faktor T2T^2 gleicht die kleineren Gradienten bei höheren Temperaturen näherungsweise aus. Dies ist eine Trainingseinstellung; beim Einsatz verwendet der Student für diese Softmax-Funktion normalerweise wieder T=1T=1.

Ein Präfix ist der Text, der vor dem nächsten Token bereits vorliegt: der Prompt und alle bisherigen Antwort-Tokens. Beispielsweise ist „The animal is a“ das Präfix vor „cat“. In der Abbildung lesen beide Models dasselbe Präfix und erzeugen Wahrscheinlichkeiten für das nächste Token. Links stammen die Präfixe aus gespeicherten Beispielen. Rechts stammen sie aus Antworten, die der Student während des Trainings erzeugt. Beide Abläufe verwenden den Vergleichsloss, um nur den Student zu aktualisieren.

Klassische Distillation gibt feste Eingaben an einen unveränderten Teacher und einen trainierbaren Student weiter und vergleicht dann ihre Wahrscheinlichkeiten. On-Policy-Distillation sampelt zuerst Text vom Student; anschließend vergleichen Teacher und Student die Wahrscheinlichkeiten des nächsten Tokens an Präfixen dieses Texts. Nur der Student wird aktualisiert.Klassische Distillation gibt feste Eingaben an einen unveränderten Teacher und einen trainierbaren Student weiter und vergleicht dann ihre Wahrscheinlichkeiten. On-Policy-Distillation sampelt zuerst Text vom Student; anschließend vergleichen Teacher und Student die Wahrscheinlichkeiten des nächsten Tokens an Präfixen dieses Texts. Nur der Student wird aktualisiert.

Es gibt auch Distillation ohne Wahrscheinlichkeitsabgleich. Die distillierten Models von DeepSeek-R1, §2.4, wurden mit 800.000 Beispielen feinabgestimmt, die zuvor mithilfe von DeepSeek-R1 zusammengestellt worden waren. Sie verwendeten einen überwachten Loss auf gespeichertem Text; während dieses Trainings musste kein Teacher ausgeführt werden.

Die ausführliche Distillation-Lektüre erklärt den gemischten Loss und die Temperaturskalierung.

On-Policy-Distillation: Der Student steht im KL an erster Stelle

Ein Student, der eigene Antworten erzeugt, kann Präfixe erreichen, also Prompt und bisherige Antwort, die im Trainingsdataset nie vorkamen. On-Policy-Distillation liefert Rückmeldungen zu Text, den der Student tatsächlich erzeugt. Für einen Prompt sampelt der Student eine Antwort. Ein unveränderter Teacher bewertet dann dieselben Präfixe. Das Training vergleicht die beiden Models somit dort, wo der Student Anleitung braucht.

On-Policy beschreibt diese Herkunft des Trainingstexts. Der Begriff allein schreibt keinen bestimmten Loss vor. Die Beispiele hier verwenden Reverse-KL mit dem Student an erster Stelle:

DKL(qθ∥p)=Ey∼qθ[log⁡qθ(y∣x)p(y∣x)].D_{\mathrm{KL}}(q_\theta\Vert p) =\mathbb E_{y\sim q_\theta}\left[\log\frac{q_\theta(y\mid x)}{p(y\mid x)}\right].

Für einen festen Prompt xx ist qθq_\theta der Student mit trainierbaren Parametern θ\theta, pp der Teacher und yy eine vom Student gesampelte Antwort. Der Ausdruck vergleicht die Wahrscheinlichkeit, die der Student seiner Antwort zuweist, mit der Wahrscheinlichkeit des Teachers. Über Student-Antworten gemittelt bestraft er den Student dafür, Text eine hohe Wahrscheinlichkeit zu geben, den der Teacher für unwahrscheinlich hält.

Dies unterscheidet sich von der Forderung, alle möglichen Ausgaben des Teachers abzudecken. Wenn ein kleiner Student die vollständige Verteilung des Teachers nicht nachbilden kann, kann Reverse-KL eine Konzentration auf einige wahrscheinliche Teacher-Ausgaben begünstigen. MiniLLM, §2.1, Gleichung (1), verwendet diese Zielfunktion. Das tatsächliche Trainingsverfahren mischt außerdem Teacher-Samples in die Generierung und passt den Gradienten an, um Varianz und Längenverzerrung zu verringern; das Diagramm zeigt den grundlegenden On-Policy-Aufbau, nicht alle Anpassungen von MiniLLM.

Im folgenden einfachen Beispiel weisen Teacher und Student „cat“ und „dog“ nach demselben Präfix unterschiedliche Wahrscheinlichkeiten zu. KL multipliziert das logarithmierte Verhältnis jedes Tokens mit dessen Gewichtungsfaktor und addiert die beiden Terme. Ein anderes erstes Argument verschiebt das größere Gewicht von „cat“ zu „dog“. Die Balken zeigen die Gewichtungsfaktoren, nicht die vollständigen Loss-Beiträge.

Nach demselben Präfix betragen die Teacher-Wahrscheinlichkeiten für cat und dog 0.8 und 0.2; die Student-Wahrscheinlichkeiten betragen 0.4 und 0.6. Forward-KL gewichtet die logarithmierten Teacher-zu-Student-Verhältnisse mit 0.8 und 0.2. Reverse-KL gewichtet die logarithmierten Student-zu-Teacher-Verhältnisse mit 0.4 und 0.6.Nach demselben Präfix betragen die Teacher-Wahrscheinlichkeiten für cat und dog 0.8 und 0.2; die Student-Wahrscheinlichkeiten betragen 0.4 und 0.6. Forward-KL gewichtet die logarithmierten Teacher-zu-Student-Verhältnisse mit 0.8 und 0.2. Reverse-KL gewichtet die logarithmierten Student-zu-Teacher-Verhältnisse mit 0.4 und 0.6.

Von Sequenz-KL zu Token-KL

Die Wahrscheinlichkeit einer Antwort setzt sich aus ihren Wahrscheinlichkeiten für das jeweils nächste Token zusammen. An jedem Präfix einer Student-Antwort sagen beide Models das nächste Token vorher. Ihre Unterschiede lassen sich daher Position für Position vergleichen und dann über die Antwort aufsummieren. DeepSeek-V4, §5.1.2, Gleichung (29), verwendet Reverse-KL-Distillation von mehreren spezialisierten Teacher-Models in einen Student. Diese Teachers sind eigenständige Models, keine Experten-Teilnetze innerhalb einer Mixture-of-Experts-Schicht.

An Position tt sei qt(v)=qθ(v∣x,y<t)q_t(v)=q_\theta(v\mid x,y_{<t}) die Wahrscheinlichkeit des Students für das mögliche Token vv. Das Präfix enthält den Prompt xx und die vorhergehenden Antwort-Tokens y<ty_{<t}; pt(v)p_t(v) ist die Wahrscheinlichkeit des Teachers an demselben Präfix. Dann gilt:

dt=∑v∈Vqt(v)log⁡qt(v)pt(v),DKL(qθ∥p)=Ey∼qθ[∑tdt].\begin{aligned} d_t&=\sum_{v\in V}q_t(v)\log\frac{q_t(v)}{p_t(v)},\\ D_{\mathrm{KL}}(q_\theta\Vert p) &=\mathbb E_{y\sim q_\theta}\left[\sum_t d_t\right]. \end{aligned}

VV ist das Vokabular und dtd_t die Next-Token-KL an einem Präfix. Die innere Summe bedeutet: „Alle möglichen Tokens vergleichen, gewichtet mit der Wahrscheinlichkeit des Students“. Diese Vergleiche werden über die Positionen summiert und dann über die vom Student erzeugten Antworten gemittelt. Der äußere Erwartungswert sampelt weiterhin Text, auch wenn jedes dtd_t das ganze Vokabular aufsummiert.

Ein gesampeltes Token oder das gesamte Vokabular

Der Teacher kann nur das vom Student gewählte Token bewerten oder Wahrscheinlichkeiten für das gesamte Vokabular liefern. Die erste Option überträgt weniger Informationen; die zweite vergleicht alle möglichen nächsten Tokens an diesem Präfix. DeepSeek-V4 berichtet von instabilem Training mit seiner Ein-Token-Schätzung und verwendet den Vergleich über das ganze Vokabular. Dadurch wird der Token-Vergleich an einem gegebenen Präfix exakt; die Präfixe stammen weiterhin aus gesampelten Antworten.

Beide Models lesen „The animal is a“ und liefern die unten gezeigten Wahrscheinlichkeiten. Die hervorgehobenen Zeilen gehen in den Loss ein, mit dem der Student aktualisiert wird. Die Variante mit vollständigem Vokabular umfasst „dog“ und „fox“, obwohl der Student „cat“ gewählt hat; die Ein-Token-Variante schätzt diesen Vergleich allein anhand des gewählten Tokens.

Am Präfix „The animal is a“ sampelt der Student cat. Eine Ein-Token-Schätzung vergleicht nur die Wahrscheinlichkeiten für cat. Die KL über das vollständige Vokabular umfasst cat, dog und fox in diesem Beispielvokabular und gewichtet jedes logarithmierte Verhältnis mit der Student-Wahrscheinlichkeit. Das Präfix bleibt gesampelt.Am Präfix „The animal is a“ sampelt der Student cat. Eine Ein-Token-Schätzung vergleicht nur die Wahrscheinlichkeiten für cat. Die KL über das vollständige Vokabular umfasst cat, dog und fox in diesem Beispielvokabular und gewichtet jedes logarithmierte Verhältnis mit der Student-Wahrscheinlichkeit. Das Präfix bleibt gesampelt.

Mit den drei Zeilen ergibt sich dt=0.5log⁡(0.5/0.6)+0.2log⁡(0.2/0.3)+0.3log⁡(0.3/0.1)≈0.157d_t=0.5\log(0.5/0.6)+0.2\log(0.2/0.3)+0.3\log(0.3/0.1)\approx0.157. Das logarithmierte Verhältnis für das gesampelte „cat“ allein beträgt ungefähr −0.182-0.182. Ein einzelnes Sample kann negativ sein, obwohl die vollständige KL nichtnegativ ist.

Die ausführliche On-Policy-Lektüre behandelt die Herleitung von Sequenzen zu Tokens, Sampling-Korrekturen und Gradientenschätzungen.

DPO: ein logistischer Loss auf zwei logarithmierten Verhältnissen

DPO beginnt mit einem gespeicherten Präferenzpaar: einem Prompt, einer bevorzugten und einer abgelehnten Antwort. Das Paar kann aus menschlichen Bewertungen früherer Model-Ausgaben stammen. Während des Trainings liest das Model jede gespeicherte Antwort Token für Token und erfasst ihre Wahrscheinlichkeit unter Berücksichtigung des Prompts und der vorhergehenden Tokens. Die Summe dieser Log-Wahrscheinlichkeiten ergibt die Log-Wahrscheinlichkeit der Antwort. Das bewertet, wie wahrscheinlich der Text unter dem Model ist; das gespeicherte Präferenzlabel gibt an, welche Antwort besser ist.

Bei einem üblichen RLHF-Verfahren trainieren Präferenzpaare zuerst ein separates Reward-Model. Anschließend trainiert Reinforcement Learning das Sprachmodell anhand dieses Rewards. DPO, §4, Gleichung (7), trainiert das Sprachmodell stattdessen direkt mit den Paaren. Ein unverändertes Referenzmodel bewertet dieselben Antworten zum Vergleich. Das Training verändert nur das aktuelle Model; für diesen Schritt muss keines der beiden Models neue Antworten erzeugen.

Für jede Antwort wird ihre Referenz-Log-Wahrscheinlichkeit von ihrer aktuellen Log-Wahrscheinlichkeit abgezogen. Diese Änderung nennen wir ihren Zuwachs. DPO vergleicht den Zuwachs der bevorzugten Antwort mit dem der abgelehnten Antwort:

g(y)=log⁡πθ(y∣x)πref(y∣x),Δ=g(yw)−g(yl),LDPO=E(x,yw,yl)∼D[−log⁡σ(βΔ)].\begin{aligned} g(y)&=\log\frac{\pi_\theta(y\mid x)}{\pi_{\mathrm{ref}}(y\mid x)},\\ \Delta&=g(y_w)-g(y_l),\\ \mathcal L_{\mathrm{DPO}} &=\mathbb E_{(x,y_w,y_l)\sim\mathcal D} \bigl[-\log\sigma(\beta\Delta)\bigr]. \end{aligned}

Hier ist xx der Prompt, ywy_w und yly_l sind seine bevorzugte und abgelehnte Antwort, und πθ\pi_\theta und πref\pi_{\mathrm{ref}} sind das aktuelle Model und das Referenzmodel. Die Sigmoidfunktion σ\sigma bildet eine Zahl auf (0,1)(0,1) ab, und β\beta ist ein positiver Skalierungsfaktor. Die letzte Zeile bedeutet: „Den eingeklammerten Paar-Loss über die gespeicherten Paare im Dataset D\mathcal D mitteln“. Ein größeres Δ\Delta verringert diesen Loss; das Training minimiert LDPO\mathcal L_{\mathrm{DPO}}, indem es die Parameter θ\theta des aktuellen Models verändert.

Wenn beispielsweise die bevorzugte Antwort 2 an Log-Wahrscheinlichkeit gewinnt und die abgelehnte Antwort 1 verliert, gilt Δ=2−(−1)=3\Delta=2-(-1)=3. Die Präferenz hat sich relativ zur Referenz in die gewünschte Richtung verändert. Der Loss hat keinen separaten KL-Term und erzwingt keinen maximalen Abstand zur Referenz.

Die Differenz allein kann ein Problem verbergen: Die bevorzugte Antwort kann unwahrscheinlicher werden, während die abgelehnte Antwort noch schneller unwahrscheinlicher wird. Der Loss verbessert sich trotzdem. Smaug untersucht diesen Fehler; prüfe daher neben dem Präferenzloss auch die einzelnen Antwortwahrscheinlichkeiten.

Die beiden folgenden Fälle ergeben dasselbe Δ=3\Delta=3 und damit bei festem β\beta denselben Paar-Loss. Nur im linken Fall wird die bevorzugte Antwort wahrscheinlicher als unter der Referenz.

Zwei Fälle verwenden dieselbe Achse für Änderungen der Log-Wahrscheinlichkeit. Bevorzugt plus 2 und abgelehnt minus 1 ergeben eine Zuwachsdifferenz von 3. Bevorzugt minus 1 und abgelehnt minus 4 ergeben ebenfalls 3, obwohl beide Antwortwahrscheinlichkeiten gegenüber der Referenz gesunken sind.Zwei Fälle verwenden dieselbe Achse für Änderungen der Log-Wahrscheinlichkeit. Bevorzugt plus 2 und abgelehnt minus 1 ergeben eine Zuwachsdifferenz von 3. Bevorzugt minus 1 und abgelehnt minus 4 ergeben ebenfalls 3, obwohl beide Antwortwahrscheinlichkeiten gegenüber der Referenz gesunken sind.

In der nächsten Abbildung gehört die DPO-Rückmeldung zu einem gespeicherten Paar. GRPO erzeugt während des Trainings vier Antworten auf „2 + 2?“. Eine Antwortprüfung vergibt 1 für „4“ und sonst 0. Der Vergleich jedes Rewards mit dem Gruppendurchschnitt ergibt einen Advantage: Positive Rückmeldung begünstigt die Tokens der gesampelten Antwort, negative Rückmeldung wirkt ihnen entgegen. Der nächste Abschnitt zeigt diese Berechnung und die Grenzen ihrer Wirkung.

DPO bewertet zwei gespeicherte Antworten mit dem aktuellen Model und dem unveränderten Referenzmodel und aktualisiert dann das aktuelle Model. GRPO sampelt 3, 4, 4 und 5 für die Frage 2 plus 2. Eine Prüfung vergibt die Rewards 0, 1, 1 und 0. Gruppenrelative Advantages von minus 1, plus 1, plus 1 und minus 1 steuern die Aktualisierung des aktuellen Models.DPO bewertet zwei gespeicherte Antworten mit dem aktuellen Model und dem unveränderten Referenzmodel und aktualisiert dann das aktuelle Model. GRPO sampelt 3, 4, 4 und 5 für die Frage 2 plus 2. Eine Prüfung vergibt die Rewards 0, 1, 1 und 0. Gruppenrelative Advantages von minus 1, plus 1, plus 1 und minus 1 steuern die Aktualisierung des aktuellen Models.

Die ausführliche DPO-Lektüre leitet den Loss aus der durch Rewards regularisierten Zielfunktion ab und erläutert seinen Gradienten.

GRPO: die PPO-Zielfunktion mit Gruppenbaseline

Group Relative Policy Optimization sampelt mehrere Antworten auf dieselbe Frage, bewertet sie und vergleicht jeden Score mit dem Gruppendurchschnitt. Eine richtige Antwort unter überwiegend falschen Antworten erhält positive Rückmeldung; eine falsche Antwort unter richtigen Antworten erhält negative Rückmeldung. Für Mathematik oder Code können eine Antwortprüfung oder Tests die Bewertung übernehmen. Andere Aufgaben können ein Reward-Model verwenden.

DeepSeekMath, §4.1.1, führte GRPO als Variante von Proximal Policy Optimization (PPO) ein. DeepSeek-R1, §2.2.1, verwendete es später ebenfalls. Die wesentliche Änderung betrifft die Berechnung des Advantage, des relativen Rewards, der die Aktualisierung steuert:

Ai=ri−mean⁡(r1,…,rG)std⁡(r1,…,rG).A_i=\frac{r_i-\operatorname{mean}(r_1,\ldots,r_G)}{\operatorname{std}(r_1,\ldots,r_G)}.

rir_i ist der Reward von Antwort ii und GG die Anzahl der Antworten auf eine Frage. Durch Subtraktion des Mittelwerts erkennen wir, welche Antworten über oder unter dem Gruppendurchschnitt liegen. Die Division durch die Standardabweichung skaliert diese Differenzen. In der dargestellten Gruppe (0,1,1,0)(0,1,1,0) beträgt der Mittelwert 0.50.5 und die Populationsstandardabweichung 0.50.5. Daraus ergeben sich die Advantages (−1,1,1,−1)(-1,1,1,-1).

Was sich gegenüber PPO ändert

PPO trainiert üblicherweise ein separates Value-Model, das den künftigen Reward aus einer unvollständigen Antwort schätzt. GRPO verwendet stattdessen die Gruppenscores als Baseline und trainiert daher dieses Value-Model nicht. Es benötigt weiterhin Reward-Rückmeldungen, und seine ursprüngliche Zielfunktion verwendet weiterhin ein Referenzmodel für eine KL-Strafe. Das Entfernen des Value-Models beseitigt diese anderen Aufgaben nicht.

Der folgende Vergleich zeigt, welches Value-Model ersetzt wird. Die untere Zeile zeigt zwei weitere Trainingseingaben. Das alte oder Sampling-Model ist die Version, die diesen Antwortbatch erzeugt hat; seine erfassten Token-Wahrscheinlichkeiten bleiben unverändert, während das aktuelle Model auf diesem Batch trainiert. Ein separates Referenzmodel, das während dieser Aktualisierungen ebenfalls unverändert bleibt, liefert Wahrscheinlichkeiten für den KL-Vergleich. Beide Vergleiche bewerten dasselbe Token nach demselben Präfix.

PPO verwendet Reward-Rückmeldungen und ein trainiertes Value-Model zur Schätzung der Advantages. GRPO normalisiert stattdessen jeden Antwortreward anhand der Rewards derselben Frage. Das Wahrscheinlichkeitsverhältnis zwischen aktuellem und Sampling-Model und die KL zwischen aktuellem und Referenzmodel erfüllen im Training unterschiedliche Aufgaben.PPO verwendet Reward-Rückmeldungen und ein trainiertes Value-Model zur Schätzung der Advantages. GRPO normalisiert stattdessen jeden Antwortreward anhand der Rewards derselben Frage. Das Wahrscheinlichkeitsverhältnis zwischen aktuellem und Sampling-Model und die KL zwischen aktuellem und Referenzmodel erfüllen im Training unterschiedliche Aufgaben.

GRPO behält die PPO-Zielfunktion mit geclipptem Wahrscheinlichkeitsverhältnis bei. Hatte ein gesampeltes Token die Wahrscheinlichkeit 0.200.20 und weist das aktuelle Model ihm am selben Präfix nun 0.240.24 zu, beträgt das Verhältnis 0.24/0.20=1.20.24/0.20=1.2. Die Zielfunktion gewichtet dieses Verhältnis mit dem Advantage der Antwort. Das Training verändert das aktuelle Model, um diese Zielfunktion zu erhöhen. Eine separate KL-Strafe wirkt einer Abweichung von der Referenz entgegen.

Für eine gesampelte Antwort bezeichnet sts_t den Prompt und die Antwort-Tokens vor Position tt, und yty_t das Token an dieser Position. Der im Diagramm dargestellte Teil von Gleichung (3) aus DeepSeekMath lautet:

ρt=πθ(yt∣st)πold(yt∣st),Ct=min⁡ ⁣(ρtA,clip⁡(ρt,1−ε,1+ε)A).\begin{aligned} \rho_t&=\frac{\pi_\theta(y_t\mid s_t)}{\pi_{\mathrm{old}}(y_t\mid s_t)},\\ C_t&=\min\!\bigl(\rho_t A,\\ &\qquad\operatorname{clip}(\rho_t,1-\varepsilon,1+\varepsilon)A\bigr). \end{aligned}

πθ\pi_\theta ist das aktuelle Model, πold\pi_{\mathrm{old}} die Sampling-Kopie und AA der Gruppen-Advantage dieser Antwort. Die Breite ε\varepsilon legt das Clipping-Intervall fest. clip⁡\operatorname{clip} bedeutet: „Ein Verhältnis außerhalb des Intervalls durch den nächstgelegenen Endpunkt ersetzen“. min⁡\min bedeutet anschließend: „Den kleineren der beiden mit dem Advantage gewichteten Terme nehmen“. CtC_t ist ein zu maximierender Beitrag. GRPO zieht seine KL-Strafe ab und mittelt über Tokens innerhalb der Antworten, Antworten innerhalb der Gruppen und gesampelte Fragen.

Mit A=1A=1, ε=0.2\varepsilon=0.2 und ρt=1.5\rho_t=1.5 sind die beiden Terme 1.51.5 und 1.21.2, also gilt Ct=1.2C_t=1.2. Mit A=−1A=-1 und ρt=0.5\rho_t=0.5 sind sie −0.5-0.5 und −0.8-0.8, also gilt Ct=−0.8C_t=-0.8. Dies sind die beiden flachen Abschnitte des Diagramms.

Das Diagramm zeigt den Beitrag eines einzelnen Tokens bei festem Advantage. Ein positiver Advantage begünstigt ein höheres Verhältnis; ein negativer begünstigt ein niedrigeres. Clipping macht die Linie dort flach, wo eine günstige Änderung diesen Term nicht weiter verbessert. Es setzt keine feste Grenze für die tatsächlichen Wahrscheinlichkeiten des Models.

Bei Clipping-Breite 0.2 und Advantage plus 1 steigt die geclippte Zielfunktion mit dem Verhältnis bis 1.2 und bleibt danach flach. Bei Advantage minus 1 bleibt sie unter 0.8 flach und fällt, wenn das Verhältnis über 0.8 steigt. Das tatsächliche Wahrscheinlichkeitsverhältnis kann beide Schwellen überschreiten.Bei Clipping-Breite 0.2 und Advantage plus 1 steigt die geclippte Zielfunktion mit dem Verhältnis bis 1.2 und bleibt danach flach. Bei Advantage minus 1 bleibt sie unter 0.8 flach und fällt, wenn das Verhältnis über 0.8 steigt. Das tatsächliche Wahrscheinlichkeitsverhältnis kann beide Schwellen überschreiten.

Die Sampling-Kopie und die Referenz können mit denselben Weights beginnen, dienen aber unterschiedlichen Vergleichen. Die Sampling-Kopie wird beim Sammeln eines neuen Batches aktualisiert; ihre erfassten Wahrscheinlichkeiten bleiben der Nenner für Aktualisierungen auf diesem Batch.

Wenn jede Antwort denselben Reward erhält, gibt es keine relative Präferenz zu lernen. Die Advantage-Formel dividiert dann null durch null. Implementierungen brauchen dafür eine Regel, etwa Advantages von null zu erzeugen; DAPO, §3.2, schließt diese Gruppen stattdessen aus und sampelt Ersatzgruppen.

Die ausführliche GRPO-Lektüre behandelt die vollständige Zielfunktion, Clipping-Fälle, Gradienten, den gesampelten KL-Schätzer und die Auswirkungen einer Mittelung pro Antwort oder pro Token.

Optionale Erläuterungen der Gleichungen

Die verlinkten ausführlichen Texte enthalten Herleitungen und Prüfungen mit kleinen Zahlen. Sie wurden mit den Coding-Agent-Skills aus paper-math-skills erstellt und sind in paper-math-readings gesammelt.

Quellen und Versionen