跳至主要内容

高级迁移学习技术概览

了解领域自适应、多任务学习和小样本学习等高级迁移学习技术如何提升模型性能与泛化能力。
Stanislav Karzhev
Stanislav Karzhev

创建您的免费账户

实践并学习路径你的进步。

使用 Google 继续继续即表示您接受我们的 使用条款,我们的 隐私政策 ,并且您的数据存储在美国。

迁移学习在机器学习中变得至关重要:将预训练模型适配到新的相关任务,以节省时间和资源。这一策略在计算机视觉、自然语言处理和语音识别等领域尤为高效,这些领域拥有大量可用的预训练数据。

尽管广受欢迎,微调这一常见的迁移学习方法常会遇到过拟合或预训练模型领域与目标数据不匹配的问题,即所谓的领域移位。当面对复杂任务或数据有限时,基础的迁移学习可能难以应对,导致适配不足。

为克服这些挑战,新一代高级迁移学习技术应运而生,专注解决领域差异、数据稀缺以及高效模型适配的需求。本文将探讨部分高级迁移学习策略。

高级迁移学习技术的类型

传统方法如先在大规模数据集上预训练,再在特定任务数据集上微调,是基础但有时不足以应对现实需求。更高级的方法应对领域移位、数据稀缺和多任务学习等细微问题。

高级方法引入了更高的复杂性与灵活性,旨在传统方法力有未逮的场景中提升模型表现。

领域自适应

领域自适应技术用于处理源域(预训练)与目标域(微调)之间的差异。当预训练模型遇到与训练数据差异显著的数据时,传统微调常产生次优结果。领域自适应通过对齐数据分布来降低这种不匹配,帮助模型在新环境中更好泛化。

多任务学习(MTL)

MTL 允许模型同时学习多个相关任务。通过在任务间共享知识,模型能够利用共性并提升所有任务的表现。该策略在任务互补时尤为有效,例如在自然语言处理中联合学习文本分类与情感分析。MTL 通过引导模型关注共享模式,提升泛化能力并降低过拟合风险。

小样本学习

另一项关键技术是小样本学习,旨在用极少的标注数据训练模型。传统方法需要大量标注数据,而小样本方法可仅用少量示例迁移知识。诸如元学习与原型网络等方法能使模型快速适配新任务,在个性化推荐或小众图像分类等领域尤为宝贵。

自监督学习(SSL)

SSL 利用未标注数据训练模型,当标注数据稀缺或不可用时,是一项强大的工具。该技术从数据自身构造监督信号,解决如预测图像缺失部分或重构打乱文本等预训练任务。采用 SSL 预训练的模型在多种应用中展现出较强的可迁移性。

零样本学习

零样本学习更进一步,使模型在没有任何标注数据的情况下处理全新任务。其通过辅助信息(如语义关系或任务描述)实现,即使面对新类别也能泛化。在新类别频繁出现的动态领域,这一能力尤为重要,例如处理新造词或俚语的 NLP 应用。

advanced transfer learning techniques

使用 napkin.ai 生成的示意图

领域自适应技术

如前所述,领域自适应是迁移学习的一个分支,关注训练模型所用数据与部署环境所遇数据之间存在显著差异(“领域移位”)的情形。

在典型的迁移学习中,通常假设源域与目标域具有相似特征,但在许多现实场景中,两者差异很大。最终这会削弱性能,因为在一套数据上训练的模型难以泛化到新条件。

领域自适应技术旨在弥合这一差距,通过最小化源域与目标域数据分布之间的差异,使在一个领域训练的模型能在另一个领域有效工作。因此,当某一领域有大量标注数据而另一领域稀缺或缺失时,领域自适应价值更大。

为应对领域自适应的挑战,已经发展出多种技术——其中两类最常见的方法是对抗式方法与领域不变特征学习。

对抗式领域自适应

一种广泛使用的领域自适应技术是对抗式学习。在此框架中,模型被训练以学习对目标任务有用的特征,同时确保这些特征在源域与目标域之间不可区分。

我们可借助判别器实现这一点,即一种神经网络,用于区分源域与目标域数据。模型以对抗方式训练,学会“欺骗”判别器,使特征表示具有领域不变性。如果这听起来似曾相识,是因为该技术受生成对抗网络(GAN)启发,后者采用类似设置来生成逼真的合成数据。

