Műszaki ÚTMUTATÓ

PagedAttention és vLLM

A PagedAttention egy memóriakezelési technika, amely a nyelvi modell figyelmi gyorsítótárát kis, újrafelhasználható blokkokban tárolja egy nagy összefüggő darab helyett.

Áttekintés

A PagedAttention egy memóriakezelési technika, amely a nyelvi modell figyelmi gyorsítótárát kis, újrafelhasználható blokkokban tárolja egy nagy összefüggő darab helyett. Ez hajtja a vLLM-et, egy nyílt forráskódú kiszolgálómotort, amely drámaian megnöveli, hogy egy GPU hány kérést tud kezelni.

A PagedAttention és a vLLM egy olyan műszaki építőelem, amely nagymértékben befolyásolja a modell minőségét, az infrastruktúra költségeit, a késleltetést és a megbízhatóságot.

Mély merülés

Amikor egy nyelvi modell szöveget generál, minden látott jogkivonathoz megőrzi a „KV gyorsítótárat” (kulcs- és értékvektorok), így a következő token a teljes kontextusra képes figyelni. Hagyományosan minden kérés egy nagy, összefüggő GPU-memóriát foglalt le, a lehető legnagyobb hosszra méretezve, így hatalmas mennyiséget pazaroltak el, amikor a sorozatok rövidebbek vagy változó hosszúságúak voltak. A UC Berkeley 2023-as vLLM-anyagában bemutatott PagedAttention a virtuális memória lapozás ötletét az operációs rendszerektől kölcsönzi: a KV gyorsítótárat fix méretű blokkokra osztja, amelyek bárhol elhelyezhetők a memóriában, és igény szerint lefoglalhatók. A keresőtábla leképezi a logikai token pozíciókat fizikai blokkokra. Ez szinte kiküszöböli a memória töredezettségét, és lehetővé teszi a blokkok megosztását, például ugyanazon prompt több kimenetén.

Technikai betekintés

A KV gyorsítótár fix méretű oldalakra van felosztva, amelyek mindegyike egy meghatározott számú token kulcsait és értékeit tartalmazza. A sorozatonkénti blokktáblázat a logikai pozíciókat a fizikai oldalhelyekhez rendeli hozzá, így a sorozat gyorsítótárának nem kell szomszédosnak lennie. Mivel az azonos előtagok (megosztott rendszerprompt vagy sugárkereső ágak) ugyanazokra a fizikai oldalakra mutathatnak írás-másolás révén, a memória újrafelhasználásra kerül a duplikált helyett, ami több mint 60%-ról néhány százalékra csökkenti a veszteséget.

A PagedAttention és a vLLM elsajátítása

A PagedAttention egy memóriakezelési technika, amely a nyelvi modell figyelmi gyorsítótárát kis, újrafelhasználható blokkokban tárolja egy nagy összefüggő darab helyett. Ez hajtja a vLLM-et, egy nyílt forráskódú kiszolgálómotort, amely drámaian megnöveli, hogy egy GPU hány kérést tud kezelni. A PagedAttention és a vLLM egy olyan műszaki építőelem, amely nagymértékben befolyásolja a modell minőségét, az infrastruktúra költségeit, a késleltetést és a megbízhatóságot. A mélyebb megértés érdekében kezelje a PagedAttention-t és a vLLM-et működési modellként, ne egyetlen jellemzőként: határozza meg a kívánt eredményeket, tisztázza a feltételezéseket, és válassza szét azt, amit a rendszer megbízhatóan képes elvégezni, attól, ami még szakértői megítélést igényel.

A gyakorlatban a PagedAttentiont és a vLLM-et használó erős csapatok optimalizálják az architektúrát, az adatokat és az infrastruktúrát a megbízhatóság és a költségek szempontjából. Dokumentálják az explicit sikerkritériumokat, tesztelik a valósághű adatokat és munkafolyamatokat, és a megfigyelt hibaminták alapján iterálnak, nem pedig egyszeri benchmark győzelmek alapján. Ez az a hely, ahol az elméleti megértés tartós képességgé válik a termék, a politika és a műveletek között.

