Accelerating Training of Transformer-Based Language Models with Progressive Layer Dropping

作者: Minjia Zhang, Yuxiong He

摘要

Recently, Transformer-based language models have demonstrated remarkable performance across many NLP domains. However, the unsupervised pre-training step of these models suffers from unbearable overall computational expenses. Current methods for accelerating the pre-training either rely on massive parallelism with advanced hardware or are not applicable to language models.

PDF打开原文