MiniMax-01用闪电注意力实现超长上下文建模,性能媲美GPT-4o。
MiniMax-01: Scaling Foundation Models with Lightning Attention
- 采用闪电注意力与专家混合架构,每令牌激活459亿参数
- 训练时支持百万级上下文,推理可达400万令牌,成本可控
- 适合需要超长文本处理的场景,如法律、科研文档分析
我们推出MiniMax-01系列,包括可处理超长上下文的MiniMax-Text-01和视觉语言模型MiniMax-VL-01。核心是闪电注意力及其高效扩展机制。通过结合混合专家(MoE)架构,构建了含32个专家、总计4560亿参数的模型,每令牌仅激活459亿参数。开发了优化的并行策略与高效计算-通信重叠技术,使数百亿参数模型在数百万令牌上下文中实现高效训练与推理。MiniMax-Text-01训练时上下文窗口达100万令牌,推理可外推至400万令牌,且成本低廉。MiniMax-VL-01基于5120亿视觉语言令牌持续训练。标准与自研基准测试表明,其性能媲美GPT-4o和Claude-3.5-Sonnet,上下文长度提升20-32倍。代码已公开于https://github.com/MiniMax-AI。
原文摘要 · Abstract (English)
We introduce MiniMax-01 series, including MiniMax-Text-01 and MiniMax-VL-01, which are comparable to top-tier models while offering superior capabilities in processing longer contexts. The core lies in lightning attention and its efficient scaling. To maximize computational capacity, we integrate it with Mixture of Experts (MoE), creating a model with 32 experts and 456 billion total parameters, of which 45.9 billion are activated for each token. We develop an optimized parallel strategy and highly efficient computation-communication overlap techniques for MoE and lightning attention. This approach enables us to conduct efficient training and inference on models with hundreds of billions of parameters across contexts spanning millions of tokens. The context window of MiniMax-Text-01 can reach up to 1 million tokens during training and extrapolate to 4 million tokens during inference at an affordable cost. Our vision-language model, MiniMax-VL-01 is built through continued training with 512 billion vision-language tokens. Experiments on both standard and in-house benchmarks show that our models match the performance of state-of-the-art models like GPT-4o and Claude-3.5-Sonnet while offering 20-32 times longer context window. We publicly release MiniMax-01 at https://github.com/MiniMax-AI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。