用分层张量专家提升微调效率,让小模型也能处理复杂任务。
TuckA: Hierarchical Compact Tensor Experts for Efficient Fine-Tuning
- 用张量分解构建多专家结构,参数量随专家数增长仍很紧凑。
- 在多个任务上性能接近全量微调,且比传统方法更适应数据多样性。
- 适合需要高效微调的场景,尤其适合资源受限的部署环境。
在基础模型时代,高效微调预训练模型是关键挑战。参数高效微调(PEFT)通过仅更新每层少量适配权重,实现与全量微调相当的性能。传统PEFT方法依赖单一适配器,其低秩矩阵难以捕捉复杂任务中数据的多样性。为此,本文提出Tucker适配(TuckA),具备四大特性:(i) 使用Tucker分解构建紧凑3D张量,每切片自然作为专家,低秩结构使参数量随专家数增长仍高效;(ii) 引入分层策略,将专家按不同粒度组织,兼顾局部与全局数据模式;(iii) 设计批量级路由机制,相比逐层路由,路由器参数量减少L倍(L为适配层数量);(iv) 提出数据感知初始化,基于理论分析实现零损失的专家负载均衡。在自然语言理解、图像分类和数学推理等多个基准上实验验证了TuckA的有效性,为PEFT提供了新而高效的解决方案。
原文摘要 · Abstract (English)
Efficiently fine-tuning pre-trained models for downstream tasks is a key challenge in the era of foundation models. Parameter-efficient fine-tuning (PEFT) presents a promising solution, achieving performance comparable to full fine-tuning by updating only a small number of adaptation weights per layer. Traditional PEFT methods typically rely on a single expert, where the adaptation weight is a low-rank matrix. However, for complex tasks, the data's inherent diversity poses a significant challenge for such models, as a single adaptation weight cannot adequately capture the features of all samples. To address this limitation, we explore how to integrate multiple small adaptation experts into a compact structure to defeat a large adapter. Specifically, we propose Tucker Adaptation (TuckA), a method with four key properties: (i) We use Tucker decomposition to create a compact 3D tensor where each slice naturally serves as an expert. The low-rank nature of this decomposition ensures that the number of parameters scales efficiently as more experts are added. (ii) We introduce a hierarchical strategy that organizes these experts into groups at different granularities, allowing the model to capture both local and global data patterns. (iii) We develop an efficient batch-level routing mechanism, which reduces the router's parameter size by a factor of $L$ compared to routing at every adapted layer (where $L$ is the number of adapted layers) (iv) We propose data-aware initialization to achieve loss-free expert load balancing based on theoretical analysis. Extensive experiments on benchmarks in natural language understanding, image classification, and mathematical reasoning speak to the efficacy of TuckA, offering a new and effective solution to the PEFT problem.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。