根据重要性动态选层,让低秩适配更省资源
FLoE: Fisher-Based Layer Selection for Efficient Sparse Adaptation of Low-Rank Experts
- 用费舍尔信息评估各层重要性,只在关键层部署适配器
- 自动分配最优低秩参数,无需人工调参和网格搜索
- 适合算力有限、需快速适配的场景
参数高效微调(PEFT)已成为将预训练大语言模型适配到下游任务的主流方法,显著降低内存与计算开销。然而,现有多数PEFT技术在所有层均匀部署LoRA适配器,忽视了各层贡献差异与任务特异性秩需求,导致参数冗余,适应效率欠佳。为此,我们提出FLoE,引入两项创新:(i) 基于费舍尔信息的重要性评分机制,动态识别MoE架构中任务关键的Transformer层,实现适配器稀疏部署;(ii) 基于贝叶斯优化的秩分配器,自动确定特定数据集上的最优LoRA秩,避免耗时的网格搜索。在多种LLM与基准测试上的实验表明,FLoE实现了出色的效率-精度权衡,在资源受限环境中尤其适用。
原文摘要 · Abstract (English)
Parameter-Efficient Fine-Tuning (PEFT) methods have emerged as a widely adopted strategy for adapting pre-trained Large Language Models (LLMs) to downstream tasks, significantly reducing memory and computational costs. However, most existing PEFT techniques uniformly deploy LoRA adapters across all layers, disregarding the intrinsic heterogeneity of layer contributions and task-specific rank requirements. This uniform paradigm leads to redundant parameter allocation and suboptimal adaptation efficiency. To address these limitations, we propose FLoE, a novel PEFT framework that introduces two key innovations: (i) a Fisher information-guided importance scoring mechanism to dynamically identify task-critical transformer layers for MoE-based low-rank adaptation, enabling sparse adapter deployment; and (ii) a Bayesian optimization-driven rank allocator that automatically determines optimal LoRA ranks on specific datasets without exhaustive grid search. Extensive experiments across diverse LLMs and benchmarks reveal that FLoE achieves impressive efficiency-accuracy trade-offs, making FLoE particularly advantageous in resource-constrained environments that necessitate rapid adaptation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。