Spekulerande avkodning snabbar upp AI-träningen med 2,5 gånger
Baserad på forskning av Hayate Iso, Tiyasa Mitra, Sudipta Mondal, Rasoul Shafipour, Venmugil Elango
Att träna toppmoderna AI-modeller stöter på ett tak. När språkmodellerna blir smartare blir processen att finslipa dem genom förstärkningsinlärning alltmer tidskrävande. Flaskhalsen handlar inte bara om beräkningskraft, utan om hur modellen genererar sin egen träningsdata i realtid. Om vi inte kan påskynda denna fas riskerar hela utvecklingscykeln att stanna upp, vilket gör effektivitet till en kritisk nödvändighet för att AI-utvecklingen ska fortskrida.
Forskare har vänt sig till spekulativ avkodning för att lösa problemet utan att offra noggrannheten. Tänk det som ett smart genvägsalternativ. Istället för att generera varje ord i en strikt sekvens, förutspår systemet flera sannolika nästa ord samtidigt och verifierar dem mot huvudmodellen. Denna teknik bevarar den ursprungliga modellens exakta utdelning medan tiden som krävs för att skapa rullningsdata minskar drastiskt. Det möjliggör både synkron och asynkron bearbetning, vilket gör det kompatibelt med olika befintliga spekulativa mekanismer som förtränade huvuden eller mindre utkastmodeller.
Resultaten är slående. I tester på en modell med åtta miljarder parametrar med synkron förstärkningsinlärning ökade genomströmningen av rullningsdata med 1,8 gånger. När detta projiceras på större modeller med 235 miljarder parametrar med asynkrona metoder kan den totala hastighetsökningen under träningen nå upp till 2,5 gånger. Det innebär att utvecklare kan träna mer kapabla resonemangsmodeller i en bråkdel av den tid som tidigare krävdes.
Slutsatsen är klar: spekulativ avkodning är inte längre bara en inference-trick för drift; det har blivit ett avgörande verktyg för själva träningen. Genom att integrera denna förlustfri acceleration direkt i förstärkningsinlärningspipelinen kan forskare kringgå de nuvarande genereringsflaskhalsarna. Detta banar väg för snabbare och mer effektiv utveckling av AI-system i nästa generation, utan att kompromissa med kvalitet eller precision.