СледваСледващо ръководство
Диференциална поверителност
технически
Техническо РЪКОВОДСТВО
PagedAttention manages transformer key/value caches using fixed-size blocks mapped from logical token positions.
Чуйте това ръководствоAI-разказан · ~2 min
This reduces cache-allocation waste and supports sharing, while vLLM’s actual throughput depends on the workload and system.
Когато езиков модел генерира текст, той запазва „KV кеш“ (вектори на ключ и стойност) за всеки токен, който е видял, така че следващият токен да може да се погрижи за пълния контекст. Традиционно всяка заявка запазваше една голяма непрекъсната плоча GPU памет, оразмерена за максималната възможна дължина, като се губеха огромни количества, когато последователностите бяха по-къси или с различна дължина. PagedAttention, въведен в доклада за vLLM от 2023 г. от Калифорнийския университет в Бъркли, заимства идеята за пейджинг на виртуална памет от операционните системи: той разделя KV кеша на блокове с фиксиран размер, които могат да живеят навсякъде в паметта и да бъдат разпределени при поискване. Справочна таблица картографира позициите на логически токени към физически блокове. Това почти елиминира фрагментирането на паметта и позволява споделянето на блокове, например между множество изходи от една и съща подкана.
Хостинг на LLM API с отворен код, където vLLM обслужва много едновременни потребители на чат от един GPU при висока пропускателна способност
Споделяне на дълга системна подкана в хиляди заявки чрез кеширане на префикси, така че да се обработва веднъж, а не многократно
Търсене с бягащ лъч или множество извадки завършвания, които споделят KV блокове за общата подкана чрез копиране при запис
Намаляване на загубата на GPU памет от фрагментация, така че доставчикът да може да пакетира повече едновременни сесии на един и същ хардуер
vLLM се превърна в основа за изводи с отворен код по подразбиране и идеите на PagedAttention вече се появяват в повечето стекове за обслужване. Очаквайте по-задълбочено кеширане на префикси (повторно използване на кеширани системни подкани между потребителите), дезагрегирано предварително попълване и декодиране на отделни машини, по-интелигентни политики за изваждане и тясна интеграция с квантуване и спекулативно декодиране. Тъй като контекстните прозорци се разрастват в милиони токени, ефективното странично управление на KV става още по-централно за поддържане на достъпното обслужване.
Безплатен бюлетин
Три проверени AI истории всяка делнична сутрин, написани на достъпен език. Безплатно завинаги, без реклами.
Един имейл всеки делничен ден. Отписване с едно кликване. Никога не продаваме и не споделяме вашия имейл адрес.
Тествайте себе си
Незабавна обратна връзка за всеки отговор и сертификат за споделяне с подлежащ на проверка идентификационен номер, след като преминете курс.
Подкрепете безплатното AI образование. AI Understanding е 501(c)(3) с нестопанска цел — без реклами, без платен достъп, никога. Направете дарение
PagedAttention manages transformer key/value caches using fixed-size blocks mapped from logical token positions. This reduces cache-allocation waste and supports sharing, while vLLM’s actual throughput depends on the workload and system.
KV кешът съдържа вектори на ключове и стойности за всеки предишен токен, позволявайки на модела да се погрижи за пълния контекст, без да го изчислява отново на всяка стъпка.
PagedAttention заема страниране на виртуална памет: KV кешът е разделен на страници с фиксиран размер, които могат да живеят навсякъде, картографирани от блокова таблица.
Резервирането на една голяма непрекъсната плоча на заявка, оразмерена за максимална дължина, губи огромни количества GPU памет. Странирането разпределя малки блокове при поискване, намалявайки отпадъците.
Идентични префикси (споделени подкани или клонове за търсене на лъч) препращат към едни и същи физически KV страници, като се копират само когато клон се отклонява.
vLLM и алгоритъмът PagedAttention излязоха от UC Berkeley в документ от 2023 г. и бързо се превърнаха в широко използвана машина за обслужване с отворен код.
Продължавай да учиш
Още ръководства, избрани за тази тема
СледваСледващо ръководство
Диференциална поверителност
технически