Optimizing Speculative Decoding for Real-Time LLM Serving

Optimalisatie van speculatieve decodering voor real-time LLM-serving

Hoe het koppelen van conceptmodellen en boomgebaseerde verificatie het genereren van tokens per seconde met 2,5x verhoogt zonder de verwarring te verminderen.

Versnelde autoregressieve generatie

Autoregressieve taalgeneratie is gebonden aan geheugenbandbreedte: elke tokengeneratie vereist het laden van miljarden modelparameters van HBM in SRAM om een enkele voorwaartse doorgang te verwerken.

1. Het concept- en verificatieparadigma

Bij speculatieve decodering wordt een lichtgewicht conceptmodel (bijvoorbeeld 1B-3B) gecombineerd met een doelmodel met hoge capaciteit (bijvoorbeeld 70B). Het conceptmodel genereert speculatief $K$ kandidaat-tokens, die het doelmodel parallel verifieert binnen een enkele voorwaartse doorgang.

2. Wiskundige acceptatiecriteria

Met behulp van aangepaste afwijzingssteekproeven worden kandidaat-tokens geaccepteerd of afgewezen om ervoor te zorgen dat de uitvoerverdeling wiskundig identiek blijft aan de rechtstreekse bemonstering uit het doelmodel.

3. Op bomen gebaseerde conceptverificatie (Medusa & EAGLE)

In plaats van eenvoudige lineaire conceptreeksen, stelt boomgebaseerde verificatie meerdere vertakkingskandidaten tegelijkertijd op, waardoor de gemiddelde acceptatiegraad boven de 80% komt en 2,2x tot 3,0x versnellingen in productie-API's oplevert.

💬 Discussion 1
Guest
Avatar
  • Ruben van den Berg (collapsed)
    Liet het stuk over „Optimalisatie van speculatieve decodering voor real-time LLM-serving“ zien aan onze lead developer. Die staarde twintig seconden roerloos naar het scherm en liep zonder iets te zeggen naar het koffiezetapparaat. Moeten we dat zien als akkoord of als een stille noodkreet??
    0