RWKV-7 "Goose" con Evoluzione Dinamica Espressiva dello StatoRWKV-7 "Goose" with Expressive Dynamic State Evolution
Presentiamo RWKV-7 "Goose", una nuova architettura per la modellazione di sequenze, insieme a modelli linguistici pre-addestrati che stabiliscono un nuovo stato dell'arte nelle prestazioni downstream su scala di 3 miliardi di parametri per task multilingue, eguagliando le attuali prestazioni di punta per la lingua inglese nonostante siano stati addestrati su un numero significativamente inferiore di token rispetto ad altri modelli top da 3B. Tuttavia, i modelli RWKV-7 richiedono solo un utilizzo di memoria costante e un tempo di inferenza costante per token. RWKV-7 introduce una nuova formulazione generalizzata della regola delta con gate vettoriali e tassi di apprendimento in-context, oltre a una regola rilassata di sostituzione dei valori. Dimostriamo che RWKV-7 è in grado di eseguire il tracciamento dello stato e riconoscere tutti i linguaggi regolari, mantenendo al contempo la parallelizzabilità dell'addestramento. Ciò supera le capacità dei Transformer sotto le congetture di complessità standard, che sono limitati a TC^0. Per dimostrare la capacità di modellazione linguistica di RWKV-7, presentiamo anche un corpus multilingue open source esteso da 3,1 trilioni di token, e addestriamo quattro modelli RWKV-7 con un numero di parametri compreso tra 0,19 miliardi e 2,9 miliardi su questo dataset. Per promuovere apertura, riproducibilità e adozione, rilasciamo i nostri modelli e l'elenco dei componenti del dataset su https://huggingface.co/RWKV, e il nostro codice di addestramento e inferenza su https://github.com/RWKV/RWKV-LM, tutto sotto licenza Apache 2.0.