通过系数调优扩展注意力特征空间,实现高效微调大模型。
Coeff-Tuning: A Graph Filter Subspace View for Tuning Attention-Based Large Models
- 将多头注意力映射为图卷积滤波子空间,仅训练少量系数。
- 在多个下游任务上超越基线方法,参数增量可忽略不计。
- 适合追求高效微调且希望保持模型性能的研究者。
基于Transformer的大规模预训练模型展现出强大的泛化能力,多种参数高效微调(PEFT)方法被提出以在下游任务上以极小的计算和内存开销定制模型。以往的PEFT方法主要从张量分解视角出发,试图通过寻找最小参数子集来有效调整线性变换。本文提出一种正交视角:将注意力操作表示为图卷积,并将多头注意力映射视为卷积滤波子空间,每个注意力映射作为子空间元素。我们提出通过学习一组少量组合系数,从原始多头注意力映射构建更具表达力的滤波子空间来微调大模型。理论上和实验上均证明,经过调优的滤波子空间能有效拓展多头注意力的特征空间,进一步提升Transformer容量。我们还通过残差参数化稳定微调过程,并通过在可调系数上直接应用丢弃法设计正则化以增强泛化能力。可调系数参数极少,且可与已有PEFT方法以即插即用方式结合。大量实验表明,本方法在性能上优于现有PEFT基线,附加参数可忽略不计。
原文摘要 · Abstract (English)
Transformer-based large pre-trained models have shown remarkable generalization ability, and various parameter-efficient fine-tuning (PEFT) methods have been proposed to customize these models on downstream tasks with minimal computational and memory budgets. Previous PEFT methods are primarily designed from a tensor-decomposition perspective that tries to effectively tune the linear transformation by finding the smallest subset of parameters to train. Our study adopts an orthogonal view by representing the attention operation as a graph convolution and formulating the multi-head attention maps as a convolutional filter subspace, with each attention map as a subspace element. In this paper, we propose to tune the large pre-trained transformers by learning a small set of combination coefficients that construct a more expressive filter subspace from the original multi-head attention maps. We show analytically and experimentally that the tuned filter subspace can effectively expand the feature space of the multi-head attention and further enhance the capacity of transformers. We further stabilize the fine-tuning with a residual parameterization of the tunable subspace coefficients, and enhance the generalization with a regularization design by directly applying dropout on the tunable coefficient during training. The tunable coefficients take a tiny number of parameters and can be combined with previous PEFT methods in a plug-and-play manner. Extensive experiments show that our approach achieves superior performances than PEFT baselines with neglectable additional parameters.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。