让大模型同时精通科学与通用任务,避免知识遗忘。
Innovator: Scientific Continued Pretraining with Fine-grained MoE Upcycling
- 将通用模型细粒度拆分为学科专家+通用共享专家
- 3000亿token训练后科学任务平均提升25%,通用能力保留99%
- 适合需要跨学科推理的科研场景
具备科学与通用知识的大语言模型是科学通用智能的基础。然而,直接使用科学数据对大模型进行持续预训练常导致灾难性遗忘,严重损害通用能力。本文提出Innovator,通过在持续预训练过程中将预训练的密集模型升级为细粒度混合专家(MoE)模型,使不同专家学习不同学科的科学知识,同时保留一个共享专家处理通用任务。Innovator采用四阶段上链训练范式:(1) 领域特定数据上的科学专家诱导,(2) 通过前馈网络维度分解实现细粒度专家分裂,(3) 科学感知路由预热,(4) 混合数据集上的通才-专家融合训练。该范式实现通用知识与各科学领域的解耦,避免领域间负向干扰。模型总参数53.3B,激活参数13.3B,基于Qwen2.5-7B扩展,包含1个共享通用专家和64个专用科学专家(每次激活8个)。在300B token、三级质量控制数据上训练后,Innovator在30项科学任务上实现25%平均提升,胜率70%,同时保持99%通用任务性能。进一步微调的Innovator-Reason在复杂科学问题求解中表现优异,推理能力提升超30%。
原文摘要 · Abstract (English)
A large language model (LLM) with knowledge in both scientific and general tasks is the foundation of science general intelligence. However, directly continued pretraining an LLM using science data usually leads to catastrophic forgetting, which indicates severe degradation in general ability. In this report, we present Innovator, which solves this problem by upcycling a pre-trained dense LLM into a fine-grained Mixtures-of-Experts model during continued pretraining, where different experts are expected to learn science knowledge in different disciplines, and a shared expert is utilized for general tasks. Innovator introduces a four-stage upcycle training paradigm: (1) Scientific Expert Induction on discipline-specific data, (2) Fine-grained Expert Splitting via FFN dimension decomposition, (3) Science-Aware Routing warmup, and (4) Generalist-Scientist Integration training on hybrid datasets. Such a paradigm enables knowledge in the general domain, and different scientific disciplines can be decoupled, avoiding the negative influence among knowledge in different domains. With 53.3B total parameters and 13.3B activated, Innovator extends Qwen2.5-7B using a shared general expert and 64 specialized scientific experts with 8 activated. Trained on 300B tokens with tri-level quality-controlled data, Innovator achieves 25% average improvement across 30 scientific tasks with a win rate as 70%, while retaining 99% performance in general tasks. Furthermore, Innovator-Reason, which is post-trained from Innovator for reasoning boosting, exhibits excellent reasoning performance in solving complex scientific problems with improvements over 30%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。