arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2202.07959cs.CL
收藏

EdgeFormer:一种用于器件上Seq2seq生成的参数高效Transformer

EdgeFormer: A Parameter-Efficient Transformer for On-Device Seq2seq Generation

Microsoft(微软)

Tao Ge, Si-Qing Chen, Furu Wei

摘要:

我们介绍了EdgeFormer --一个参数高效的Transformer,用于在严格的计算和内存限制下生成设备上的seq 2seq。与以前的参数高效的Transformers相比,EdgeFormer应用了两个新的原则,成本效益的参数化,使其能够在相同的参数预算下表现更好;此外,EdgeFormer还通过层自适应创新进一步增强,该创新旨在改善具有共享层的网络。大量的实验表明,EdgeFormer可以有效地优于以前的参数高效的Transformer基线,并在计算和内存限制下取得有竞争力的结果。鉴于有希望的结果,我们发布了EdgeLM-EdgeFormer的预训练版本,这是第一个公开可用的预训练的设备上seq 2seq模型,可以很容易地针对seq 2seq任务进行微调,并获得强大的结果,从而在实践中促进设备上seq 2seq生成。

英文摘要:

We introduce EdgeFormer -- a parameter-efficient Transformer for on-device seq2seq generation under the strict computation and memory constraints. Compared with the previous parameter-efficient Transformers, EdgeFormer applies two novel principles for cost-effective parameterization, allowing it to perform better given the same parameter budget; moreover, EdgeFormer is further enhanced by layer adaptation innovation that is proposed for improving the network with shared layers. Extensive experiments show EdgeFormer can effectively outperform previous parameter-efficient Transformer baselines and achieve competitive results under both the computation and memory constraints. Given the promising results, we release EdgeLM -- the pretrained version of EdgeFormer, which is the first publicly available pretrained on-device seq2seq model that can be easily fine-tuned for seq2seq tasks with strong results, facilitating on-device seq2seq generation in practice.

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

↑