arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2005.00175cs.CL
收藏

选择信息上下文改进语言模型微调

Selecting Informative Contexts Improves Language Model Finetuning

UT Austin(德克萨斯大学奥斯汀分校);Intel Labs(英特尔研究院)

Richard Antonello, Nicole Beckage, Javier Turek, Alexander Huth

摘要:

语言模型微调对于现代自然语言处理是必不可少的,但计算昂贵且耗时。此外,微调的有效性受到包括负面影响性能的训练样本的限制。为了提高语言模型微调的整体训练效率和最终性能,本文提出了一种通用的微调方法--信息增益过滤。我们将一个例子的信息增益定义为在该例子上训练后测试度量的改进。然后训练二级学习者近似这个量。在微调过程中,这个学习者选择信息丰富的例子,跳过没有信息的例子。我们证明了我们的方法在数据集、微调任务和语言模型架构上都有一致的改进。例如,我们在books数据集上实现了54.0的中位数困惑度,而标准微调为57.3。我们提出的统计证据,提供了深入了解我们的方法超过标准微调的改进。我们的方法的通用性,使我们提出了一个新的范式语言模型微调-我们鼓励研究人员发布预训练的二级学习者在共同的语料库,以促进高效和有效的微调,从而提高性能,减少整体能源足迹的语言模型微调。

英文摘要:

Language model fine-tuning is essential for modern natural language processing, but is computationally expensive and time-consuming. Further, the effectiveness of fine-tuning is limited by the inclusion of training examples that negatively affect performance. Here we present a general fine-tuning method that we call information gain filtration for improving the overall training efficiency and final performance of language model fine-tuning. We define the information gain of an example as the improvement on a test metric after training on that example. A secondary learner is then trained to approximate this quantity. During fine-tuning, this learner selects informative examples and skips uninformative ones. We show that our method has consistent improvement across datasets, fine-tuning tasks, and language model architectures. For example, we achieve a median perplexity of 54.0 on a books dataset compared to 57.3 for standard fine-tuning. We present statistical evidence that offers insight into the improvements of our method over standard fine-tuning. The generality of our method leads us to propose a new paradigm for language model fine-tuning -- we encourage researchers to release pretrained secondary learners on common corpora to promote efficient and effective fine-tuning, thereby improving the performance and reducing the overall energy footprint of language model fine-tuning.

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

↑