Exea LabsBlog
English英語版

結果

これまでの ExeaLM 対 Llama のすべての実行。bpb は低いほど良い。

ExeaLM
1.464
Llama
1.481

0.1B の ExeaLM と同サイズの Llama、同じ学習時間

01

ステップ数と計算量をそろえた比較

計算量をそろえた比較(同じ学習時間)ステップ数をそろえた比較
-0.04-0.020.00+0.02+0.04+0.06ハイブリッドが良い ↑Llama が良い ↓13M27M48M90M
48M の最初の試み(アテンション半分)0.1B、ハイブリッドの学習率 1.5e-3
モデルサイズごとの Llama に対するリード。ゼロより上ならハイブリッドが良い。
02

すべての実行

評価用テキストのバイトあたりビット数。低いほど良い。1 行が 1 回の学習です。列名を押すと並べ替えます。

#実行モデル層の構成予算学習率シード
1eqt_big_attn6TONNアテンション:1, 2, 4, 5, 7, 8畳み込み寄り2,000 秒0.001501.3651.4351.3271.6551.446
2eqt_big_attn6_s1TONNアテンション:1, 2, 4, 5, 7, 8畳み込み寄り2,000 秒0.001511.3671.4371.3281.6501.446
3eqt_big_llama_s1Llamaすべてアテンション2,000 秒0.00111.3691.4391.3331.6651.451
4eqt_big_llamaLlamaすべてアテンション2,000 秒0.00101.3701.4391.3321.6681.452
5eqt_100m_hybrid_lr1e3TONNアテンション:1, 2, 4, 5, 7, 8, 10, 11, 13畳み込み寄り3,000 秒0.00101.3731.4501.3391.6921.464
6eqt_big_hybrid_lr15e4TONNアテンション:1, 3, 5, 7, 9畳み込み寄り2,000 秒0.001501.3741.4531.3371.6991.466
7hconv_6kTONNアテンション:1, 3, 5畳み込み寄り6,000 ステップ0.00201.3901.4551.3431.6761.466
8eqt_big_hybridTONNアテンション:1, 3, 5, 7, 9畳み込み寄り2,000 秒0.00201.3791.4561.3411.7121.472
9llama_6kLlamaすべてアテンション6,000 ステップ0.00201.3951.4591.3481.6861.472
10eqt_100m_hybridTONNアテンション:1, 2, 4, 5, 7, 8, 10, 11, 13畳み込み寄り3,000 秒0.001501.3861.4621.3471.7021.474
11eqt_big_hybrid_lr1e3TONNアテンション:1, 3, 5, 7, 9畳み込み寄り2,000 秒0.00101.3771.4581.3411.7231.475
12eqt_100m_llamaLlamaすべてアテンション3,000 秒0.00101.3911.4681.3521.7131.481
13mid_hybridTONNアテンション:1, 3, 5, 7畳み込み寄り3,000 ステップ0.00211.4131.4811.3681.7021.491
14big_hybrid_3kTONNアテンション:1, 3, 5, 7, 9畳み込み寄り3,000 ステップ0.00201.3951.4771.3601.7351.492
15mid_hybridTONNアテンション:1, 3, 5, 7畳み込み寄り3,000 ステップ0.00201.4121.4841.3731.7021.493
16eqt_mid_hybridTONNアテンション:1, 3, 5, 7畳み込み寄り1,000 秒0.00211.4151.4861.3731.7011.494
17eqt_mid_hybridTONNアテンション:1, 3, 5, 7畳み込み寄り1,000 秒0.00201.4181.4921.3761.7191.502
18eqt_mid_llamaLlamaすべてアテンション1,000 秒0.00111.4181.4901.3801.7201.502
19eqt_mid_llamaLlamaすべてアテンション1,000 秒0.00101.4171.4941.3791.7301.505
20big_llama_3kLlamaすべてアテンション3,000 ステップ0.00101.4171.4961.3811.7271.505
21mid_llama_lr1e3Llamaすべてアテンション3,000 ステップ0.00111.4351.5081.3971.7411.520
22eqt_small_hybridTONNアテンション:1, 3, 5畳み込み寄り600 秒0.00211.4451.5121.3951.7381.522
23eqt_small_llamaLlamaすべてアテンション600 秒0.00201.4401.5111.3941.7451.522
24mid_hybrid_lr1e3TONNアテンション:1, 3, 5, 7畳み込み寄り3,000 ステップ0.00101.4261.5121.3901.7681.524
25hyb135TONNアテンション:1, 3, 53,000 ステップ0.00201.4491.5161.3991.7361.525
26hyb135_searchTONN+探索アテンション:1, 3, 53,000 ステップ0.00201.4481.5171.4001.7341.525
27mid_llama_lr1e3Llamaすべてアテンション3,000 ステップ0.00101.4341.5141.3981.7541.525
28hyb135_memTONN+メモリアテンション:1, 3, 53,000 ステップ0.00201.4491.5181.4021.7341.526
29hconv_25TONNアテンション:2, 5畳み込み寄り3,000 ステップ0.00211.4521.5171.3991.7431.528
30hconv_attn4TONNアテンション:1, 2, 4, 5畳み込み寄り3,000 ステップ0.00211.4481.5161.3991.7491.528
31hconv_searchTONN+探索アテンション:1, 3, 5畳み込み寄り3,000 ステップ0.00211.4481.5161.4011.7511.529
32hconv_024TONNアテンション:0, 2, 4畳み込み寄り3,000 ステップ0.00201.4511.5221.4051.7411.530
33hyb135_convTONNアテンション:1, 3, 5畳み込み寄り3,000 ステップ0.00211.4511.5191.4021.7481.530
34eqt_small_hybridTONNアテンション:1, 3, 5畳み込み寄り600 秒0.00201.4501.5241.4041.7461.531
35hconv_25TONNアテンション:2, 5畳み込み寄り3,000 ステップ0.00201.4511.5231.4061.7441.531
36hconv_searchTONN+探索アテンション:1, 3, 5畳み込み寄り3,000 ステップ0.00201.4521.5221.4061.7441.531
37hyb135_convTONNアテンション:1, 3, 5畳み込み寄り3,000 ステップ0.00201.4521.5231.4061.7451.532
38eqt_small_llamaLlamaすべてアテンション600 秒0.00211.4491.5231.4011.7541.532
39hconv_attn4TONNアテンション:1, 2, 4, 5畳み込み寄り3,000 ステップ0.00201.4531.5271.4071.7601.537
40mid_llamaLlamaすべてアテンション3,000 ステップ0.00211.4471.5221.4011.7771.537
41mid_llamaLlamaすべてアテンション3,000 ステップ0.00201.4441.5241.4021.7781.537
42hconv_memTONN+メモリアテンション:1, 3, 5畳み込み寄り3,000 ステップ0.00201.4561.5321.4101.7641.540
43hconv_memTONN+メモリアテンション:1, 3, 5畳み込み寄り3,000 ステップ0.00211.4551.5281.4101.7731.542
44hconv_024TONNアテンション:0, 2, 4畳み込み寄り3,000 ステップ0.00211.4611.5301.4111.7781.545
45archLlamaすべてアテンション3,000 ステップ0.00201.4611.5341.4131.7751.546
46llama_lr3e3Llamaすべてアテンション3,000 ステップ0.00301.4651.5381.4151.7691.547
47llama_lr1e3Llamaすべてアテンション3,000 ステップ0.00101.4681.5481.4251.7761.554
48archLlamaすべてアテンション3,000 ステップ0.00211.4731.5451.4261.8031.562
49rp4TONNすべて TONN3,000 ステップ0.00201.4751.5551.4241.8691.581
50hyb135_searchTONN+探索アテンション:1, 3, 53,000 ステップ0.00211.4551.5431.4071.9331.584
51archTONN+メモリすべて TONN3,000 ステップ0.00211.4751.5591.4211.9061.590
52hyb135_s1_curveTONNアテンション:1, 3, 53,000 ステップ0.00211.4611.5511.4121.9491.593
53hyb135_s1TONNアテンション:1, 3, 53,000 ステップ0.00211.4611.5511.4121.9501.594
54TONN+探索すべて TONN3,000 ステップ0.00201.4791.5661.4221.9481.604
55TONNすべて TONN3,000 ステップ0.00201.4831.5701.4241.9591.609
56TONN+探索すべて TONN3,000 ステップ0.00211.4741.5631.4171.9931.611
57TONNすべて TONN3,000 ステップ0.00211.4771.5671.4192.0001.616
58archTONN+メモリすべて TONN3,000 ステップ0.00201.4911.5781.4331.9641.616
03

学習曲線

ExeaLM ハイブリッドLlama
4.05.06.07.08.005001000150020002500学習時間(秒)ExeaLM ハイブリッド 3.835Llama 3.963
時間に対する学習損失。48M に切り替えると 2:1 構成の最初の勝利が見られます。
04

速度

1 層の順伝播と逆伝播(ミリ秒)
GI127 → 25 ms
SM145 → 45 ms
改善前プレフィックススキャン後アテンションブロック 18 ms
トークンあたりの学習コスト(Llama を 1 とした値、0.1B)
純粋な TONN1.54× / 2.66×
ハイブリッド(アテンション半分)1.13× / 1.66×
ハイブリッド 2:1(標準)1.01× / 1.49×
メモリ回路は休止検索ありLlama = 1
プレフィックススキャン前後の回路の時間と、Llama に対するトークンあたりのコスト。
05

実験の流れ

03:0006:0009:0012:0015:0018:00ハイブリッド層(ランダムな初期配置)畳み込み寄りの初期配置学習率と長い学習大きいモデル(ステップ数をそろえる)計算量をそろえた比較層の並べ方ハイブリッドでの検索とスウォーム速度
Mac で動いたすべての実行を開始時刻順に。