Kimi K2 - Modèle de langage MoE de pointe
(github.com/MoonshotAI)- Kimi K2 de Moonshot AI est un modèle de langage Mixture-of-Experts (MoE) de pointe doté de 1 trillion de paramètres
- Il adopte la méthode d’optimisation Muon durant l’entraînement pour résoudre les problèmes de stabilité à grande échelle
- Il met l’accent sur l’intelligence agentique pour l’usage d’outils, le raisonnement et la résolution autonome de problèmes
- Il démontre des performances de premier plan dans divers benchmarks en codage, mathématiques et tâches générales
- Il est pratique à déployer et exploiter, avec une API compatible OpenAI/Anthropic et un environnement prenant en charge des moteurs flexibles
Pourquoi Kimi K2 est important
- Kimi K2 est le dernier modèle de langage Mixture-of-Experts (MoE) développé par Moonshot AI
- Avec une échelle de 1 trillion de paramètres et une méthode d’optimisation innovante (Muon), il offre de hautes performances et une grande stabilité dans le domaine des grands modèles de langage
- Comparé aux modèles open source haute performance existants, il atteint l’état de l’art mondial (SOTA) et le meilleur niveau open source dans divers cas d’usage réels comme le codage, les mathématiques et l’utilisation d’outils
- Il se distingue par sa capacité à entraîner rapidement et de manière stable de très grands modèles, ainsi que par sa flexibilité pour prendre en charge divers scénarios d’utilisation de l’IA
1. Présentation du modèle
- Kimi K2 est un modèle de langage MoE de pointe avec 1 trillion (1T) de paramètres au total et 32 milliards (32B) de paramètres actifs
- Il utilise l’optimiseur Muon pour résoudre efficacement l’instabilité de l’entraînement des modèles à grande échelle
- Il est spécialisé dans des capacités de haut niveau comme l’usage d’outils, le raisonnement complexe et les agents autonomes
Caractéristiques principales
- Entraînement à grande échelle : le modèle à 1 trillion de paramètres a été préentraîné sur 15,5 trillions de tokens, sans instabilité durant l’apprentissage
- Optimiseur MuonClip : il combine l’algorithme Muon, spécialisé pour les grands modèles, avec de nouvelles techniques d’optimisation afin d’assurer la stabilité
- Agentic Intelligence : conçu en pensant à l’usage d’outils, au raisonnement complexe et à la résolution autonome de problèmes
Types de modèles
- Kimi-K2-Base : modèle de base adapté au fine-tuning personnalisé et à l’usage par les chercheurs
- Kimi-K2-Instruct : modèle post-entraînement (post-training) optimisé pour le chat et l’exécution d’agents généralistes
2. Résumé du modèle
- Architecture : Mixture-of-Experts (MoE)
- Paramètres totaux : 1 trillion (1,000,000,000,000)
- Paramètres actifs : 32 milliards (32B)
- Nombre de couches : 61 (avec couche Dense)
- Nombre de couches Dense : 1
- Attention hidden dimension : 7168
- MoE hidden dimension (par expert) : 2048
- Attention Head : 64
- Nombre d’experts : 384
- Nombre d’experts sélectionnés par token : 8
- Nombre d’experts partagés : 1
- Taille du vocabulaire : 160K
- Longueur de contexte : 128K
- Mécanisme d’attention : MLA
- Fonction d’activation : SwiGLU
3. Résultats d’évaluation
Performances du modèle Instruction
- Il enregistre des performances de premier plan sur divers benchmarks couvrant les tâches de codage, l’usage d’outils, les mathématiques/sciences et les tâches générales
- Sur SWE-bench, LiveCodeBench, OJBench, MultiPL-E, TerminalBench, AceBench, Tau2, AIME, MATH-500 et d’autres évaluations de code/outils, mathématiques/logique et tâches générales, il atteint des performances SOTA ou au meilleur niveau de sa catégorie
- Il enregistre 65,8 % de pass@1 sur SWE-bench Verified et 47,3 % sur SWE-bench Multilingual, avec des résultats particulièrement remarquables en environnement de codage agentique
- Il affiche aussi une excellente précision sur des tests scientifiques et mathématiques comme MATH-500, AIME, HMMT et CNMO
- Sur diverses tâches générales comme MMLU (culture générale) et SimpleQA, il obtient des performances supérieures à celles de modèles open source et commerciaux concurrents
Performances du modèle Base
- Sur des benchmarks de référence comme MMLU, TriviaQA et GPQA-Diamond, il obtient des scores parmi les meilleurs de sa catégorie parmi les modèles open source
- Il prend globalement l’avantage sur les grands modèles de base open source en codage, mathématiques et évaluations en chinois
4. Déploiement et exécution sur moteurs
- L’API Kimi K2 (compatible OpenAI/Anthropic) est disponible sur https://platform.moonshot.ai
- Les checkpoints du modèle (block-fp8) sont également pris en charge sur Huggingface (https://huggingface.co/moonshotai/Kimi-K2-Instruct)
- Moteurs d’inférence recommandés : vLLM, SGLang, KTransformers, TensorRT-LLM et autres environnements compatibles
5. Exemples d’utilisation du modèle
Interface de chat
- Après avoir lancé un service d’inférence local, il est possible d’interagir directement depuis un client compatible OpenAI (comme l’API Chat Completions)
- Température recommandée : 0.6, et il est également recommandé d’utiliser le prompt System dans sa forme par défaut
Fonction d’appel d’outils
- Kimi-K2-Instruct dispose d’une puissante capacité d’appel d’outils (tool-calling)
- À chaque requête, l’utilisateur peut fournir la liste des outils disponibles, et le modèle décide de manière autonome s’il faut les utiliser et à quel moment les exécuter
- Il est possible de démontrer des exemples et des messages de résultat sur l’ensemble du pipeline
- Le moteur doit prendre en charge la logique de parsing des outils de Kimi-K2
6. Licence
- Le code comme les poids du modèle sont distribués en open source sous Modified MIT License
1 commentaires
Avis Hacker News