arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.07779cs.CL
收藏

AdapterHub:一个自适应Transformers的框架

AdapterHub: A Framework for Adapting Transformers

Technical University of Darmstadt(达姆施塔特工业大学);New York University(纽约大学);University of Cambridge(剑桥大学);DeepMind(深度思维)

Jonas Pfeiffer, Andreas Rücklé, Clifton Poth, Aishwarya Kamath, Ivan Vulić, Sebastian Ruder, Kyunghyun Cho, Iryna Gurevych

摘要:

目前NLP的工作方式包括下载和微调由数百万或数十亿个参数组成的预训练模型。存储和共享这样的大型训练模型是昂贵的,缓慢的,耗时的,这阻碍了向更通用和通用的NLP方法的进展,这些方法可以从许多任务中学习。适配器--在预训练模型的每一层中插入小的学习瓶颈层--通过避免对整个模型进行完全微调来改善这个问题。然而,共享和集成适配器层并不简单。我们提出了AdapterHub,一个框架,允许动态的“嵌入式”的预训练的适配器为不同的任务和语言。该框架建立在流行的HuggingFace Transformers库之上,能够非常容易和快速地适应最先进的预训练模型(例如,BERT、RoBERTa、XLM-R)。下载、共享和训练适配器是尽可能无缝的,只需对训练脚本和专门的基础设施进行最小的更改。我们的框架使可扩展的和容易访问的共享特定于任务的模型,特别是在低资源的情况下。AdapterHub包括所有最新的适配器架构,可以在https://AdapterHub.ml上找到。

英文摘要:

The current modus operandi in NLP involves downloading and fine-tuning pre-trained models consisting of millions or billions of parameters. Storing and sharing such large trained models is expensive, slow, and time-consuming, which impedes progress towards more general and versatile NLP methods that learn from and for many tasks. Adapters -- small learnt bottleneck layers inserted within each layer of a pre-trained model -- ameliorate this issue by avoiding full fine-tuning of the entire model. However, sharing and integrating adapter layers is not straightforward. We propose AdapterHub, a framework that allows dynamic "stitching-in" of pre-trained adapters for different tasks and languages. The framework, built on top of the popular HuggingFace Transformers library, enables extremely easy and quick adaptations of state-of-the-art pre-trained models (e.g., BERT, RoBERTa, XLM-R) across tasks and languages. Downloading, sharing, and training adapters is as seamless as possible using minimal changes to the training scripts and a specialized infrastructure. Our framework enables scalable and easy access to sharing of task-specific models, particularly in low-resource scenarios. AdapterHub includes all recent adapter architectures and can be found at https://AdapterHub.ml.

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

↑