← 返回资源分享
Foundation Transformers
论文
论文
发布时间2022-10-12
发表arXiv:2210.06423
作者:Zhun Liu,Yu Wu,Li Dong,Wenhui Wang,Furu Wei,Vishrav Chaudhary,Xia Song,Saksham Singhal,Shuming Ma,Payal Bajaj,Shaohan Huang,Barun Patra,Zhiliang Peng,Hongyu Wang,Alon Benhaim
详细介绍
A big convergence of model architectures across language, vision, speech, and multimodal is emerging. However, under the same name "Transformers", the above areas use different implementations for better performance, e.g., Post-LayerNorm for BERT, and Pre-LayerNorm for GPT and vision Transformers. We call for the development of Foundation Transformer for true general-purpose modeling, which serves as a go-to architecture for various tasks and modalities with guaranteed training stability. In this work, we introduce a Transformer variant, named Magneto, to fulfill the goal. Specifically, we propose Sub-LayerNorm for good expressivity, and the initialization strategy theoretically derived from DeepNet for stable scaling up. Extensive experiments demonstrate its superior performance and better stability than the de facto Transformer variants designed for various applications, including language modeling (i.e., BERT, and GPT), machine translation, vision pretraining (i.e., BEiT), speech recognition, and multimodal pretraining (i.e., BEiT-3).
代码仓库 (4)
microsoft/unilm官方PyTorch
microsoft/torchscale官方PyTorch
fkodom/dilated-attention-pytorchPyTorch
qwopqwop200/Magneto-pytorchPyTorch
