arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2211.08682cs.CL
收藏

预训练语言模型的层归一化参数有效调整

Parameter-Efficient Tuning on Layer Normalization for Pre-trained Language Models

Zhejiang Lab(之江实验室);Beihang University(北京航空航天大学)

Wang Qi, Yu-Ping Ruan, Yuan Zuo, Taihao Li

摘要:

由于现有预训练语言模型的规模,传统的语言模型微调面临越来越大的困难,使得参数有效调整成为前沿研究的热点。该领域中的先前方法将可调适配器添加到Transformer块的MHA或/和FFN中以使PLM实现可转移性。然而,作为Transformer体系结构的重要组成部分,层归一化对参数有效调整的作用被忽略了。本文首次提出了LN调谐方法,通过仅用0. 03%的参数对层归一化模块的增益和偏置项进行调谐,具有很高的时间效率,明显优于基线的0. 1%的可调参数。进一步,我们研究了将LN调优与以前的调优相结合的统一框架,发现:(1)将前缀调优、基于适配器的MHA调优方法和LN调优相结合的统一框架实现了SOTA性能。(2)同时调整MHA和LayerNorm的统一框架可以获得性能的改善,而同时调整FFN和LayerNorm的统一框架会导致性能下降。烧蚀研究证实了激光调光的参数不丰富,并对激光调光有了进一步的认识。

英文摘要:

Conventional fine-tuning encounters increasing difficulties given the size of current Pre-trained Language Models, which makes parameter-efficient tuning become the focal point of frontier research. Previous methods in this field add tunable adapters into MHA or/and FFN of Transformer blocks to enable PLMs achieve transferability. However, as an important part of Transformer architecture, the power of layer normalization for parameter-efficent tuning is ignored. In this paper, we first propose LN-tuning, by tuning the gain and bias term of Layer Normalization module with only 0.03\% parameters, which is of high time-efficency and significantly superior to baselines which are less than 0.1\% tunable parameters. Further, we study the unified framework of combining LN-tuning with previous ones and we find that: (1) the unified framework of combining prefix-tuning, the adapter-based method working on MHA, and LN-tuning achieves SOTA performance. (2) unified framework which tunes MHA and LayerNorm simultaneously can get performance improvement but those which tune FFN and LayerNorm simultaneous will cause performance decrease. Ablation study validates LN-tuning is of no abundant parameters and gives a further understanding of it.

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

↑