Az építészeti döntések évekig növelik a teljesítményt és a működési költségeket. Ugyanakkor az egyik benchmark optimalizálása elrejtheti a rendszer általános gyengeségeit. A legrugalmasabb megközelítés a kísérleti sebesség és az irányítási fegyelem kombinálása: kísérleti kísérletek futtatása, bizonyítékok rögzítése, döntési naplók közzététele és a biztosítékok folyamatos frissítése a modell viselkedésének, a felhasználói elvárásoknak és a szabályozási követelményeknek megfelelően.

Stratégiai hatás

Az építészeti döntések évekig növelik a teljesítményt és a működési költségeket.

Az építészeti döntések évekig növelik a teljesítményt és a működési költségeket. A kiváló minőségű telepítéseknél ez mérhető működési szabályokká, tulajdonosi határokká és ismétlődő felülvizsgálati rituálékká alakul át, így a csapatok növelhetik a bizalmat a kétértelműség skálázása helyett.

A technikai oktatás segít a csapatoknak a megfelelő verem kiválasztásában, nem csak a legújabb készletben.

A technikai oktatás segít a csapatoknak a megfelelő verem kiválasztásában, nem csak a legújabb készletben. A kiváló minőségű telepítéseknél ez mérhető működési szabályokká, tulajdonosi határokká és ismétlődő felülvizsgálati rituálékká alakul át, így a csapatok növelhetik a bizalmat a kétértelműség skálázása helyett.

A jobb mérnöki döntések csökkentik a termelés megbízhatósági incidenseit.

A jobb mérnöki döntések csökkentik a termelés megbízhatósági incidenseit. A kiváló minőségű telepítéseknél ez mérhető működési szabályokká, tulajdonosi határokká és ismétlődő felülvizsgálati rituálékká alakul át, így a csapatok növelhetik a bizalmat a kétértelműség skálázása helyett.

A PagedAttention és a vLLM jövője

A vLLM alapértelmezett nyílt forráskódú következtetési gerincvé vált, és a PagedAttention ötletei a legtöbb kiszolgálóveremben megjelennek. Mélyebb előtag-gyorsítótárazásra (a gyorsítótárazott rendszerkérdések újrafelhasználása a felhasználók között), a különálló gépeken lebontott előtöltésre és dekódolásra, intelligensebb kilakoltatási szabályzatokra, valamint a kvantálás és a spekulatív dekódolás szoros integrációjára számíthat. Ahogy a kontextusablakok több millió tokenekké nőnek, a hatékony lapozott KV-kezelés még központibbá válik a kiszolgálás megfizethető megőrzése szempontjából.

Valós megvalósítás

Nyílt forráskódú LLM API tárolása, ahol a vLLM sok egyidejű csevegőfelhasználót szolgál ki egyetlen GPU-ról nagy átviteli sebességgel

Hosszú rendszerparancs megosztása több ezer kérelem között az előtag gyorsítótárazásával, így egyszeri feldolgozásra kerül, nem ismétlődően

Futósugár-keresés vagy több mintavételezett befejezés, amelyek megosztják a KV-blokkokat a közös prompthoz másolás-írás útján

A GPU-memória-pazarlás csökkentése a töredezettségből, hogy a szolgáltató több egyidejű munkamenetet csomagolhasson ugyanarra a hardverre

Megvalósítási minták

PagedAttention és vLLM a gyakorlatban

Nyílt forráskódú LLM API hosztolása, ahol a vLLM sok egyidejű csevegőfelhasználót szolgál ki egyetlen GPU-ról nagy átviteli sebességgel.

Nyílt forráskódú LLM API hosztolása, ahol a vLLM sok egyidejű csevegőfelhasználót szolgál ki egyetlen GPU-ról nagy átviteli sebességgel. A csapatok általában jobb eredményeket érnek el, ha előre meghatározzák a minőségi küszöböket, fenntartják az emberi eszkalációs útvonalat a szélsőséges esetekhez, és nyomon követik a termelékenység növekedését és a hibaköltségeket is.

