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

EasyTransfer --面向NLP应用的简单可扩展深度迁移学习平台

EasyTransfer -- A Simple and Scalable Deep Transfer Learning Platform for NLP Applications

Alibaba Group(阿里巴巴集团);State Key Lab of CAD&CG, Zhejiang University(浙江大学计算机辅助设计与图形学国家重点实验室)

Minghui Qiu, Peng Li, Chengyu Wang, Hanjie Pan, Ang Wang, Cen Chen, Xianyan Jia, Yaliang Li, Jun Huang, Deng Cai, Wei Lin

摘要:

文献已经见证了将预训练语言模型(PLM)和迁移学习(TL)算法成功地应用于广泛的自然语言处理(NLP)应用,然而为此目的构建易于使用和可扩展的TL工具包并不容易。为了弥合这一差距,EasyTransfer平台旨在为NLP应用程序开发深度TL算法。EasyTransfer后端具有高性能和可扩展的引擎,可实现高效的训练和推理,并集成了全面的深度TL算法,使工业规模TL应用程序的开发更加容易。在EasyTransfer中,内置的数据和模型并行策略,结合AI编译器优化,比社区版本的分布式训练快4.0倍。EasyTransfer支持ModelZoo中的各种NLP模型,包括主流PLM和多模态模型。它还具有各种内部开发的TL算法,以及用于NLP应用程序的AppZoo。该工具包方便用户快速开始模型训练、评估和在线部署。EasyTransfer目前部署在阿里巴巴,支持多种业务场景,包括商品推荐、个性化搜索、会话式问答等。在真实数据集和在线应用上的大量实验表明,EasyTransfer适合在线生产,并具有各种应用的尖端性能。EasyTransfer的源代码发布在Github(https://github.com/alibaba/EasyTransfer)。

英文摘要:

The literature has witnessed the success of leveraging Pre-trained Language Models (PLMs) and Transfer Learning (TL) algorithms to a wide range of Natural Language Processing (NLP) applications, yet it is not easy to build an easy-to-use and scalable TL toolkit for this purpose. To bridge this gap, the EasyTransfer platform is designed to develop deep TL algorithms for NLP applications. EasyTransfer is backended with a high-performance and scalable engine for efficient training and inference, and also integrates comprehensive deep TL algorithms, to make the development of industrial-scale TL applications easier. In EasyTransfer, the built-in data and model parallelism strategies, combined with AI compiler optimization, show to be 4.0x faster than the community version of distributed training. EasyTransfer supports various NLP models in the ModelZoo, including mainstream PLMs and multi-modality models. It also features various in-house developed TL algorithms, together with the AppZoo for NLP applications. The toolkit is convenient for users to quickly start model training, evaluation, and online deployment. EasyTransfer is currently deployed at Alibaba to support a variety of business scenarios, including item recommendation, personalized search, conversational question answering, etc. Extensive experiments on real-world datasets and online applications show that EasyTransfer is suitable for online production with cutting-edge performance for various applications. The source code of EasyTransfer is released at Github (https://github.com/alibaba/EasyTransfer).

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

↑