arXiv:2505.09519cs.CL2025-05NeurIPS被引 7

将专家混合与矩阵分解结合,让提示调优更高效且通用。

PT-MoE: An Efficient Finetuning Framework for Integrating Mixture-of-Experts into Prompt Tuning

  • 用矩阵分解和专家路由增强提示调优,实现参数共享与动态适配。
  • 在17个数据集上超越现有方法,问答任务F1提升1.49点,数学准确率提高10.75点。
  • 适合需要高效跨任务泛化的模型微调场景,尤其关注低参数高性能应用。

参数高效微调(PEFT)方法在适配大语言模型方面展现出潜力,但现有方法存在反直觉现象:将路由模块引入提示调优(PT)可提升训练效率,却未必改善性能;通过矩阵分解减少参数可在特定领域提升表现。受此启发并基于PT的模块化特性,我们提出PT-MoE框架,将矩阵分解与专家混合(MoE)路由结合,实现高效提示调优。在17个数据集上的实验表明,PT-MoE在问答(QA)和数学问题求解任务中均达当前最优性能:相比PT,QA任务F1提升1.49点,数学准确率提升10.75点;相比LoRA,QA任务F1提升2.13点,数学准确率提升0.44点,同时参数量比LoRA减少25%。分析显示,尽管PT在问答任务中占优,而LoRA在数学任务中表现更好,但矩阵分解与MoE在PT-MoE中的融合带来互补优势:分解促进专家间高效参数共享,MoE实现动态适应,共同提升跨任务一致性与泛化能力。消融研究进一步揭示了路由机制与结构组件的作用,为未来PEFT方法设计提供启示。

原文摘要 · Abstract (English)

Parameter-efficient fine-tuning (PEFT) methods have shown promise in adapting large language models, yet existing approaches exhibit counter-intuitive phenomena: integrating router into prompt tuning (PT) increases training efficiency yet does not improve performance universally; parameter reduction through matrix decomposition can improve performance in specific domains. Motivated by these observations and the modular nature of PT, we propose PT-MoE, a novel framework that integrates matrix decomposition with mixture-of-experts (MoE) routing for efficient PT. Results across 17 datasets demonstrate that PT-MoE achieves state-of-the-art performance in both question answering (QA) and mathematical problem solving tasks, improving F1 score by 1.49 points over PT and 2.13 points over LoRA in QA tasks, while enhancing mathematical accuracy by 10.75 points over PT and 0.44 points over LoRA, all while using 25% fewer parameters than LoRA. Our analysis reveals that while PT methods generally excel in QA tasks and LoRA-based methods in math datasets, the integration of matrix decomposition and MoE in PT-MoE yields complementary benefits: decomposition enables efficient parameter sharing across experts while MoE provides dynamic adaptation, collectively enabling PT-MoE to demonstrate cross-task consistency and generalization abilities. These findings, along with ablation studies on routing mechanisms and architectural components, provide insights for future PEFT methods.

参数高效专家混合提示调优微调框架

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