构建统一基准LoRALib,评估LoRA-MoE方法在40个任务上的表现。
LoRALib: A Standardized Benchmark for Evaluating LoRA-MoE Methods
- 标准化40个下游任务数据集与17种模型架构的训练流程。
- 发现LoRAMoE在3种方法中表现最优,任务相关LoRA优先选择更有效。
- 开源了680个LoRA模块和测试工具,助力公平比较。
作为参数高效微调(PEFT)方法,低秩适配(LoRA)能显著降低存储与计算成本,但其对单一任务强适应性常伴随跨任务泛化能力不足。为改善此问题,现有工作将LoRA与专家混合(MoE)结合,通过专家模块与路由机制增强模型适应性。然而,现有LoRA-MoE方法在模型、数据集、超参数及评估方式上缺乏统一标准,难以进行公平比较。为此,我们提出统一基准LoRALib:将40个下游任务的数据集统一格式,使用相同超参数微调出680个LoRA模块,覆盖17种模型架构。基于该LoRA库,我们在公开测试工具OpenCompass上对3种代表性LoRA-MoE方法及不同LoRA选择机制进行了大规模实验。结果表明,LoRAMoE表现最佳,且优先选择与目标任务相关的LoRA可进一步提升性能。研究结果有望推动后续工作。数据集与模型库已开源于HuggingFace。
原文摘要 · Abstract (English)
As a parameter efficient fine-tuning (PEFT) method, low-rank adaptation (LoRA) can save significant costs in storage and computing, but its strong adaptability to a single task is often accompanied by insufficient cross-task generalization capabilities. To improve this, existing work combines LoRA with mixture-of-experts (MoE) to enhance the model's adaptability through expert modules and routing mechanisms. However, existing LoRA-MoE methods lack unified standards in models, datasets, hyperparameters, and evaluation methods, making it difficult to conduct fair comparisons between different methods. To this end, we proposed a unified benchmark named LoRALib. Specifically, we standardized datasets from $40$ downstream tasks into a unified format, fine-tuned them using the same hyperparameters and obtained $680$ LoRA modules across $17$ model architectures. Based on this LoRA library, we conduct large-scale experiments on $3$ representative LoRA-MoE methods and different LoRA selection mechanisms using the open-sourced testing tool OpenCompass. Extensive experiments show that LoRAMoE performs best, and that prioritizing LoRAs relevant to the target task can further improve the performance of MoE. We hope these findings will inspire future work. Our datasets and LoRA library are available at https://huggingface.co/datasets/YaoLuzjut/LoRAOcean_dataset and https://huggingface.co/YaoLuzjut/models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。