边缘设备上高效微调大模型,仅更新不到0.1%参数仍保持性能。
Task-Aware Parameter-Efficient Fine-Tuning of Large Pre-Trained Models at the Edge
- 根据权重与激活值计算参数重要性,精准识别关键参数。
- 任务参数均匀分布,更新量<0.1%时仍保持下游任务性能。
- 兼容稀疏加速与LoRA,适合资源受限的边缘场景。
大规模语言模型(LLMs)在决策、推理和问答等任务中表现卓越,已被广泛部署于边缘设备。然而,在边缘端对LLM进行特定任务微调面临计算成本高、存储与能源资源有限的挑战。为此,我们提出TaskEdge——一种面向边缘设备的任务感知参数高效微调框架,通过仅更新任务相关参数来优化资源利用。首先设计融合权重与输入激活的参数重要性评估准则;其次提出模型无关的任务特定参数分配算法,确保参数在模型中均匀分布而非集中于局部区域。实验表明,TaskEdge可将更新参数比例控制在0.1%以下,显著降低计算开销与内存占用,同时维持下游任务性能。此外,该框架可无缝集成结构化稀疏以利用NVIDIA专用稀疏张量核心实现加速,并兼容LoRA实现高效的稀疏低秩适应。
原文摘要 · Abstract (English)
Large language models (LLMs) have achieved remarkable success in various tasks, such as decision-making, reasoning, and question answering. They have been widely used in edge devices. However, fine-tuning LLMs to specific tasks at the edge is challenging due to the high computational cost and the limited storage and energy resources at the edge. To address this issue, we propose TaskEdge, a task-aware parameter-efficient fine-tuning framework at the edge, which allocates the most effective parameters to the target task and only updates the task-specific parameters. Specifically, we first design a parameter importance calculation criterion that incorporates both weights and input activations into the computation of weight importance. Then, we propose a model-agnostic task-specific parameter allocation algorithm to ensure that task-specific parameters are distributed evenly across the model, rather than being concentrated in specific regions. In doing so, TaskEdge can significantly reduce the computational cost and memory usage while maintaining performance on the target downstream tasks by updating less than 0.1\% of the parameters. In addition, TaskEdge can be easily integrated with structured sparsity to enable acceleration by NVIDIA's specialized sparse tensor cores, and it can be seamlessly integrated with LoRA to enable efficient sparse low-rank adaptation. Extensive experiments on various tasks demonstrate the effectiveness of TaskEdge.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。