Kimi K2 - Mô hình ngôn ngữ Mixture-of-Experts (MoE) tiên tiến nhất
(github.com/MoonshotAI)- Kimi K2 của Moonshot AI là một mô hình ngôn ngữ Mixture-of-Experts (MoE) tiên tiến nhất với 1 nghìn tỷ tham số
- Trong quá trình huấn luyện, mô hình áp dụng kỹ thuật tối ưu hóa Muon để giải quyết các vấn đề ổn định ở quy mô lớn
- Tập trung vào trí tuệ tác tử cho sử dụng công cụ, suy luận và giải quyết vấn đề tự chủ
- Chứng minh hiệu năng hàng đầu trên nhiều benchmark ở các tác vụ lập trình, toán học và tác vụ tổng quát
- Triển khai và sử dụng thuận tiện, cung cấp API tương thích OpenAI/Anthropic và hỗ trợ môi trường engine linh hoạt
Vì sao Kimi K2 quan trọng
- Kimi K2 là mô hình ngôn ngữ Mixture-of-Experts (MoE) mới nhất do Moonshot AI phát triển
- Với quy mô 1 nghìn tỷ tham số và phương pháp tối ưu hóa đổi mới (Muon), mô hình mang lại hiệu năng và độ ổn định cao trong lĩnh vực mô hình ngôn ngữ lớn
- So với các mô hình mã nguồn mở hiệu năng cao hiện có, Kimi K2 đạt trình độ tiên tiến toàn cầu (SOTA) và thuộc nhóm tốt nhất trong giới mã nguồn mở ở nhiều lĩnh vực ứng dụng thực tế như lập trình, toán học và sử dụng công cụ
- Mô hình có thế mạnh ở khả năng huấn luyện nhanh và ổn định cho mô hình lớn, cũng như tính linh hoạt để hỗ trợ nhiều kịch bản ứng dụng AI khác nhau
1. Giới thiệu mô hình
- Kimi K2 là mô hình ngôn ngữ MoE tiên tiến nhất với 1 nghìn tỷ (1T) tham số tổng và 32 tỷ (32B) tham số hoạt động
- Sử dụng optimizer Muon để giải quyết hiệu quả tình trạng bất ổn trong huấn luyện mô hình quy mô lớn
- Được tối ưu cho các năng lực cấp cao như khai thác công cụ, suy luận phức tạp và tác tử tự chủ
Đặc điểm chính
- Huấn luyện quy mô lớn: tiền huấn luyện mô hình 1 nghìn tỷ tham số trên 15,5 nghìn tỷ token mà không gặp tình trạng mất ổn định trong huấn luyện
- Optimizer MuonClip: kết hợp thuật toán Muon chuyên cho mô hình quy mô lớn với các kỹ thuật tối ưu hóa mới để đảm bảo độ ổn định
- Agentic Intelligence: được thiết kế với mục tiêu sử dụng công cụ, suy luận phức tạp và giải quyết vấn đề tự chủ
Các phiên bản mô hình
- Kimi-K2-Base: mô hình nền tảng phù hợp cho fine-tuning tùy chỉnh và nhu cầu của nhà nghiên cứu
- Kimi-K2-Instruct: mô hình post-training được tối ưu cho trò chuyện và vận hành tác tử tổng quát
2. Tóm tắt mô hình
- Kiến trúc: Mixture-of-Experts (MoE)
- Tổng số tham số: 1 nghìn tỷ (1,000,000,000,000)
- Tham số hoạt động: 32 tỷ (32B)
- Số lớp: 61 (bao gồm Dense layer)
- Số Dense Layer: 1
- Attention hidden dimension: 7168
- MoE hidden dimension (mỗi expert): 2048
- Attention Head: 64
- Số lượng expert: 384
- Số expert được chọn trên mỗi token: 8
- Số expert chia sẻ: 1
- Kích thước từ vựng: 160K
- Độ dài ngữ cảnh: 128K
- Cơ chế Attention: MLA
- Hàm kích hoạt: SwiGLU
3. Kết quả đánh giá
Hiệu năng của mô hình Instruction
- Đạt hiệu năng hàng đầu trên nhiều benchmark ở các hạng mục như bài toán lập trình, sử dụng công cụ, toán học/STEM và tác vụ tổng quát
- Trên các benchmark như SWE-bench, LiveCodeBench, OJBench, MultiPL-E, TerminalBench, AceBench, Tau2, AIME, MATH-500, mô hình cho thấy hiệu năng SOTA hoặc tốt nhất cùng hạng ở các nhóm mã, công cụ, toán học, logic và tác vụ tổng quát
- Ghi nhận pass@1 65.8% trên SWE-bench Verified và 47.3% trên SWE-bench Multilingual, đồng thời thể hiện kết quả nổi bật trong môi trường Agentic Coding
- Đạt độ chính xác xuất sắc trong các bài kiểm tra STEM như MATH-500, AIME, HMMT và CNMO
- Cũng đạt hiệu năng cao hơn nhiều mô hình mã nguồn mở/thương mại cạnh tranh ở các tác vụ tổng quát như MMLU (kiến thức tổng quát) và SimpleQA
Hiệu năng của mô hình Base
- Ghi nhận kết quả thuộc nhóm cao nhất trong các mô hình mã nguồn mở cùng hạng trên những benchmark tiêu biểu như MMLU, TriviaQA và GPQA-Diamond
- Thể hiện ưu thế toàn diện so với các mô hình nền tảng mã nguồn mở lớn ở các bài đánh giá về lập trình, toán học và tiếng Trung
4. Triển khai và vận hành engine
- Có thể sử dụng API Kimi K2 (tương thích OpenAI/Anthropic) tại https://platform.moonshot.ai
- Checkpoint mô hình (block-fp8) cũng được hỗ trợ trên Huggingface (https://huggingface.co/moonshotai/Kimi-K2-Instruct)
- Các engine suy luận được khuyến nghị: vLLM, SGLang, KTransformers, TensorRT-LLM, tương thích với nhiều môi trường khác nhau
5. Ví dụ sử dụng mô hình
Giao diện trò chuyện
- Sau khi chạy dịch vụ suy luận cục bộ, có thể tương tác trực tiếp qua client tương thích OpenAI như Chat Completions API
- Temperature được khuyến nghị: 0.6, đồng thời cũng nên dùng System prompt ở dạng mặc định
Chức năng gọi công cụ
- Kimi-K2-Instruct có năng lực gọi công cụ (tool-calling) mạnh mẽ
- Người dùng chỉ cần truyền danh sách công cụ khả dụng cho từng yêu cầu, mô hình sẽ tự chủ quyết định việc sử dụng công cụ và thời điểm thực thi
- Có thể trình diễn ví dụ và thông điệp kết quả xuyên suốt toàn bộ pipeline
- Cần có hỗ trợ logic phân tích công cụ Kimi-K2 từ phía engine
6. Giấy phép
- Cả mã nguồn và trọng số mô hình đều được phát hành mã nguồn mở theo Modified MIT License
1 bình luận
Ý kiến trên Hacker News