InayofuataMwongozo unaofuata
Adhabu ya Kurudia na Vidhibiti vya Kusimbua
Lugha AI
Lugha AI MWONGOZO
Medusa ni mbinu ya kubahatisha-decoding ambayo huweka 'vichwa' vya ziada vya utabiri kwenye modeli ya lugha ili iweze kukisia tokeni nyingi za siku zijazo mara moja.
Sikiliza mwongozo huuInaonyeshwa na AI · ~2 min
By verifying these guesses in a single forward pass, it speeds up text generation roughly 2-3x without changing the model's output distribution.
Normal language models generate one token per forward pass, which is slow because each step must wait for the previous one. Medusa adds lightweight feed-forward heads on top of the frozen base model; each head predicts a token a few positions ahead (head 1 predicts the next token, head 2 the token after, and so on). These predictions form a tree of candidate continuations. The full model then verifies the whole tree in one pass using a 'tree attention' mask, accepting the longest prefix that matches what the model would have produced anyway. For Medusa-1, the backbone stays frozen and original-model rejection sampling can preserve the base model's output distribution, enabling lossless inference acceleration. Medusa-2 jointly trains the heads and backbone with a special recipe that preserves the backbone's next-token capability while improving head prediction. The paper also studies typical acceptance, which can increase acceptance and maintain similar generation quality but does not make the same exact distribution-matching guarantee.
Kupunguza muda wa kuchelewa kujibu chatbot kwa kukubali tokeni nyingi zilizothibitishwa kwa kila pasi ya mbele
Kuongeza kasi ya usaidizi wa kukamilisha msimbo ambapo mfuatano wa tokeni unaotabirika ni rahisi kukisia
Kupunguza gharama ya makisio kwa API za LLM zenye trafiki nyingi bila kupeleka muundo tofauti wa rasimu
Kuongeza kasi ya utengenezaji wa maandishi ya umbo refu kama vile muhtasari huku tukiweka towe sawa na usimbaji wa kawaida
Usimbaji wa kubahatisha unakuwa wa kawaida katika rafu za makisio ya uzalishaji, na mbinu zinazojitosheleza kama vile Medusa, ambazo huepuka kuhitaji muundo tofauti wa rasimu, zinavutia kwa sababu ni rahisi kutumia. Kazi ya siku zijazo inachanganya vichwa vya mtindo wa Medusa na ubashiri wa kipengele cha mtindo wa EAGLE, ujenzi bora wa miti na uthibitishaji wa kufahamu maunzi. Tarajia muunganisho mkali zaidi katika mifumo ya kuhudumia, urekebishaji kiotomatiki wa umbo la mti kwa kila mzigo wa kazi, na michanganyiko yenye ukandamizaji wa akiba ya KV ili muda wa kusubiri ushuke bila GPU za ziada au kupoteza ubora.
Jarida halali
Hadithi tatu za AI zilizothibitishwa kila asubuhi ya wiki, zimetengenezwa kwa Kiingereza rahisi. Bila malipo daima, hakuna matangazo.
Barua pepe moja kila siku ya wiki. Jiondoe kwa bonyeza moja. Hatutauza au kushiriki anwani yako.
Jijaribu mwenyewe
Maoni ya haraka juu ya kila jibu, na cheti kinachoweza kushirikishwa na kitambulisho kinachoweza kuthibitishwa baada ya kupita kozi.
Changia elimu ya AI bila malipo. AI Understanding ni 501(c)(3) isiyo na faida — hakuna matangazo, hakuna kizuizi cha malipo, daima. Toa mchango
Medusa ni mbinu ya kubahatisha-decoding ambayo huweka 'vichwa' vya ziada vya utabiri kwenye modeli ya lugha ili iweze kukisia tokeni nyingi za siku zijazo mara moja. Kwa kuthibitisha makadirio haya kwa kupitisha moja kwa moja, huongeza kasi ya uundaji wa maandishi takriban 2-3x bila kubadilisha usambazaji wa matokeo ya muundo.
Kila kichwa cha Medusa kinatabiri ishara nafasi kadhaa katika siku zijazo, hivyo ishara kadhaa zinaweza kupendekezwa mara moja.
The Medusa paper identifies Medusa-1, which freezes the backbone, as enabling lossless acceleration; exact rejection sampling verifies accepted candidates against the base model distribution.
Watahiniwa huunda mti, na kinyago cha kuzingatia mti huruhusu kielelezo cha msingi kuthibitisha matawi yote kwa njia moja ya mbele.
Medusa huambatisha vichwa vyepesi kwa muundo uliopo, kwa hivyo hakuna haja ya kutoa mafunzo na kuhudumia mtandao tofauti wa rasimu.
Medusa kwa ujumla hufikia takriban punguzo la 2-3x la latency ya kizazi kulingana na mzigo wa kazi.
Endelea kujifunza
Miongozo zaidi imechaguliwa kwa mada hii
InayofuataMwongozo unaofuata
Adhabu ya Kurudia na Vidhibiti vya Kusimbua
Lugha AI