预训练模型的兴起给自然语言处理（NLP）带来了“新面貌”。

近年来，Google、Facebook、OpenAI、微软、百度等人工智能“头部玩家”推出多个颇具影响的预训练模型，并反复迭代出十多个版本。无论学术界还是业界，人们对大规模预训练模型“热情高涨”。

日前，来自清华大学的一支研究团队提出一种简单高效的NLP学习框架。不同于当下NLP社区主流的“大规模预训练+下游任务微调”的范式，这一框架无需进行大规模预训练，同时将训练效率提升两个数量级，并在多个NLP任务上，实现了比肩甚至超出预训练模型的性能。近日，相关研究以预印本形式，在arXiv上发表。

预训练模型的“内功”

预训练模型在自然语言处理领域蓬勃发展，近年来在多个子方向取得了颠覆性的成果。

“自然语言处理的‘预训练’过程，就像武侠小说中，练武之人的‘修炼内功’。”上海对外经贸大学副研究员邵浩说，“一个人要成为武林高手，需要有扎实的“内功”，内功修炼好之后，再去学各种招式就非常容易上手，并能发挥其最大效用。”

随着深度学习的发展，模型参数显著增长，从而需要越来越大的数据集，用于充分训练模型参数。然而，因大部分NLP任务的标注成本极为高昂，尤其是句法和语义相关的任务，构建大规模标注数据集尤为困难。

相比较而言，大规模无标注数据集相对易于构建。为更好地利用海量无标签文本数据，常规的做法是首先从这些数据中学到较好的文本表示，然后再将其用于其他任务。许多研究表明，在大规模无标注语料中训练的预训练语言模型，可以使多方面NLP任务获得显著的性能提升。

通过海量无标注语料来预训练神经网络模型，可以让人工智能更利于下游NLP任务的完成。预训练模型的作者已经设计出了基准模型，这样，使用者就可以在自己的NLP数据集上应用该模型，而无需从头开始构建模型来解决类似的问题。尽管后续过程需要进行一些微调，但这为人们节省了大量的时间和计算资源。

2018年，无监督的双向预训练语言模型ELMo被提出，这种上下文相关的文本表示方法在多个典型任务上表现惊艳，能有效处理一词多义问题。紧随其后，GPT，BERT等预训练语言模型相继被提出，预训练模型技术开始在NLP领域大放异彩，并在各种下游任务中遍地开花。

任务驱动模型出场

“预训练语言模型因其强大的性能被广泛关注，基于‘预训练—微调’的范式也成为许多NLP任务的标准方法。”清华大学交叉信息研究院助理教授、RecurrentAI联合创始人杨植麟对《中国科学报》说，“然而，当前通用语言模型的预训练成本极其高昂，这使得只有少数资源充足的研究机构或组织能够对其展开探索。”

为解决上述问题，杨植麟团队提出的一种完全不需要预训练语言模型的高效学习框架。这一框架从通用语料中筛选出与下游任务相关的子集，并将语言建模任务与下游任务进行联合训练。

该论文第一作者、清华大学计算机科学实验班（姚班）大四本科生姚星丞介绍说，提出任务驱动的语言模型的想法源于一个基本的观察：人类可以通过对关键信息的学习，在有限的时间和精力投入情况下，快速掌握某一任务技能。例如，在临近考试时，学生仅根据考纲复习浏览若干相关章节的要点即可应对考试，而不必学习所有可能的知识点。与之类似，预训练语言模型在某一下游任务上的优良表现，“很有可能因为来自于语料中与下游任务相关的数据”。

基于这一判断，该团队提出任务驱动的语言模型（TLM），它仅利用从大规模通用语料中提取的少量与下游任务相关的数据，就可以取得与全量数据类似的结果。

“相较于传统的预训练模型RoBERTa（基于BERT的改进模型，使用更大的批次和更多的数据对模型进行更长的训练），TLM仅需要约1%的训练时间与1%的语料，即可在众多NLP任务上，表现出比肩甚至超出预训练模型的性能。”姚星丞说，“我们目前也正在尝试将任务驱动的方法推广到更大规模的模型上，如GPT-3或T5。”

跳出预训练范式

为了从大规模通用语料中抽取关键数据，TLM以任务数据作为查询对象，用基于稀疏特征的BM25算法作为召回算法，对通用语料库进行相似数据的召回。

“除已有的下游任务数据以外，其余的语料均通过BM25算法进行相似性匹配而自动筛选，不需要人工做额外的选择与标记。”姚星丞说。“TLM基于任务数据和召回数据，同时优化任务目标和语言建模目标，从零开始进行联合训练。”

为了测试TLM的性能，研究人员在8项NLP分类任务上，从三个不同规模展开了对比实验。这8项任务涵盖了计算机科学、生物医药、新闻、评论等4个领域，包括了训练样本数量小于5000的低资源任务和训练样本数量大于20000的高资源任务，任务类型覆盖了话题分类，情感分类，实体关系抽取等。

测试结果显示，和对应“预训练—微调”基准相比，TLM实现了相当甚至更优的性能。平均而言，TLM减少了两个数量级规模的训练计算量以及训练语料的规模。整体来说，预训练模型以极高的成本学习尽可能多的，和任务无关的知识，而TLM以非常低的成本，针对每个任务学习相关知识。

“当我们有少数特定目标的任务需要解决的时候(例如希望对少量几个数据集进行研究)，TLM会是非常高效的。”姚星丞说，“而需要一次性解决大量任务时(例如工业界构建一个NLP平台为多方提供相似的服务)，预训练模型仍然具有优势。”

此外，TLM是任务驱动的，所以可以给研究人员更大的自由度，从而自定义策略进行标记、序列长度、数据表示、超参数的调整等等，从而达到提高性能和效率的目的。

“TLM的提出，让NLP研究跳脱出‘预训练—微调’范式成为可能，这有利于推动NLP研究公平化。”杨植麟解释说，预训练本身严重依赖大量的计算资源，这一限制使大多数NLP研究者只能专注于对微调算法的研究。然而微调算法的性能上限，很大程度上受预训练模型性能的约束。而TLM可以让大多数研究人员可以以较低的代价和较高的效率，基于最先进的解决方案对模型架构、损失函数、算法等方面进一步自由探索。

杨植麟认为，未来会有更多有趣的研究可以在TLM的基础上展开。例如，如何经济地达到更大规模预训练模型的表现效果；如何提升TLM的通用性与可迁移性；可否利用TLM进行小样本或零样本学习等。此外，还可以将预训练模型和TLM结合，从而在通用性和效率之间实现更好的权衡。

相关论文信息：https://arxiv.org/pdf/2111.04130.pdf

项目地址：https://github.com/yaoxingcheng/TLM