Chiosa
Machine Learning

QUILT accélère le prefill des longs contextes en partageant le travail des requêtes

2 min ·

En termes simples

QUILT est une méthode informatique qui accélère l’attention éparse dans les modèles de langage à long contexte. Au lieu de traiter chaque requête séparément, elle regroupe les requêtes proches, charge une seule fois les entrées clé-valeur qu’elles ont sélectionnées en commun, puis les réutilise.

La méthode organise aussi ce partage pour l’adapter aux tuiles de calcul des accélérateurs et chevauche la préparation des données avec le calcul de l’attention. Dans les tests sur GLM-5.3 et DeepSeek-3.2, la plus forte réduction rapportée pour la latence du noyau d’attention éparse était de 55,1 % sur quatre tâches à long contexte, tandis que les écarts absolus moyens de score sur 21 jeux de données restaient faibles.

La recherche

À mesure que les fenêtres de contexte s’allongent, le prefill doit traiter davantage de données clé-valeur sélectionnées avant l’apparition du premier jeton.

On ne sait pas encore clairement si les requêtes voisines partagent suffisamment de travail pour qu’un noyau puisse le réutiliser, au lieu de charger, convertir et calculer plusieurs fois les mêmes entrées.

QUILT regroupe les requêtes éparses voisines et réutilise leurs entrées clé-valeur communes, réduisant ainsi la latence du noyau et le délai avant le premier jeton.

Les auteurs ont comparé QUILT à OPS-Transformer, un noyau optimisé d’attention éparse, dans le même environnement de service XYServe.

Ils ont testé GLM-5.3 et DeepSeek-3.2 sur quatre tâches à long contexte avec le parallélisme de tenseurs et de séquences, puis comparé leurs scores officiels sur 21 jeux de données.

QUILT a réduit le nombre d’entrées clé-valeur traitées, ainsi que la latence du noyau et le délai avant le premier jeton de sortie. Pour GLM-5.3, la latence moyenne du noyau a baissé de 53,8 % avec le parallélisme de tenseurs sur les quatre tâches, tandis que l’écart absolu moyen de score sur 21 jeux de données était de 0,42 point.

Ensemble, ces résultats étayent un prefill plus rapide dans les tests, avec peu de variation mesurée des scores.

Les résultats se limitent à GLM-5.3 et DeepSeek-3.2, aux tâches LongBench, à quatre charges de travail pour les performances et la latence avant le premier jeton, à 21 jeux de données pour la précision, ainsi qu’à un banc d’essai équipé de 16 NPU Ascend 910C.

Ils ne permettent pas d’établir que les mêmes gains seraient obtenus avec d’autres modèles, tâches, configurations, conditions de service ou accélérateurs.

Analyse

Détails

Article suivantSoftmaxLift reduces key retrieval for long-context attention