arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.13151cs.CLcs.LGstat.ML

面向资源高效的语言模型预训练在线优化的多臂老虎机方法:动态掩码的应用案例

Multi-armed bandits for resource efficient, online optimization of language model pre-training: the use case of dynamic masking

Basque Center for Applied Mathematics(巴斯克应用数学中心);IKERBASQUE - Basque Foundation for Science(伊克巴斯克-巴斯克科学基金会);Columbia University(哥伦比亚大学);Walmart Global Tech(沃尔玛全球技术);eBay Inc.(易贝公司)

Iñigo Urteaga, Moulay-Zaïdane Draïdia, Tomer Lancewicki, Shahram Khadivi

AI总结:

本文提出基于高斯过程汤普森采样的多臂老虎机框架GP-TS,通过顺序选择动态掩码超参数加速Transformer语言模型预训练,在更少轮数下降低MLM损失并保持下游性能,避免昂贵网格搜索,实现资源高效优化。

摘要:

我们设计并评估了一个贝叶斯优化框架,用于基于Transformer的语言模型(TLMs)的资源高效预训练。TLM预训练需要高计算资源,并引入许多尚未解决的设计选择,例如选择其预训练超参数。我们提出一个多臂老虎机框架,用于TLM预训练超参数的顺序选择,旨在以资源高效的方式优化语言模型性能。我们设计了一个汤普森采样算法,以掩码语言模型(MLM)预训练目标的高斯过程代理奖励模型进行顺序最小化。与使用固定掩码概率的MLM预训练不同,所提出的基于高斯过程的汤普森采样(GP-TS)通过顺序选择能够提升性能的掩码超参数来加速预训练。我们通过实验证明GP-TS能够高效地预训练语言模型,即在多种设置下以更少的训练轮数实现更低的MLM损失。此外,GP-TS预训练的TLM在下游任务上取得了有竞争力的性能,同时避免了昂贵的超参数网格搜索。GP-TS为高效且优化的TLM预训练提供了一个交互式框架,通过规避代价高昂的超参数选择,实现了显著的计算节省。

英文摘要:

We design and evaluate a Bayesian optimization framework for resource efficient pre-training of Transformer-based language models (TLMs). TLM pre-training requires high computational resources and introduces many unresolved design choices, such as selecting its pre-training hyperparameters. We propose a multi-armed bandit framework for the sequential selection of TLM pre-training hyperparameters, aimed at optimizing language model performance, in a resource efficient manner. We design a Thompson sampling algorithm, with a surrogate Gaussian process reward model of the Masked Language Model (MLM) pre-training objective, for its sequential minimization. Instead of MLM pre-training with fixed masking probabilities, the proposed Gaussian process-based Thompson sampling (GP-TS) accelerates pre-training by sequentially selecting masking hyperparameters that improve performance. We empirically demonstrate how GP-TS pre-trains language models efficiently, i.e., it achieves lower MLM loss in fewer epochs, across a variety of settings. In addition, GP-TS pre-trained TLMs attain competitive downstream performance, while avoiding expensive hyperparameter grid search. GP-TS provides an interactive framework for efficient and optimized TLM pre-training that, by circumventing costly hyperparameter selection, enables substantial computational savings.

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

↑