File size: 2,351 Bytes
115e671
 
 
 
 
 
c90dfa5
 
 
 
 
 
 
 
 
 
 
115e671
 
 
 
 
 
 
 
824a209
 
 
 
 
115e671
 
2699f75
115e671
 
2699f75
115e671
 
 
 
 
2699f75
115e671
824a209
 
 
 
 
d672854
 
824a209
07e64f4
 
 
 
 
 
 
 
 
c90dfa5
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
---
language: en
tags:
- hummingbird
- causal-lm
license: mit
datasets:
- wikimedia/wikipedia
- qwedsacf/grade-school-math-instructions
- HuggingFaceH4/instruction-dataset
- alespalla/chatbot_instruction_prompts
- MBZUAI/LaMini-instruction
- hendrycks/competition_math
- lighteval/MATH
- camel-ai/math
- microsoft/orca-math-word-problems-200k
pipeline_tag: text-generation
---

# 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.

<div align="center">
  <img src="figs/Hummingbird.jpg" width="400"/>
</div>


## 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](https://arxiv.org/abs/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.
<div align="center">
  <img src="figs/history.png" width="700"/>
</div>

## Team
The design and training of Hummingbird has been done jointly by [Mehran Hosseini](https://mehranhosseini.com) and [Peyman Hosseini](https://peymanhosseini.net).

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}
}
```