This is Hummingbird 0.0, a 1B proof-of-concept causal language model based on Efficient Attention, which reduces the number of parameters in the attention layer by 50% compared to standard Multi-Head Attention.
This version of Hummingbird is only meant to demonstrate Efficient Attention for use in causal language modelling. It has been trained on only 15 Billion tokens and is not safeguarded. Therefore, we do not recommend using it as a chatbot.
Model Details
The model consists of 1.1 Billion parameters with the following specifications:
Parameter
size
# Transformer Blocks
10
Model Dimension
3072
# Heads
1
The Attention Mechanism used is based on our newly proposed Efficient Attention from our paper, You Need to Pay Better Attention: Rethinking the Mathematics of Attention Mechanism (arXiv:2403.01643). We have chosen the number of heads to be 1 as an interesting case study since all current LMs use multiple heads.
The loss plot below illustrates the model's performance during training. For comparison, when trained on 15 billion tokens, Hummingbird achieves a slightly lower loss than TinyLlama, a model of similar size.
If you use Efficient Attention or Hummingbird, please cite our paper:
@article{Hosseinis24BetterAttention,
title = {You Need to Pay Better Attention: Rethinking the Mathematics of Attention Mechanism},
author = {Hosseini, Mehran and Hosseini, Peyman},
journal = {arXiv preprint arXiv:2403.01643},
year = {2024}
}
