CPT: a pre-trained unbalanced transformer for both Chinese language understanding and generation

Yunfan Shao,Zhichao Geng,Yitao Liu,Junqi Dai,Hang Yan,Fei Yang,Zhe Li,Hujun Bao,Xipeng Qiu
DOI: https://doi.org/10.1007/s11432-021-3536-5
2024-04-13
Science China Information Sciences
Abstract:In this paper, we take the advantage of previous pre-trained models (PTMs) and propose a novel Chinese pre-trained unbalanced transformer (CPT). Different from previous Chinese PTMs, CPT is designed to utilize the shared knowledge between natural language understanding (NLU) and natural language generation (NLG) to boost the performance. CPT consists of three parts: a shared encoder, an understanding decoder, and a generation decoder. Two specific decoders with a shared encoder are pre-trained with masked language modeling (MLM) and denoising auto-encoding (DAE) tasks, respectively. With the partially shared architecture and multi-task pre-training, CPT can (1) learn specific knowledge of both NLU or NLG tasks with two decoders and (2) be fine-tuned flexibly that fully exploits the potential of the model. Moreover, the unbalanced transformer saves the computational and storage cost, which makes CPT competitive and greatly accelerates the inference of text generation. Experimental results on a wide range of Chinese NLU and NLG tasks show the effectiveness of CPT.
computer science, information systems,engineering, electrical & electronic
What problem does this paper attempt to address?