arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2104.12470cs.CL
收藏

简单高效的Transformer:大型NLP模型的可扩展推理解决方案

Easy and Efficient Transformer : Scalable Inference Solution For large NLP model

NetEase Inc.(网易公司);Hong Kong Baptist University(香港浸会大学)

Gongzheng Li, Yadong Xi, Jingzhen Ding, Duan Wang, Bai Liu, Changjie Fan, Xiaoxi Mao, Zeng Zhao

摘要:

最近,大规模的基于transformer的模型已被证明是有效的,在许多领域的各种任务。然而,将它们应用于工业生产需要繁琐和繁重的工作,以减少推理成本。为了填补这一空白,我们引入了一个可扩展的推理解决方案:简单和高效的Transformer(EET),包括一系列的Transformer推理优化的算法和实现水平。首先,我们为长输入和大隐藏大小设计高度优化的内核。其次,我们提出了一个灵活的CUDA内存管理器,以减少部署大型模型时的内存占用。与最先进的Transformer推理库(Faster Transformer v4.0)相比,EET在使用A100 GPU的Transformer解码器层上可以实现平均1.40- 4.20倍的加速

英文摘要:

Recently, large-scale transformer-based models have been proven to be effective over various tasks across many domains. Nevertheless, applying them in industrial production requires tedious and heavy works to reduce inference costs. To fill such a gap, we introduce a scalable inference solution: Easy and Efficient Transformer (EET), including a series of transformer inference optimization at the algorithm and implementation levels. First, we design highly optimized kernels for long inputs and large hidden sizes. Second, we propose a flexible CUDA memory manager to reduce the memory footprint when deploying a large model. Compared with the state-of-the-art transformer inference library (Faster Transformer v4.0), EET can achieve an average of 1.40-4.20x speedup on the transformer decoder layer with an A100 GPU

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

↑