通过正交训练提升低秩适配专家模型的多样性,显著减少专家数量同时保持性能。
OMoE: Diversifying Mixture of Low-Rank Adaptation by Orthogonal Finetuning
- 采用格拉姆-施密特过程强制专家表示正交,提升模型多样性。
- 在多个常识推理数据集上,仅用少量专家即实现稳定性能提升。
- 方法轻量高效,适合资源受限场景下的参数高效微调应用。
为低秩适配(LoRA)构建混合专家(MoE)架构是参数高效微调(PEFT)的一个新兴方向,具有模块化设计和优异性能潜力。然而,简单增加专家数量并不能保证显著提升。本文通过定性分析发现,原始MoE中专家会坍缩为相似表示,限制了模块化能力和计算效率。进一步分析表明,此前MoE变体的性能受限于专家间缺乏多样性。为此,我们提出正交混合专家(OMoE),一种资源高效的MoE变体,通过正交方式训练专家以促进多样性。在OMoE中,采用格拉姆-施密特过程,强制专家表示位于Stiefel流形上。该方法直接作用于架构,不改变学习目标,且不损害最优性。实验在多个常识推理基准上验证,相比当前最佳方法,OMoE能以显著更少的专家数实现稳定、高效的性能提升。
原文摘要 · Abstract (English)
Building mixture-of-experts (MoE) architecture for Low-rank adaptation (LoRA) is emerging as a potential direction in parameter-efficient fine-tuning (PEFT) for its modular design and remarkable performance. However, simply stacking the number of experts cannot guarantee significant improvement. In this work, we first conduct qualitative analysis to indicate that experts collapse to similar representations in vanilla MoE, limiting the capacity of modular design and computational efficiency. Ulteriorly, Our analysis reveals that the performance of previous MoE variants maybe limited by a lack of diversity among experts. Motivated by these findings, we propose Orthogonal Mixture-of-Experts (OMoE), a resource-efficient MoE variant that trains experts in an orthogonal manner to promote diversity. In OMoE, a Gram-Schmidt process is leveraged to enforce that the experts' representations lie within the Stiefel manifold. By applying orthogonal constraints directly to the architecture, OMoE keeps the learning objective unchanged, without compromising optimality. Our method is simple and alleviates memory bottlenecks, as it incurs minimal experts compared to vanilla MoE models. Experiments on diverse commonsense reasoning benchmarks demonstrate that OMoE can consistently achieve stable and efficient performance improvement when compared with the state-of-the-art methods while significantly reducing the number of required experts.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。