PagedAttention és vLLM a gyakorlatban

Hosszú rendszerkérdések megosztása több ezer kérelem között az előtag gyorsítótárazásával, így az egyszeri feldolgozásra kerül, nem pedig ismételten.

Hosszú rendszerparancs megosztása több ezer kérés között az előtag gyorsítótárazásával, így az egyszeri feldolgozásra kerül, nem pedig ismétlődően. A csapatok általában jobb eredményeket érnek el, ha előre meghatározzák a minőségi küszöbértékeket, emberi eszkalációs útvonalat tartanak a szélsőséges eseteknél, és nyomon követik a termelékenység növekedését és a hibaköltségeket az idő múlásával.

PagedAttention és vLLM a gyakorlatban

Futósugár-keresés vagy több mintavételezett befejezés, amelyek megosztják a KV-blokkokat a közös prompthoz másolás-írás útján.

Futósugár-keresés vagy többszörös mintavételezett befejezések, amelyek megosztják a KV-blokkokat a közös prompthoz másolás-írás-írás útján. A csapatok általában jobb eredményeket érnek el, ha előre meghatározzák a minőségi küszöbértékeket, megtartják az emberi eszkalációs útvonalat a szélsőséges eseteknél, és nyomon követik a termelékenység növekedését és a hibaköltségeket az idő múlásával.

PagedAttention és vLLM a gyakorlatban

A GPU-memória-pazarlás csökkentése a töredezettség miatt, hogy a szolgáltató több egyidejű munkamenetet is csomagolhasson ugyanarra a hardverre.

A GPU-memória-pazarlás csökkentése annak érdekében, hogy a szolgáltató több egyidejű munkamenetet csomagolhasson ugyanarra a hardverre. A csapatok általában jobb eredményeket érnek el, ha előre meghatározzák a minőségi küszöbértékeket, megtartják az emberi eszkalációs útvonalat a szélsőséges eseteknél, és nyomon követik a termelékenység növekedését és a hibaköltségeket is.

Kockázatok és védőkorlátok

!

Egy benchmark optimalizálása elrejtheti a rendszer általános hiányosságait.

!

Az infrastrukturális és karbantartási költségeket gyakran alábecsülik.

!

A biztonsági és megfigyelhetőségi hiányosságok a rendszerek bonyolultabbá válásával nőhetnek.

Végrehajtási ütemterv

1

A megvalósítás előtt határozza meg a késleltetési, minőségi és költségcélokat.

A megvalósítás előtt határozza meg a késleltetési, minőségi és költségcélokat. Minden lépést bizonyítékkapuként kell kezelni: ha a feltételek nem teljesülnek, szüneteltesse a közzétételt, zárja be a rést, és csak ezután bővítse a felhasználást.

2

Benchmark reális terhelési és adatviszonyok mellett.

Benchmark reális terhelési és adatviszonyok mellett. Minden lépést bizonyítékkapuként kell kezelni: ha a feltételek nem teljesülnek, szüneteltesse a közzétételt, zárja be a rést, és csak ezután bővítse a felhasználást.

3

Műszerfigyelés a hibák, az eltolódás és a felhasználói hatások szempontjából.

Műszerfigyelés a hibák, az eltolódás és a felhasználói hatások szempontjából. Minden lépést bizonyítékkapuként kell kezelni: ha a feltételek nem teljesülnek, szüneteltesse a közzétételt, zárja be a rést, és csak ezután bővítse a felhasználást.

4

A méretezés előtt készítse elő a visszagörgetési és az incidensre adott válaszútvonalakat.

A méretezés előtt készítse elő a visszagörgetési és az incidensre adott válaszútvonalakat. Minden lépést bizonyítékkapuként kell kezelni: ha a feltételek nem teljesülnek, szüneteltesse a közzétételt, zárja be a rést, és csak ezután bővítse a felhasználást.

Folytassa a felfedezést