领域不变特征学习

另一种方法专注于学习不受领域特异性变化影响的表示。其将两个领域的输入数据映射到共享的特征空间,在该空间内源域与目标域更加对齐。最大均值差异(MMD)与 CORAL(相关对齐)等技术可在此空间中最小化统计差异,确保模型在应用到目标域时更有效泛化。

应用场景

在领域移位不可避免的诸多真实应用中,领域自适应已被证明非常有用:

  • 在 NLP 中,领域自适应常用于将训练于一种语言或方言的模型迁移至另一种语言或方言。
    • 例如,一个用英语训练的情感分析模型可能需要适配到同一语言的区域方言,或将训练于正式文本的模型适配至非正式、口语化的语言,如社交媒体数据。
  • 在计算机视觉中,当训练数据与目标数据的图像特性存在差异时,常需进行领域自适应。
    • 例如,训练于自然光照条件下物体图像的模型,可能需要适配到昏暗或人工光照环境。
  • 自动驾驶系统需要在不同地理区域都具备稳健表现,此处领域自适应再次派上用场。
    • 在一个国家训练用于识别路标、行人或车辆的模型,可能需要适配到另一地区略有不同的道路状况、天气模式,甚至交通规则。

applications of domain adaption techniques

多任务学习与迁移学习的协同

随着机器学习任务愈发复杂,同时解决多个相关问题的能力可显著提升模型的性能与效率。MTL 通过同时训练多个任务来实现这一点,而非孤立地专注于单一任务。

什么是多任务学习?

MTL 的强大之处在于利用任务之间的关系与共享结构,使模型学习到更丰富的特征表示。从一个任务中获得的洞见可提升另一个任务的表现,从而提高整体学习效率。

在 MTL 中,常用两种主要策略:

  • 硬参数共享:该方法让模型在各任务间共享大部分参数,仅任务特定层相互独立。通过促使模型学习对多任务泛化良好的表示,从而降低过拟合。
  • 软参数共享:此法为每个任务保留独立参数,但通过某种正则化使这些参数相似,在不完全共享的情况下鼓励共享学习。

A diagram illustrating an example of transfer learning and multitask learning.

迁移学习与多任务学习的示例。来源:Kim 等,2021

多任务学习的优势

MTL 在复杂的机器学习场景中具有多项优势,因而是一种颇具价值的方法。

更强的泛化能力

通过跨多个任务学习,模型接触更广泛的数据并学习到更通用的表示。这通常能提升各个任务的性能,尤其当这些任务彼此相关时。

数据效率

在标注数据稀缺的情况下,MTL 能更好地利用可用数据。模型从多个任务学习,能借助相关任务的洞见提升性能。此类跨任务学习加速了学习进程,降低了每个任务对大型数据集的依赖。

模型效率

与为每个任务分别训练模型不同,MTL 将学习整合到单一模型中,从而降低计算成本与内存占用,更便于在资源受限环境中部署。

多任务学习的应用

MTL 已在多个领域成功应用,尤其在 NLP 与计算机视觉中有诸多用例。

自然语言处理

在自然语言处理中,MTL 常用于联合训练多个相关任务。例如,一个语言模型可同时执行机器翻译与文本分类。通过在这些任务之间共享学到的表示,模型能更好地理解语言结构与上下文,从而惠及两类任务。

计算机视觉

在计算机视觉中,MTL 可用于目标检测与图像分割等任务。我们可以训练一个模型在识别图像中目标的同时,学习对这些目标进行分类。共享学习使模型既能识别位置,也能识别类别,从而更通用且高效。

小样本与自监督学习

在标注数据稀缺或缺失的场景中,小样本学习与 SSL 成为弥补传统监督方法局限的有力技术。

这些方法以创新方式使模型在极少标注数据或大量未标注数据的帮助下适配新任务,是高级迁移学习领域的重要工具。

小样本学习

如前所述,小样本学习旨在让模型仅凭少量标注示例即可快速适配新任务。

