Hummingbird 0.0 Release

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.

Team

The design and training of Hummingbird has been done jointly by Mehran Hosseini and Peyman Hosseini.

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}
}
Downloads last month
19
Safetensors
Model size
1.14B params
Tensor type
BF16
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Datasets used to train PeymanHosseini/Hummingbird