Kimi K2 - modelo de linguagem Mixture-of-Experts (MoE) de ponta
(github.com/MoonshotAI)- Kimi K2 da Moonshot AI é um modelo de linguagem Mixture-of-Experts (MoE) de ponta com 1 trilhão de parâmetros
- Introduz a técnica de otimização Muon no processo de treinamento para resolver problemas de estabilidade em larga escala
- Foca em inteligência agêntica para uso de ferramentas, raciocínio e resolução autônoma de problemas
- Comprova desempenho de alto nível em programação, matemática e tarefas gerais em diversos benchmarks
- É conveniente para implantação e uso, com API compatível com OpenAI/Anthropic e suporte flexível a ambientes de execução
Por que o Kimi K2 é importante
- Kimi K2 é o mais recente modelo de linguagem Mixture-of-Experts (MoE) desenvolvido pela Moonshot AI
- Com escala de 1 trilhão de parâmetros e um método inovador de otimização (Muon), oferece alto desempenho e estabilidade no campo dos grandes modelos de linguagem
- Em comparação com modelos open source de alto desempenho já existentes, alcança nível SOTA global e o melhor patamar entre open source em várias aplicações práticas, como programação, matemática e uso de ferramentas
- Destaca-se tanto na questão de treinar modelos grandes com rapidez e estabilidade quanto na flexibilidade para dar suporte a diversos cenários de uso de inteligência artificial
1. Introdução ao modelo
- Kimi K2 é um modelo de linguagem MoE de ponta com 1 trilhão (1T) de parâmetros totais e 32 bilhões (32B) de parâmetros ativos
- Usa o otimizador Muon para resolver de forma eficaz a instabilidade no treinamento de modelos em larga escala
- É especializado em capacidades de alto nível, como uso de ferramentas, raciocínio complexo e agentes autônomos
Principais características
- Treinamento em larga escala: o modelo de 1 trilhão de parâmetros foi pré-treinado com 15,5 trilhões de tokens, sem instabilidade durante o treinamento
- Otimizador MuonClip: garante estabilidade ao combinar o algoritmo Muon, voltado para modelos de grande escala, com uma nova técnica de otimização
- Agentic Intelligence: projetado com foco em uso de ferramentas, raciocínio complexo e resolução autônoma de problemas
Tipos de modelo
- Kimi-K2-Base: modelo base adequado para fine-tuning customizado e uso por pesquisadores
- Kimi-K2-Instruct: modelo pós-treinamento (post-training) otimizado para chat e execução de agentes gerais
2. Resumo do modelo
- Arquitetura: Mixture-of-Experts (MoE)
- Parâmetros totais: 1 trilhão (1.000.000.000.000)
- Parâmetros ativos: 32 bilhões (32B)
- Número de camadas: 61 (incluindo Dense layer)
- Número de Dense Layers: 1
- Attention hidden dimension: 7168
- MoE hidden dimension (por especialista): 2048
- Attention Head: 64
- Número de especialistas: 384
- Número de especialistas selecionados por token: 8
- Número de especialistas compartilhados: 1
- Tamanho do vocabulário: 160K
- Comprimento de contexto: 128K
- Mecanismo de Attention: MLA
- Função de ativação: SwiGLU
3. Resultados de avaliação
Desempenho do modelo Instruction
- Registra desempenho de alto nível em diversos benchmarks, incluindo tarefas de programação, uso de ferramentas, matemática/ciências exatas e tarefas gerais
- Em várias categorias de código e ferramentas, matemática e lógica, e tarefas gerais — como SWE-bench, LiveCodeBench, OJBench, MultiPL-E, TerminalBench, AceBench, Tau2, AIME e MATH-500 — mostra desempenho SOTA ou no nível dos melhores da categoria
- Registra pass@1 de 65,8% no SWE-bench Verified e 47,3% no SWE-bench Multilingual, com resultados especialmente marcantes também em ambientes de Agentic Coding
- Mostra excelente precisão também em testes de exatas, como MATH-500, AIME, HMMT e CNMO
- Garante desempenho superior em diversas tarefas gerais, como MMLU e SimpleQA, em comparação com modelos open source e comerciais concorrentes
Desempenho do modelo Base
- Registra resultados de topo entre modelos open source da mesma categoria em benchmarks representativos como MMLU, TriviaQA e GPQA-Diamond
- Garante vantagem geral em relação a grandes modelos base open source em programação, matemática e avaliação em chinês
4. Implantação e execução em engines
- A API do Kimi K2 (compatível com OpenAI/Anthropic) pode ser usada em https://platform.moonshot.ai
- Os checkpoints do modelo (block-fp8) também são disponibilizados no Huggingface (https://huggingface.co/moonshotai/Kimi-K2-Instruct)
- Engines de inferência recomendadas: vLLM, SGLang, KTransformers, TensorRT-LLM e outras, com compatibilidade para diversos ambientes
5. Exemplos de uso do modelo
Interface de chat
- Após executar o serviço local de inferência, é possível interagir diretamente por meio de um cliente compatível com OpenAI, como a Chat Completions API
- temperature recomendado: 0.6; também é recomendado usar o prompt de sistema na forma padrão
Recurso de chamada de ferramentas
- O Kimi-K2-Instruct tem forte capacidade de chamada de ferramentas (tool-calling)
- Se o usuário fornecer a lista de ferramentas disponíveis em cada solicitação, o modelo decide de forma autônoma quando usar e executar as ferramentas
- É possível demonstrar exemplos e mensagens de resultado ao longo de todo o pipeline
- É necessário suporte da engine à lógica de parsing de ferramentas do Kimi-K2
6. Licença
- Tanto o código quanto os pesos do modelo são distribuídos como open source sob a Modified MIT License
1 comentários
Comentários do Hacker News