然而,在许多应用中收集大规模数据集并不现实,如罕见疾病诊断或小众产品识别。小样本学习通过从相关任务迁移知识,并对模型进行少量微调,在极少新数据下取得良好表现。

下文讨论小样本学习中的两个关键策略。

元学习

元学习,又称“学会学习”,通过训练模型在少量数据下快速适配新任务。其思想是在多样任务上训练模型,使其能够泛化所学并迅速调整到未见过的任务。

诸如模型无关元学习(MAML)等框架,使模型学得一组可仅用少量样本就能微调到新任务的最优参数。

原型网络

另一常见方法是原型网络,通过在学习到的特征空间中计算样本间距离来进行分类。在这种设置中,每个类别由一个原型表示(即该类别样本在特征空间中的均值)。

随后根据新数据点与这些原型的接近程度进行分类。原型网络在小样本学习中行之有效,因为其每类仅需极少示例即可生成有意义的分类原型。

自监督学习

SSL 通过生成监督信号,使模型从海量未标注数据中提取有意义的表示,从而无需人工标注。模型从未标注数据中学习后,可在较小的标注数据集上进行微调,在下游任务上取得强劲表现。

影响力最大的 SSL 技术之一是对比学习,其通过比较正负样本对来训练模型。目标是在特征空间中将相似数据点(正样本对)拉近、将不相似数据点(负样本对)推远。

在自然语言处理中,掩码语言建模是常见的 SSL 方法,应用于 BERT 等模型。该方法对输入部分进行掩码,并训练模型利用上下文预测缺失内容。该预训练策略帮助模型学习稳健、通用的表示,可微调用于各种 NLP 任务。

应用

小样本与自监督学习广泛用于数据限制是主要障碍的领域。以下是各自的一些关键应用。

小样本学习在自定义目标检测中的应用

在自定义目标检测中,小样本学习尤为有用:识别稀有或专业化目标时,模型可能只能从少量标注示例中学习。

在工业环境中,检测制造过程中的缺陷可能只有极少实例。借助小样本学习,可将预训练于通用目标检测任务的模型快速适配到这些特定对象的检测,且仅需极少额外训练数据。

自监督学习在语言模型中的应用

自监督学习通过 GPT 和 BERT 等模型革新了自然语言处理。这些模型学习通用语言表示,可用相对较小的标注数据集进行微调以完成特定任务。例如,使用 SSL 预训练的语言模型可微调用于情感分析、文本分类或摘要,且常能取得最先进的效果。

高级迁移学习技术的实践

机器学习库和框架大幅简化了高级迁移学习方法的实现。诸如PyTorch、TensorFlow与Hugging Face Transformers等工具提供了必要的灵活性与预构建组件,加速开发进程。

框架与工具

对于希望实现高级迁移学习技术的实践者,以下库与框架提供了广泛支持:

  • PyTorch:一款灵活、易用的框架,广泛用于研究与实际机器学习项目。PyTorch 的动态图机制非常适合元学习与多任务学习等任务。
  • TensorFlow:另一款流行的深度学习框架,TensorFlow 通过其高级 API——Keras,为迁移学习与领域自适应提供了强大支持。TensorFlow 的tensorflow_hub 也提供了可用于新任务微调的预训练模型。
  • Hugging Face Transformers:处理自然语言任务的理想库,提供多种预训练模型,如 BERT、GPT 与 T5,可轻松微调用于特定下游任务。

挑战与注意事项

实现高级迁移学习技术并非易事,需要注意以下关键因素。

灾难性遗忘

在新任务上微调可能导致模型“遗忘”先前学到的内容。诸如弹性权重固化(EWC)等技术可帮助保留预训练模型的权重,对保持先验知识至关重要。

应对计算复杂度

领域自适应与多任务学习等技术可能显著增加计算需求。训练对抗式领域自适应模型或多任务架构会因额外复杂性而需要更多资源。我们需要优化模型,并使用诸如混合精度训练等工具以高效处理大模型。

确保公平与缓解偏差

基于带偏数据预训练的迁移学习模型,可能在新任务中延续甚至放大这些偏差,导致预测偏斜。在医疗与金融等敏感领域,这一风险尤为关键。为确保公平性,应定期评估模型输出是否存在偏差,并采用诸如重加权、对抗式去偏或公平表示学习等缓解技术。

