Keyless Attention: Un Nuovo Meccanismo per Ridurre del 50% il Sovraccarico della KV-Cache
Un nuovo meccanismo di attenzione chiamato Keyless Attention è stato introdotto dai ricercatori, che elimina le rappresentazioni delle chiavi implementando una proiezione specializzata di routing nello spazio dei valori. Questa innovazione porta a una Value-Only Cache, riducendo del 50% la memoria e il sovraccarico di accesso della KV-cache rispetto ai metodi di attenzione tradizionali, migliorando al contempo il throughput di decodifica. Test condotti su cinque modelli in quattro architetture diverse indicano che Keyless Attention eguaglia o supera l'attenzione QKV standard in termini di perplexity per quattro dei cinque modelli, e dimostra prestazioni solide nei benchmark di valutazione downstream.
Fatti principali
- Keyless Attention elimina le rappresentazioni delle chiavi dal calcolo dell'attenzione.
- Introduce una proiezione dedicata di routing nello spazio dei valori.
- Value-Only Cache riduce del 50% la memoria e il sovraccarico di accesso della KV-cache.
- Migliora il throughput di decodifica rispetto all'attenzione standard.
- Testato su cinque modelli e quattro architetture.
- Eguaglia o supera l'attenzione QKV standard in perplexity su quattro dei cinque modelli.
- Prestazioni competitive nei benchmark di valutazione downstream.
- Pubblicato su arXiv con ID 2606.21848.
Entità
—