Kimi K2 - modelo de lenguaje Mixture-of-Experts (MoE) de vanguardia
(github.com/MoonshotAI)- Kimi K2 de Moonshot AI es un modelo de lenguaje Mixture-of-Experts (MoE) de vanguardia con 1 billón de parámetros
- Durante el entrenamiento, incorporó la técnica de optimización Muon para resolver problemas de estabilidad a gran escala
- Se enfoca en la inteligencia agéntica para el uso de herramientas, razonamiento y resolución autónoma de problemas
- Demuestra rendimiento de primer nivel en programación, matemáticas y tareas generales en diversos benchmarks
- Su despliegue y uso son convenientes, con API compatible con OpenAI/Anthropic y soporte flexible de motores
Por qué importa Kimi K2
- Kimi K2 es el más reciente modelo de lenguaje Mixture-of-Experts (MoE) desarrollado por Moonshot AI
- Con una escala de 1 billón de parámetros y un método de optimización innovador (Muon), ofrece alto rendimiento y estabilidad en el ámbito de los modelos de lenguaje de gran escala
- Frente a modelos open source de alto rendimiento ya existentes, logra nivel de vanguardia global (SOTA) y el mejor nivel open source en varios campos de aplicación reales como programación, matemáticas y uso de herramientas
- Destaca por su capacidad para entrenar modelos grandes de forma rápida y estable, así como por su flexibilidad para soportar diversos escenarios de uso de inteligencia artificial
1. Introducción al modelo
- Kimi K2 es un modelo de lenguaje MoE de vanguardia con 1 billón (1T) de parámetros totales y 32 mil millones (32B) de parámetros activos
- Utiliza el optimizador Muon para resolver de manera efectiva la inestabilidad del entrenamiento de modelos a gran escala
- Está especializado en capacidades de alto nivel como uso de herramientas, razonamiento complejo y agentes autónomos
Características principales
- Entrenamiento a gran escala: el modelo de 1 billón de parámetros fue preentrenado con 15.5 billones de tokens, sin inestabilidad durante el entrenamiento
- Optimizador MuonClip: asegura estabilidad al combinar el algoritmo Muon, especializado para modelos a gran escala, con nuevas técnicas de optimización
- Agentic Intelligence: fue diseñado pensando en el uso de herramientas, el razonamiento complejo y la resolución autónoma de problemas
Tipos de modelo
- Kimi-K2-Base: modelo base adecuado para fine-tuning personalizado y uso por investigadores
- Kimi-K2-Instruct: modelo post-training optimizado para chat y operación de agentes generales
2. Resumen del modelo
- Arquitectura: Mixture-of-Experts (MoE)
- Parámetros totales: 1 billón (1,000,000,000,000)
- Parámetros activos: 32 mil millones (32B)
- Número de capas: 61 (incluye Dense layer)
- Número de Dense Layer: 1
- Attention hidden dimension: 7168
- MoE hidden dimension (por experto): 2048
- Attention Head: 64
- Número de expertos: 384
- Número de expertos seleccionados por token: 8
- Número de expertos compartidos: 1
- Tamaño de vocabulario: 160K
- Longitud de contexto: 128K
- Mecanismo de Attention: MLA
- Función de activación: SwiGLU
3. Resultados de evaluación
Rendimiento del modelo Instruction
- Registra rendimiento de primer nivel en diversos benchmarks de tareas de programación, uso de herramientas, matemáticas/ciencias e ingeniería y tareas generales
- Muestra rendimiento SOTA o de máximo nivel equivalente en áreas de código y herramientas, matemáticas y lógica, y tareas generales, incluyendo SWE-bench, LiveCodeBench, OJBench, MultiPL-E, TerminalBench, AceBench, Tau2, AIME y MATH-500
- Logró 65.8% de pass@1 en SWE-bench Verified y 47.3% en SWE-bench Multilingual, con resultados sobresalientes también en entornos de Agentic Coding
- Ofrece precisión sobresaliente en pruebas de matemáticas y ciencias e ingeniería como MATH-500, AIME, HMMT y CNMO
- También obtiene rendimiento superior frente a modelos open source y comerciales competidores en varias tareas generales como MMLU y SimpleQA
Rendimiento del modelo Base
- Registra resultados de primer nivel entre modelos open source comparables en benchmarks representativos como MMLU, TriviaQA y GPQA-Diamond
- Asegura una ventaja general frente a grandes modelos base open source en programación, matemáticas y evaluación en chino
4. Despliegue y ejecución en motores
- La API de Kimi K2 (compatible con OpenAI/Anthropic) está disponible en https://platform.moonshot.ai
- También ofrece checkpoints del modelo (block-fp8) en Huggingface (https://huggingface.co/moonshotai/Kimi-K2-Instruct)
- Motores de inferencia recomendados: vLLM, SGLang, KTransformers, TensorRT-LLM y otros entornos compatibles
5. Ejemplos de uso del modelo
Interfaz de chat
- Tras ejecutar un servicio de inferencia local, es posible interactuar directamente desde un cliente compatible con OpenAI (como la Chat Completions API)
- temperature recomendada: 0.6; también se recomienda usar el prompt de System en su forma predeterminada
Función de llamada de herramientas
- Kimi-K2-Instruct tiene una potente capacidad de tool-calling
- Si el usuario envía una lista de herramientas disponibles en cada solicitud, el modelo decide de forma autónoma cuándo usar y ejecutar las herramientas
- Es posible demostrar ejemplos y mensajes de resultado a lo largo de todo el pipeline
- Se requiere soporte de la lógica de parsing de herramientas de Kimi-K2 en el motor
6. Licencia
- Tanto el código como los pesos del modelo se distribuyen como open source bajo la Modified MIT License
1 comentarios
Comentarios de Hacker News