arXiv:2606.16243cs.LGcs.CL2026-06被引 3

用线性规划控制过拟合,让Transformer微调更精准高效。

LiFT: Local Search via Linear Programming for Overfitting-Controlled Transformers

论文配图:LiFT: Local Search via Linear Programming for Overfitting-Controlled Transformers
图 1 · 摘自论文原文
  • 通过线性规划构建验证感知的下降方向,联合优化参数与正则化超参。
  • 在WikiText-2上测试困惑度显著降低,尤其在易过拟合场景提升明显。
  • 适合需要精细控制过拟合的模型微调任务,如小样本或高复杂度数据。

本文提出一种基于线性规划(LP)的局部搜索框架——LiFT,用于对预训练Transformer模型进行微调,并显式控制过拟合。该方法将微调建模为双层优化正则化问题,联合更新模型参数与正则化超参数。利用初始预热迭代中收集的验证梯度和训练海森信息,求解一个线性规划问题,以最小化缩放后的方向导数并保持训练最优性,从而生成聚焦的局部更新方向。该方向使参数与正则化超参数同时得到有针对性的调整,避免重复完整重训练周期。实验表明,在GPT-2 Small于WikiText-2上的微调中,LiFT通过选择性地调整变换器块与正则化参数,显著提升了测试困惑度,且在多种层数配置与正则化设置下表现一致,尤其在过拟合敏感场景下优势突出。此外,LiFT建立了Transformer微调与双层优化、局部搜索及正则化理论之间的系统性联系。

原文摘要 · Abstract (English)

This paper proposes a Linear Programming (LP)-based local search framework for fine-tuning pretrained transformer models with explicit control against overfitting. The approach formulates transformer fine-tuning as a bilevel optimization-based regularization problem, in which model parameters and regularization hyperparameters are jointly updated. Information collected during initial warm-up iterations, including validation gradients and training Hessian information, is used to construct a local descent direction by solving an LP that minimizes a scaled directional derivative while preserving training optimality. This validation-aware descent direction enables focused local updates of both parameters and regularization hyperparameters, reducing overfitting without requiring repeated full retraining cycles. The resulting method, termed Linear Programming-based Fine-Tuning (LiFT) for transformers, differs from conventional fine-tuning by systematically identifying task-specific updates rather than relying on heuristic or grid-based hyperparameter selection. Experiments on GPT-2 Small fine-tuned on WikiText-2 demonstrate that LiFT enables effective adaptation through selective tuning of transformer blocks and regularization parameters, yielding consistent improvements in test perplexity across multiple layer configurations and regularization settings, with particularly pronounced gains in overfitting-prone scenarios. Beyond empirical performance, LiFT establishes a principled connection between transformer fine-tuning, bilevel optimization, local search, and regularization theory.

Transformer微调过拟合控制线性规划

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。