AI中的“预训练”是什么意思

“预训”(Pre-training)是现代深度学习,尤其是像Transformer这类大模型的核心基础和第一步。可以把它想象成让一个模型先去“上小学和中学”,广泛学习各种通用知识,而不是直接“上大学”专攻一个专业。

预训练 核心思想:先通用,后专用

“预训练”指的是在一个大规模、无标注的通用数据集上,以自监督学习的方式训练模型,让其学习数据中潜在的基础模式、通用特征和知识表示的过程。

这个过程的目标不是让模型立刻完成某个具体任务(如情感分析或翻译),而是让它打下坚实的基础,构建起对语言、图像或其他领域的基本认知。之后,这个“博学”的模型可以通过微调(Fine-tuning) 快速适应各种下游任务。

为了更直观地理解预训练以及它与后续微调的关系,可以参考下面的表格:

方面 预训练 (Pre-training) 微调 (Fine-tuning)
学习目标 学习通用的特征、模式和知识表示 (如语言结构、语法规则、世界知识) 学习特定任务的知识和模式 (如情感分类、问答、翻译)
训练数据 海量无标注原始数据 (如互联网文本、书籍) 少量有标注任务特定数据
计算资源 需求巨大,通常需要昂贵的GPU/TPU集群训练数天甚至数月 需求相对较小,通常可以在单个高性能GPU上几小时或几天内完成
输出 一个包含了学到的通用知识的预训练模型(如BERT, GPT的初始权重) 一个专门化的模型,适应了下游任务 (如情感分析BERT, 对话生成GPT)
类比 “上中小学”:学习语文、数学、物理、化学等通识教育,为未来发展打下宽厚的基础。 “读大学或职业教育”:基于已有的通识基础,学习计算机、金融、法律等特定领域的专业知识,为具体职业做准备。

预训练是如何进行的?

以自然语言处理(NLP)中基于Transformer的模型为例,预训练通常通过自监督学习(Self-supervised Learning) 来完成。这意味着训练所需的“标签”或“监督信号”直接从原始数据本身产生,而无需人工标注。最常见的预训练任务有:

  1. 掩码语言模型(Masked Language Model, MLM):
    ◦ 做法:随机遮盖输入句子中的一些词(token),然后训练模型根据上下文来预测这些被遮盖的词。

    ◦ 代表模型:BERT。

    ◦ 目标:让模型学会双向地理解语言的深层语义和上下文关系。

  2. 自回归语言模型(Autoregressive Language Model):
    ◦ 做法:让模型逐个预测序列中的下一个词。在预测时,它只能看到当前词之前的上下文信息。

    ◦ 代表模型:GPT系列。

    ◦ 目标:让模型学会生成连贯、合乎逻辑的文本,并掌握大量的世界知识。

通过这些任务,模型在消化海量数据的过程中,逐渐学会了语法、句法、语义逻辑,甚至许多事实性知识。

为什么要进行预训练?

预训练-微调范式带来了多重好处:

  1. 降低对标注数据的依赖:许多任务难以获取大量高质量标注数据。预训练模型通过无标注数据学到了丰富的表示,微调时只需少量标注数据就能达到很好效果。
  2. 提升模型性能与泛化能力:在海量数据上学习到的通用特征表示,使得模型在面对新任务时起点更高、泛化能力更强,通常能显著超越从零开始训练的模型。
  3. 提高开发效率与降低成本:无需每次都从零开始训练一个大型模型,节省了大量的计算资源、时间和金钱。研究人员和工程师可以更快速地针对不同任务构建高性能模型。

预训练之后做什么?

预训练模型本身通常不能直接完成特定任务。它需要经过微调(Fine-tuning) 才能发挥巨大价值。

• 微调:即在预训练好的模型权重基础上,使用特定任务的有标注数据,对模型的所有参数或部分参数进行进一步的训练,使其适应特定任务的需求。

• 提示(Prompting):对于超大规模模型(如GPT-3、ChatGPT),有时不更新模型权重,而是通过设计巧妙的提示词(Prompt) 来引导模型直接输出符合要求的答案。

最后

总而言之,Transformer中的“预训练”是一个让模型在大规模数据上预先学习通用知识和特征表示的基础阶段。它为模型后续高效地适应各种下游任务奠定了坚实的基础,是现代AI领域一项非常核心和强大的技术范式。


愿你我都能在各自的领域里不断成长,勇敢追求梦想,同时也保持对世界的好奇与善意!

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