arXiv:2507.07024cs.CLcs.AI2025-07NeurIPS被引 14

FlexOlmo让模型在不共享数据的情况下训练和灵活使用,支持按需调用私有数据。

FlexOlmo: Open Language Models for Flexible Data Use

  • 采用专家混合架构,各专家独立训练于封闭数据集,通过新路由机制集成。
  • 370亿参数模型在31项任务上平均提升41%,优于已有融合方法10.1%。
  • 适合需要保护数据隐私的机构,支持推理时细粒度控制数据访问。

我们提出FlexOlmo,一种新型语言模型,支持(1)无需数据共享的分布式训练,不同模型参数可在封闭数据集上独立训练;(2)推理阶段灵活选择是否使用特定数据及其对应参数,无需重新训练。FlexOlmo采用专家混合(MoE)架构,每个专家在封闭数据集上独立训练,后续通过新的领域感知路由机制整合,无须联合训练。模型在自建的FlexMix语料库上训练,包含公开数据集与七个领域专用数据集,近似模拟封闭数据集。我们在31个下游任务上评估了最大达370亿参数(活跃参数200亿)的模型。结果表明,公共数据训练的一般专家可与多方独立训练的专家有效结合,平均相对性能提升41%,同时用户可根据数据许可或授权要求选择排除特定数据。该方法平均比现有模型融合技术高出10.1%,且在相同训练浮点运算量下超越标准无数据限制的MoE模型。本研究为受监管行业中的数据所有者和研究人员提供了兼顾数据隐私与性能提升的解决方案。

原文摘要 · Abstract (English)

We introduce FlexOlmo, a new class of language models (LMs) that supports (1) distributed training without data sharing, where different model parameters are independently trained on closed datasets, and (2) data-flexible inference, where these parameters along with their associated data can be flexibly included or excluded from model inferences with no further training. FlexOlmo employs a mixture-of-experts (MoE) architecture where each expert is trained independently on closed datasets and later integrated through a new domain-informed routing without any joint training. FlexOlmo is trained on FlexMix, a corpus we curate comprising publicly available datasets alongside seven domain-specific sets, representing realistic approximations of closed sets. We evaluate models with up to 37 billion parameters (20 billion active) on 31 diverse downstream tasks. We show that a general expert trained on public data can be effectively combined with independently trained experts from other data owners, leading to an average 41% relative improvement while allowing users to opt out of certain data based on data licensing or permission requirements. Our approach also outperforms prior model merging methods by 10.1% on average and surpasses the standard MoE trained without data restrictions using the same training FLOPs. Altogether, this research presents a solution for both data owners and researchers in regulated industries with sensitive or protected data. FlexOlmo enables benefiting from closed data while respecting data owners' preferences by keeping their data local and supporting fine-grained control of data access during inference.

语言模型数据隐私专家混合

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。