Mixture-of-Experts Meets Instruction Tuning: A Winning Combination for Large Language Models

828 篇文章

已下架不支持订阅

本文研究了Mixture-of-Experts(MoE)模型与指令微调结合在大型语言模型(LLM)中的效果。实验表明,MoE模型在指令微调后在多个任务上优于密集模型,特别是FLAN-MOE模型在减少计算成本的同时,性能超越了FLAN-PALM。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

本文是LLM系列的文章,针对《Mixture-of-Experts Meets Instruction Tuning: A Winning Combination for Large Language Models》的翻译。

摘要

稀疏混合专家(MoE)是一种神经架构设计,可用于在不增加推理成本的情况下向大型语言模型(LLM)添加可学习参数。指令调整是一种训练LLM遵循指令的技术。我们主张将这两种方法结合起来,因为我们发现MoE模型比密集模型从指令调整中受益更多。特别是,我们在三个实验设置中进行了实证研究:(i)在没有指令调整的情况下对单个下游任务进行直接微调;(ii)指令调整,然后对下游任务进行上下文小样本或零样本泛化;以及(iii)指令调整,辅以对个别下游任务的进一步微调。在第一种情况下,MoE模型总体表现不如具有相同计算能力的密集模型。然而,随着指令调整(第二和第三场景)的引入,这种叙述发生了巨大变化,指令调整独立使用或与特定任务的微调结合使用。我们最强大的型号FLAN-MOE(32B)在四个基准任务上的性能超过了FLAN-PALM(62B),而只使用了三分之一的FLOP。FLAN-MOE所体现的进步启发了在任务不可知学习的框架下重新评估大规模、高性能语言模型的设计原则。

1 引言

近年来,在越来越大和复杂的深度学习模型的发展推动下,自然语言处理(NLP)领域取得了显著进步。在这些模型中,基于Transformer的语言模型已经成为一系列NLP任务的事实标准,因为它们在捕捉复杂的语言模式和在不同上下文中进行概括方面具有无与伦比的能力。训练这种模型的一个特别成功的范例是指令调整,它通过调整它们预先训

已下架不支持订阅

### DeepSeekMoE 实现极致专家专业化 在 Mixture-of-Experts (MoE) 语言模型中,DeepSeekMoE 致力于实现专家的极致专业化。为了达到这一目标,该研究引入了一种新颖的方法来处理专家之间的知识共享和隔离问题。 #### 共享专家隔离机制 分配给不同专家的 token 可能需要一些共同的知识,这可能导致多个专家在其参数中收敛于相同的共享知识,进而造成专家参数的冗余[^2]。为了避免这种冗余并提高参数效率,DeepSeekMoE 设计了一个特殊的共享专家子集。这些共享专家始终处于激活状态,负责捕捉和整合跨上下文的通用知识。因此,其他路由专家可以专注于更特定的任务,减少了重复学习的可能性,提升了整体模型性能和资源利用效率。 #### 极致专家专业化策略 除了上述提到的共享专家外,DeepSeekMoE 还采用了多种技术手段促进各领域内专家的专业化程度: - **动态调整门控网络**:通过优化门控函数的设计,使得每个输入样本能够被最合适的少数几个专家处理,而不是平均分布到所有可用专家上; - **自适应训练方案**:根据不同阶段的学习需求灵活改变损失权重以及正则项强度等因素; - **多任务联合训练框架**:鼓励各个专精方向上的专家不仅要在单一目标任务上有出色表现,在辅助任务方面也需具备一定能力,以此增强泛化性和鲁棒性。 ```python def dynamic_gating_network(input_tensor, expert_weights): """ 动态调整门控网络示例 参数: input_tensor: 输入张量 expert_weights: 各位专家对应的权值向量 返回: selected_experts_indices: 被选中的几位专家索引列表 """ scores = tf.matmul(input_tensor, expert_weights) top_k_values, top_k_indices = tf.nn.top_k(scores, k=3) return top_k_indices.numpy().tolist() ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

UnknownBody

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值