行业案例

高级迁移学习方法正被广泛应用于各行业,为复杂的现实挑战提供创新解法。以下是一些关键应用与行业示例,展示这些技术如何推动 AI 进步。

OpenAI:GPT 模型中的小样本学习

OpenAI 的GPT-4o 模型是小样本学习的领先实践,展示出仅凭少量示例就能适配新任务的能力。GPT-4o 在海量文本数据上预训练,无需任务特定微调即可执行翻译、摘要、问答等多种任务。

用户只需提供少量期望任务的示例,模型即可据此泛化。这种小样本学习能力拓宽了大语言模型的应用范围,从创意写作到对话式智能体皆是如此。

DeepMind:医疗 AI 中的多任务学习

DeepMind 作为 AI 研究的领军者,利用多任务学习开发出可辅助从医学影像中诊断多种病症的医疗模型。例如,其用于从视网膜扫描检测眼病的 AI 系统,能够用一个模型诊断包括糖尿病视网膜病变与老年黄斑变性在内的多种病症。

这种 MTL 方法减少了为每种病症分别训练模型的需求,帮助医疗机构简化诊断流程,更快速、更准确地为患者提供护理。

Waymo:自动驾驶中的领域自适应

Waymo(谷歌的自动驾驶项目)使用领域自适应技术,提升自动驾驶模型在不同驾驶环境下的泛化能力。公司先在模拟环境中训练模型,再将其适配到不同城市与天气条件下的真实部署。

该过程确保自动驾驶车辆能够处理不同城市景观的细微差别:从旧金山拥挤的街道,到亚利桑那开阔的高速公路,而无需为每个地点进行大量再训练。

高级迁移学习的未来趋势

研究者正不断突破边界,致力于让模型更具适应性、更公平,并能在减少数据与训练时间的同时处理复杂的真实世界任务。

持续学习

持续学习(或终身学习)关注让模型在保留已学知识的同时,持续从新数据中学习。与传统迁移学习中微调后模型静态不同,持续学习允许模型不断适配新信息。

这对机器人与自动系统等动态环境至关重要,因为新情境会经常出现。在此背景下,基于记忆的策略有助于在保留过往知识与吸收新技能之间取得平衡。未来发展将持续把持续学习扩展到更复杂的领域,确保模型在不损害先前性能的情况下实现高效更新。

公平与偏差缓解

随着迁移学习在医疗、金融与执法等行业的广泛应用,解决公平与偏差问题已不容忽视。

在大型、多样数据集上预训练的模型往往携带固有偏差,在新应用中可能导致结果偏斜。为对抗这一点,研究者正探索对抗式去偏、重加权与公平表示学习等技术,以检测并缓解这些偏差。

确保公平性将继续是迁移学习研究的核心焦点,尤其在偏差预测会带来现实影响的敏感领域。

混合式迁移学习

混合式迁移学习结合了监督、无监督与自监督方法,以发挥各自优势。模型通常先通过无监督或自监督方法学习通用表示,再用标注数据针对具体任务进行微调。

这种混合策略提升了可迁移性,降低了对大型标注数据集的依赖,特别适用于标注数据有限的领域,如罕见疾病诊断或专业工业应用。

强化学习中的迁移学习

迁移学习在强化学习(RL)中也在不断取得进展:将一个环境中训练好的智能体适配到相关环境,从而减少掌握新任务所需的时间与数据,提升其解决复杂、多步问题的效率。

例如,一个在特定类型机器人系统上训练的 RL 智能体,可以以最少的再训练,将其知识应用到新的但相似的设置,大幅提升 RL 系统的通用性与可扩展性。

结论

本文探讨了高级迁移学习技术。我们审视了用于处理数据分布差异的领域自适应、用于共享知识的多任务学习,以及适用于标注数据有限情境的小样本与自监督学习。

这些技术为我们提供了有效策略来应对挑战、提升模型性能。随着 AI 持续发展,熟悉这些技术将是保持竞争力的关键。

Stanislav Karzhev

作者

Stanislav Karzhev

刚毕业的理学硕士,专攻人工智能

主题