language:
- en
library_name: transformers
license: llama2
Background
Sparse computation is increasingly recognized as an important direction in enhancing the computational efficiency of large language models (LLMs).
Prior research has demonstrated that LLMs utilizing the ReLU activation function exhibit sparse activations. Interestingly, our findings indicate that models based on SwiGLU also manifest sparse activations.
This phenomenon prompts an essential question: Which activation function is optimal for sparse LLMs? Although previous works on activation function selection have focused on the performance of LLMs, we argue that the efficiency of sparse computation should also be considered so that the LLMs can proceed with efficient inference while preserving performance.
To answer this question, we pretrain 4 LLMs with different activation functions, including ReLU, SwiGLU, ReGLU, and Squared ReLU to do more comprehensive experiments.
Dataset
We pretrain the model on 100 billion tokens, including:
- Refinedweb
- SlimPajama
Training Hyper-parameters
| Parameter | Value |
|---|---|
| Batch_Size | 4M |
| GPUs | 64xA100(80G) |
| LR_Scheduler | cosine |
| LR | 3e-4 |
Citiation:
Please kindly cite using the following BibTeX:
@misc{sparsellm,
title={Sparse Large Language Models with ReLU Activation},
author={SpaseLLM Team},
year={2023}
}