发现大模型越复杂性能反而先降后升,零大小模型反而是最佳平衡点。
Investigating the Impact of Model Complexity in Large Language Models
- 用隐马尔可夫模型建模自回归大模型,分析复杂度与泛化的关系。
- 在头微调下,模型复杂度增加时风险先升后降,出现双下降现象。
- 理论结合实验验证,适合研究模型泛化与复杂度的学者参考。
基于预训练-微调范式的大型语言模型(LLMs)在自然语言处理任务中表现卓越,但其模型复杂度对微调性能的影响尚缺乏理论理解。本文聚焦自回归型LLM,提出使用隐马尔可夫模型(HMM)进行建模,并研究复杂度与下游任务泛化能力的关系。以冻结全部预训练参数、仅训练头部的头微调范式为例,理论分析表明:风险随模型复杂度先上升后下降,呈现‘双下降’现象;初期下降为退化情形,意味着偏差与方差平衡的‘甜点’出现在模型规模为零时。研究克服了对自回归模型及下游任务的有效建模和多变量回归风险分析等挑战,通过基于HMM生成的数据实验,提供了实证支持并验证了理论洞察。
原文摘要 · Abstract (English)
Large Language Models (LLMs) based on the pre-trained fine-tuning paradigm have become pivotal in solving natural language processing tasks, consistently achieving state-of-the-art performance. Nevertheless, the theoretical understanding of how model complexity influences fine-tuning performance remains challenging and has not been well explored yet. In this paper, we focus on autoregressive LLMs and propose to employ Hidden Markov Models (HMMs) to model them. Based on the HMM modeling, we investigate the relationship between model complexity and the generalization capability in downstream tasks. Specifically, we consider a popular tuning paradigm for downstream tasks, head tuning, where all pre-trained parameters are frozen and only individual heads are trained atop pre-trained LLMs. Our theoretical analysis reveals that the risk initially increases and then decreases with rising model complexity, showcasing a "double descent" phenomenon. In this case, the initial "descent" is degenerate, signifying that the "sweet spot" where bias and variance are balanced occurs when the model size is zero. Obtaining the presented in this study conclusion confronts several challenges, primarily revolving around effectively modeling autoregressive LLMs and downstream tasks, as well as conducting a comprehensive risk analysis for multivariate regression. Our research is substantiated by experiments conducted on data generated from HMMs, which provided empirical support and alignment with our theoretical insights.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。