Transformers documentation

UL2

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v5.22.1).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

该模型于 2022-05-10 在 HF papers 上发布,并于 2023-06-20 贡献至 Hugging Face Transformers。

UL2

概述

T5 模型由 Yi Tay、Mostafa Dehghani、Vinh Q. Tran、Xavier Garcia、Dara Bahri、Tal Schuster、Huaixiu Steven Zheng、Neil Houlsby 和 Donald Metzler 在 Unifying Language Learning Paradigms 中提出。

该论文的摘要如下:

现有的预训练模型通常针对特定类别的问题进行设计。迄今为止,关于正确的架构和预训练方案应该是什么,似乎仍未达成共识。本文提出了一个统一的框架,用于预训练在各类数据集和设置下均具有普适有效性的模型。我们首先将架构原型与预训练目标这两个常被混淆的概念区分开来。接着,我们提出了自然语言处理中自监督的广义统一视角,并展示了不同的预训练目标如何相互转化,以及在不同目标之间进行插值为何是有效的。随后,我们提出了混合去噪器(Mixture-of-Denoisers, MoD),这是一种将多种预训练范式结合在一起的预训练目标。此外,我们引入了模式切换的概念,其中下游微调与特定的预训练方案相关联。我们进行了大量的消融实验来比较多种预训练目标,发现我们的方法通过在多种不同设置下超越 T5 和/或 GPT 类模型,推动了帕累托前沿。最后,通过将模型规模扩展至 200 亿参数,我们在 50 个成熟的有监督 NLP 任务上取得了 SOTA 性能,这些任务涵盖语言生成(包含自动和人工评估)、语言理解、文本分类、问答、常识推理、长文本推理、结构化知识接地和信息检索。我们的模型在上下文学习中同样表现出色,在零样本 SuperGLUE 上超越了 1750 亿参数的 GPT-3,并将 T5-XXL 在单样本摘要任务上的性能提升了三倍。

该模型由 DanielHesslow 贡献。原始代码可在此处找到。

使用技巧

  • UL2 是一种编码器-解码器模型,在多种去噪函数的混合数据上进行预训练,并在一系列下游任务上进行微调。
  • UL2 与 T5v1.1 具有相同的架构,但使用 Gated-SiLU 激活函数而非 Gated-GELU。
  • 作者发布了该架构的检查点,可在此处查看。

由于 UL2 与 T5v1.1 具有相同的架构,请参阅 T5 的文档页面 以获取 API 参考、技巧、代码示例和笔记本。

Update on GitHub