PanGu-$π$: Enhancing Language Model Architectures via Nonlinearity Compensation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Yunhe, Chen, Hanting, Tang, Yehui, Guo, Tianyu, Han, Kai, Nie, Ying, Wang, Xutao, Hu, Hailin, Bai, Zheyuan, Wang, Yun, Liu, Fangcheng, Liu, Zhicheng, Guo, Jianyuan, Zeng, Sinan, Zhang, Yinchen, Xu, Qinghua, Liu, Qun, Yao, Jun, Xu, Chao, Tao, Dacheng
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929194963828736
author Wang, Yunhe
Chen, Hanting
Tang, Yehui
Guo, Tianyu
Han, Kai
Nie, Ying
Wang, Xutao
Hu, Hailin
Bai, Zheyuan
Wang, Yun
Liu, Fangcheng
Liu, Zhicheng
Guo, Jianyuan
Zeng, Sinan
Zhang, Yinchen
Xu, Qinghua
Liu, Qun
Yao, Jun
Xu, Chao
Tao, Dacheng
author_facet Wang, Yunhe
Chen, Hanting
Tang, Yehui
Guo, Tianyu
Han, Kai
Nie, Ying
Wang, Xutao
Hu, Hailin
Bai, Zheyuan
Wang, Yun
Liu, Fangcheng
Liu, Zhicheng
Guo, Jianyuan
Zeng, Sinan
Zhang, Yinchen
Xu, Qinghua
Liu, Qun
Yao, Jun
Xu, Chao
Tao, Dacheng
contents The recent trend of large language models (LLMs) is to increase the scale of both model size (\aka the number of parameters) and dataset to achieve better generative ability, which is definitely proved by a lot of work such as the famous GPT and Llama. However, large models often involve massive computational costs, and practical applications cannot afford such high prices. However, the method of constructing a strong model architecture for LLMs is rarely discussed. We first analyze the state-of-the-art language model architectures and observe the feature collapse problem. Based on the theoretical analysis, we propose that the nonlinearity is also very important for language models, which is usually studied in convolutional neural networks for vision tasks. The series informed activation function is then introduced with tiny calculations that can be ignored, and an augmented shortcut is further used to enhance the model nonlinearity. We then demonstrate that the proposed approach is significantly effective for enhancing the model nonlinearity through carefully designed ablations; thus, we present a new efficient model architecture for establishing modern, namely, PanGu-$π$. Experiments are then conducted using the same dataset and training strategy to compare PanGu-$π$ with state-of-the-art LLMs. The results show that PanGu-$π$-7B can achieve a comparable performance to that of benchmarks with about 10\% inference speed-up, and PanGu-$π$-1B can achieve state-of-the-art performance in terms of accuracy and efficiency. In addition, we have deployed PanGu-$π$-7B in the high-value domains of finance and law, developing an LLM named YunShan for practical application. The results show that YunShan can surpass other models with similar scales on benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2312_17276
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle PanGu-$π$: Enhancing Language Model Architectures via Nonlinearity Compensation
Wang, Yunhe
Chen, Hanting
Tang, Yehui
Guo, Tianyu
Han, Kai
Nie, Ying
Wang, Xutao
Hu, Hailin
Bai, Zheyuan
Wang, Yun
Liu, Fangcheng
Liu, Zhicheng
Guo, Jianyuan
Zeng, Sinan
Zhang, Yinchen
Xu, Qinghua
Liu, Qun
Yao, Jun
Xu, Chao
Tao, Dacheng
Computation and Language
Machine Learning
The recent trend of large language models (LLMs) is to increase the scale of both model size (\aka the number of parameters) and dataset to achieve better generative ability, which is definitely proved by a lot of work such as the famous GPT and Llama. However, large models often involve massive computational costs, and practical applications cannot afford such high prices. However, the method of constructing a strong model architecture for LLMs is rarely discussed. We first analyze the state-of-the-art language model architectures and observe the feature collapse problem. Based on the theoretical analysis, we propose that the nonlinearity is also very important for language models, which is usually studied in convolutional neural networks for vision tasks. The series informed activation function is then introduced with tiny calculations that can be ignored, and an augmented shortcut is further used to enhance the model nonlinearity. We then demonstrate that the proposed approach is significantly effective for enhancing the model nonlinearity through carefully designed ablations; thus, we present a new efficient model architecture for establishing modern, namely, PanGu-$π$. Experiments are then conducted using the same dataset and training strategy to compare PanGu-$π$ with state-of-the-art LLMs. The results show that PanGu-$π$-7B can achieve a comparable performance to that of benchmarks with about 10\% inference speed-up, and PanGu-$π$-1B can achieve state-of-the-art performance in terms of accuracy and efficiency. In addition, we have deployed PanGu-$π$-7B in the high-value domains of finance and law, developing an LLM named YunShan for practical application. The results show that YunShan can surpass other models with similar scales on benchmarks.
title PanGu-$π$: Enhancing Language Model Architectures via Nonlinearity Compensation
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2312.17276