通过测试时重混专家路径,显著提升MoE大模型的准确率。
C3PO: Critical-Layer, Core-Expert, Collaborative Pathway Optimization for Test-Time Expert Re-Mixing
- 基于样本邻近性设计代理目标,动态优化关键层核心专家权重。
- 在6个基准上提升7-15%准确率,使1-3B激活参数模型超越7-9B通用模型。
- 计算高效,仅优化关键层核心专家,适合资源受限场景部署。
Mixture-of-Experts (MoE) 大语言模型在测试时存在路径选择不优的问题——预训练学习的简单专家选择导致高达10-20%的准确率提升空间。为此,我们提出一种新型测试时优化方法,针对每个测试样本联合重加权不同层的专家,实现“再混”(re-mixing)。由于测试样本真实标签未知,我们设计基于参考样本中“成功邻居”的代理目标,包括基于模式查找、核回归及相似样本平均损失三种算法。为降低计算开销,仅在关键层的核心专家上优化混合权重,性能接近全路径优化但效率显著提升。该方法称为「Critical-Layer, Core-Expert, Collaborative Pathway Optimization (C3PO)」。我们在两个近期MoE LLM上验证,覆盖六个主流基准,结果一致显示准确率提升7-15%,远超上下文学习、提示/前缀调优等常用测试时学习基线。更重要的是,1-3B活跃参数的MoE模型经优化后可超越7-9B参数的普通模型,进一步强化了MoE在效率上的优势。全面消融实验揭示了提升测试时性能的新机制。
原文摘要 · Abstract (English)
Mixture-of-Experts (MoE) Large Language Models (LLMs) suffer from severely sub-optimal expert pathways-our study reveals that naive expert selection learned from pretraining leaves a surprising 10-20% accuracy gap for improvement. Motivated by this observation, we develop a novel class of test-time optimization methods to re-weight or "re-mixing" the experts in different layers jointly for each test sample. Since the test sample's ground truth is unknown, we propose to optimize a surrogate objective defined by the sample's "successful neighbors" from a reference set of samples. We introduce three surrogates and algorithms based on mode-finding, kernel regression, and the average loss of similar reference samples/tasks. To reduce the cost of optimizing whole pathways, we apply our algorithms merely to the core experts' mixing weights in critical layers, which enjoy similar performance but save significant computation. This leads to "Critical-Layer, Core-Expert, Collaborative Pathway Optimization (C3PO)". We apply C3PO to two recent MoE LLMs and examine it on six widely-used benchmarks. It consistently improves the base model by 7-15% in accuracy and outperforms widely used test-time learning baselines, e.g., in-context learning and prompt/prefix tuning, by a large margin. Moreover, C3PO enables MoE LLMs with 1-3B active parameters to outperform LLMs of 7-9B parameters, hence improving MoE's advantages on efficiency. Our thorough ablation study further sheds novel insights on achieving test-time improvement on MoE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。