Le côté obscur de la lune et la nouvelle publication de Tsinghua : la pré-remplissage LLM peut traverser plusieurs centres de données, le débit du modèle de 1T augmente de 54 %

robot
Création du résumé en cours
ME News Actualités, le 18 avril (UTC+8), selon la surveillance de Dongcha Beating, Moonshot AI et l'Université Tsinghua ont publié le 16 avril sur arXiv un nouveau papier intitulé « Prefill-as-a-Service », proposant de faire fonctionner la phase de pré-remplissage (prefill) de l'inférence de grands modèles à travers plusieurs centres de données.
L'inférence de grands modèles se divise en deux étapes : le prefill lit une fois toutes les entrées et génère un cache KV ; le décode utilise ensuite ce cache pour produire le résultat mot par mot.
Les caractéristiques matérielles nécessaires pour ces deux étapes sont totalement différentes : le prefill consomme beaucoup de puissance de calcul, le décode nécessite une grande mémoire vidéo et une large bande passante.
La pratique courante dans l'industrie consiste à séparer ces deux étapes sur des machines différentes (séparation PD), mais cela exige une interconnexion RDMA dans le même centre de données, car le cache KV des modèles d'attention intensive peut atteindre plusieurs dizaines de Gbps par seconde, et si la transmission est lente, le GPU reste inactif.
Le tournant vient des nouveaux modèles d'attention hybride.
Les travaux expérimentaux montrent que des modèles comme Kimi Linear, MiMo-V2-Flash, Ring-2.5-1T, en combinant quelques couches d'attention complètes avec de nombreuses couches linéaires, ont réduit le débit du cache KV d'environ un ordre de grandeur, avec un ratio de compression global de 36 fois pour Ring-2.5-1T.
À ce moment-là, le cache KV peut être transféré du réseau privé RDMA vers un réseau Ethernet standard pour le décode.
La méthode spécifique de PrfaaS consiste à constituer un « cluster de pré-remplissage » dédié, qui ne route que les requêtes avec de longs contextes ou des préfixes non trouvés, tandis que les requêtes courtes restent dans le cluster PD local ;
une fois le pré-remplissage terminé, le cache KV est renvoyé via Ethernet au cluster local pour le décode.
Elle introduit également un routage basé sur un seuil de longueur, un ordonnanceur sensible à la bande passante et un pool de cache de préfixes hybrides.
Les expérimentations ont été réalisées avec un modèle hybride interne de 1T paramètres (basé sur l'architecture Kimi Linear), montrant que la capacité de service globale dépasse de 54 % celle d'une déploiement PD homogène, et de 32 % par rapport à une solution hétérogène naïve, tout en utilisant une bande passante inter-centre modérée par machine.
(Source : BlockBeats)
Voir l'original
Cette page peut inclure du contenu de tiers fourni à des fins d'information uniquement. Gate ne garantit ni l'exactitude ni la validité de ces contenus, n’endosse pas les opinions exprimées, et ne fournit aucun conseil financier ou professionnel à travers ces informations. Voir la section Avertissement pour plus de détails.
  • Récompense
  • 8
  • 1
  • Partager
Commentaire
Ajouter un commentaire
Ajouter un commentaire
PerpMoodSwing
· Il y a 3h
1T paramètres, une amélioration de 54 %, l'équipe de Tsinghua est vraiment hardcore
Voir l'originalRépondre0
GateUser-57ab9c02
· Il y a 9h
Une demande courte pour rester local est très raisonnable, évitant de faire toute une histoire.
Voir l'originalRépondre0
Cream-ColoredCross-ChainBridge
· Il y a 9h
L'hétérogénéité naïve peut être dépassée de 32 %, la différence d'infrastructure de base est grande
Voir l'originalRépondre0
CandleSitter
· Il y a 9h
PD séparation atteint de nouveaux sommets
Voir l'originalRépondre0
MidnightReconciler
· Il y a 9h
L'aspect obscur de la lune, cette série de sorties techniques est acceptable
Voir l'originalRépondre0
WatercolorGlassBottle
· Il y a 9h
Explication de la planification sensible à la bande passante, ils ont vraiment évité des pièges
Voir l'originalRépondre0
MistValleySignpost
· Il y a 9h
Comment le mécanisme de synchronisation du décodage de retour est-il conçu ?
Voir l'originalRépondre0
SeaSaltMintCandy
· Il y a 9h
Je me sens particulièrement à l'aise dans les scénarios de contexte long
Voir l'originalRépondre0
  • Épinglé