arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2502.03884cs.LGcs.AI

秩也很重要:LLM微调中适配器专家混合的分层配置

Rank Also Matters: Hierarchical Configuration for Mixture of Adapter Experts in LLM Fine-Tuning

Peking University(北京大学)

Peizhuang Cong, Wenpu Liu, Wenhan Yu, Haochen Zhao, Tong Yang

AI总结:

针对LLM微调中现有LoRA-MoE方法忽略适配器秩的问题,本文提出HILO分层方案,动态调整各层适配器专家的数量与秩,在更少可训练参数下取得更高准确率。

摘要:

大型语言模型(LLMs)已在各类任务中展现出显著成功,同时其参数规模持续增长。参数高效微调(PEFT)方法,如低秩适配(LoRA),通过大幅减少可训练参数数量来应对LLM微调的挑战。近期研究已将LoRA与专家混合(MoE)架构相结合,利用多个适配器专家和门控机制进一步提升微调性能。然而,现有方法主要侧重于调整每层适配器专家的分配以优化引入的可训练参数规模,却忽略了适配器秩这一关键因素。为此,我们提出了一种专家分配与秩配置的分层方案HILO,该方案在适配器粒度上动态调整各层适配器专家的数量和秩,以匹配模型层不同的表示复杂度。在多个基准任务上的大量实验表明,HILO在引入更少可训练参数的同时,在准确率上优于现有方法,为LLM微调提供了一种高效且实用的解决方案。

英文摘要:

Large language models (LLMs) have demonstrated remarkable success across various tasks, accompanied by a continuous increase in their parameter size. Parameter-efficient fine-tuning (PEFT) methods, such as Low-Rank Adaptation (LoRA), address the challenges of fine-tuning LLMs by significantly reducing the number of trainable parameters. Recent studies have integrated LoRA with Mixture of Experts (MoE) architectures, leveraging multiple adapter experts and gating mechanisms to further improve fine-tuning performance. However, existing approaches primarily focus on adjusting the allocations of adapter experts per layer to optimize the introduced trainable parameter size, while neglecting a critical factor of adapters' rank. To this end, we propose a hierarchical scheme for expert allocation and rank configuration, HILO, which dynamically adjusts the number and rank of adapter experts across layers, matching the varying representational complexity of model layers in adapter-granularity. Extensive experiments on multiple benchmark tasks demonstrate that HILO outperforms existing methods in accuracy while introducing fewer trainable parameters, providing an efficient and practical solution for fine-tuning LLMs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