この研究は、何を調べてきたのか
What this research investigates
小型言語モデルの部品と学習条件を変え、何が予測に役立つかを調べた実験記録です。最初に、構造と評価の読み方を説明します。
A starting point for understanding the models, their training, and the measurements in this notebook.
研究の問い
The research question
この研究では、少量の英語データで学習する小型言語モデルを作り、状態の幅、層数、FFNの幅、情報を戻す部品、学習率を変えてきました。中心となる問いは、モデル内部で一度計算した情報を次の処理に使うことで、限られた規模でも言語予測を改善できるか、というものです。
主系列はBabyLM 2026 Strict-Smallのデータを使った因果言語モデルです。文章の現在までの部分から、次のトークンを予測します。対話向けの指示学習を施したチャットモデルとは目的が異なります。別のデータや評価条件を使った古い系列も台帳に残っていますが、比較表では条件群を分けています。
These experiments study small English language models trained with limited data. They vary state width, depth, FFN width, feedback mechanisms, and learning rates. The central question is whether information already computed inside a model can be reused to improve later prediction within a small parameter and data budget.
The main series uses BabyLM 2026 Strict-Small data for causal next-token prediction. These are pretrained completion models. Older series with different data or evaluation conditions remain in the catalogue, in separate comparison cohorts.
最初に知っておく言葉
A few terms before starting
トークンは、文章をモデルへ渡すために分割した単位です。一語と一致するとは限らず、単語の一部や記号になることもあります。状態ベクトルは、その位置までに読んだ内容を計算中に表す数値の列です。重みは学習中に更新するモデルの設定値、活性は文章を処理するたびに変わる中間の数値です。
学習率(LR)は、学習で重みをどれくらい動かすかを調整する係数です。重み減衰(WD)は重みを少し縮める仕組みです。大きくすれば必ず良くなるものではありません。「主要行列」「その他」はこの実験群で使ったparameterの分類で、すべてのモデルに共通する標準用語ではありません。
Query・Key・Valueは、情報を検索するための三つの役割です。Queryは今探したい情報の特徴、Keyは照合される特徴、Valueは取り出す情報です。いずれもモデルが数値ベクトルとして作ります。行列型の記憶ではKeyとValueの対応を一枚の行列へ重ねて保存するため、過去の文章を完全にそのまま保管する仕組みではありません。
A token is a unit into which text is split for the model. It can be a word fragment or punctuation rather than a whole word. A state vector is a sequence of numbers representing information at a position during processing. Weights are model parameters updated during training; activations are intermediate values that change with the input.
Learning rate (LR) controls the scale of training updates. Weight decay (WD) gradually shrinks weights. Larger values are not necessarily better. “Main matrices” and “other parameters” are experiment-specific groupings, not standard categories shared by every language model.
Query, Key, and Value describe roles in retrieval. A Query represents what is being sought, a Key supplies features to match against, and a Value contains information to retrieve. The model generates these as vectors. A matrix memory superposes Key–Value associations, so it does not preserve every past sentence losslessly.
本体の基本部品
The basic model
各トークンは数百個の数値からなる状態ベクトルで表されます。Attentionは現在までの位置から関連情報を集め、FFNはその状態の特徴を加工します。この組を何層も重ねます。「状態幅320」はベクトルが320次元、「FFN幅1120」はFFNの途中で1120次元へ広げるという意味です。
VISTARCでは、前の層の処理前後の差分を次の層のFFNへ渡します。また、前の位置で出した候補予測の平均・分散などを次の位置へ戻します。最後のMoS3は三つの出力枝が作った確率分布を混ぜる部品です。各モデルがどの部品を持つかは詳細表で確認できます。
A token is represented by a vector with a few hundred coordinates. Attention retrieves information from the current and earlier positions; an FFN transforms the resulting features. Repeating these operations forms the model layers. A state width of 320 means 320 coordinates, while FFN width 1120 means the intermediate FFN representation has 1120 coordinates.
VISTARC passes changes made by one layer into the next layer’s FFN. It also feeds statistics of the previous position’s candidate predictions into the next position. MoS3 combines three output probability distributions. The model details identify which components each experiment actually contains.
途中で生まれた別系列
Related architecture families
後期のF付き系列は、予測と実際に来たトークンの食い違いを小さな行列へ蓄え、現在状態に応じて読み出す部品を追加しました。Fの16×16などは行列の特徴次元で、16トークン前までの履歴という意味ではありません。
さらに、D/C系列では役割を分けた状態行列を試しました。DはAttentionやFFNによる自己状態の変化、Cは候補予測の情報を記憶するためのものです。公開モデル#1は、この後期のFやD/C行列を持ちません。似た「帰還」「記憶」という名称でも、計算と保存内容は異なります。
A later family adds F, a small matrix memory of differences between predictions and arriving tokens, read according to the current state. A shape such as 16×16 describes feature dimensions, not a history of sixteen tokens.
The D/C family explores a different division: D stores changes produced by Attention and FFN processing, while C stores candidate-prediction information. Released model #1 has neither the later F module nor D/C matrices. These families reuse information in materially different ways.
読む順序
How to read the notebook
初めて読む場合は、次の「点数と比較条件」を読み、その後にseed、幅と学習率、深さ、FFN比率の章へ進むと全体をつかめます。各章のモデル番号は順位ではなく固定IDです。同じ番号から、そのモデルの部品、実際の学習率、学習時点別の結果を開けます。
本文には確認できた観測と、その範囲での解釈を書いています。再現できる法則が見つからなかった試行や、予想に反した結果も含めています。全302個のIDと988件の評価地点を残し、採点がないものを0点として扱わない方針です。
Start with the measurement guide, then read the chapters on seeds, width and learning rate, depth, and FFN ratio. A model number is a permanent identifier, not a rank. Opening it reveals components, recorded effective learning rates, and results by training age.
The articles distinguish observations from interpretations and retain unsuccessful experiments. The catalogue contains 302 permanent identities and 988 recorded evaluation points. Missing scores are not treated as zeros.