HyperAIHyperAI

Command Palette

Search for a command to run...

Au-delà de l'attention homogène : des LLMs à faible consommation mémoire grâce au cache KV approximé par Fourier

Résumé

Les grands modèles de langage (Large Language Models) peinent à faire face aux exigences de mémoire croissantes liées au cache clé-valeur (KV) lorsque la longueur du contexte s'accroît. Les méthodes de compression existantes homogénéisent les dimensions des têtes ou s'appuient sur l'élagage guidé par l'attention, souvent au détriment de la précision ou en introduisant un surcoût computationnel. Nous proposons FourierAttention, un cadre sans apprentissage qui exploite les rôles hétérogènes des dimensions des têtes des transformateurs : les dimensions inférieures privilégient le contexte local, tandis que les dimensions supérieures capturent les dépendances à longue portée. En projetant les dimensions insensibles au contexte long sur des bases de Fourier orthogonales, FourierAttention approxime leur évolution temporelle avec des coefficients spectraux de longueur fixe. Les évaluations sur les modèles LLaMA montrent que FourierAttention atteint la meilleure précision pour le contexte long sur LongBench et Needle-In-A-Haystack (NIAH). De plus, un noyau Triton personnalisé, FlashFourierAttention, a été conçu pour optimiser la mémoire grâce à des opérations de lecture-écriture simplifiées, permettant une mise en œuvre efficace sans compromettre les performances.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp