Oossa

Fireworks: rekenverschillen kunnen RL-training doen ontsporen

Een experiment van Fireworks laat zien dat afwijkende berekeningen bij de trainer en de rollout-engine de beloning kunnen laten kelderen. MoE-modellen brengen extra uitdagingen voor de afstemming met zich mee.

Kort berichtDoor Gepubliceerd door Oossa: 1 min lezen

Fireworks AI meldt dat een reinforcement-learningrun met het GLM‑5.2-model na 20 stappen instortte doordat de trainer en de rollout-engine tokenwaarschijnlijkheden verschillend berekenden. Door de berekeningen op elkaar af te stemmen, bleef de KL-divergentie nul en de beloning gedurende de volledige run van 25 stappen stabiel. Het team ontdekte ook dat kleine verschillen in de manier waarop expertoutputs worden opgeteld in een Qwen‑3.5‑MoE-model een grote k3-afwijking veroorzaakten. Dat laat zien dat MoE-routering zowel wat de selectie als de precisie betreft moet worden afgestemd.

Waarom het ertoe doet

Door de berekeningen van de trainer en rollout-engine op elkaar af te stemmen, kunnen ML-teams mislukte en kostbare RL-runs voorkomen en reinforcement learning betrouwbaarder opschalen.

Was dit artikel nuttig?
Delen

Lees ook

Oossa · Nieuwsbrief

De AI-week, uitgelegd

Elke maandag: het nieuws dat ertoe doet, in gewone taal. Gratis, geen spam.