表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

小規模言語モデルの事前学習でReLoRAを検証:11M・66M規模で通常学習と比較

ReLoRA

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理
原論文の図

ReLoRAの原論文から掲載した図です。細部はクリックして拡大できます。

原論文の図の説明を読む

Figure 2: Training trajectories of cross-entropy loss, Paloma perplexity and BLiMP score across the tiny and small models, plotted against GPU hours taken

出典:ReLoRA / arXiv ↗

概要

  • 著者らは11M・66Mパラメータの小規模言語モデルで、ReLoRAと通常の全ランク学習を各1回比較した。
  • 学習損失、Palomaのパープレキシティー、BLiMPではReLoRA版が概して下回り、66M規模で差が広がったと報告した。
  • 設定は1組で、全ランク学習からの開始も省かれている。導入判断には対象文書での再現試験が必要である。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、ReLoRAの仕組み

左から右へ読む
  1. 01英語の学習文
  2. 02低ランク行列を更新
  3. 03重みに統合・再開始
  4. 04学習済みモデル

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

ReLoRAは、言語モデルの事前学習で各回の更新を小さな行列の組に絞り、学習した内容を元の重みに繰り返し統合する方法である。累積した更新が、個々の小さな更新より多くの変化の方向を扱えることを狙う。

一方、先行研究では、小規模言語モデルの重みは利用できる次元を十分に使えていないと指摘されていた。更新をさらに絞る方法がこの制約を和らげるのか、強めるのか。著者らは11M・66Mパラメータのモデルで調べた。

手法

ReLoRAでは、モデルの線形計算を担う重み行列に、学習可能な小さな行列の組を付ける。低ランクとは、1回の更新で扱う変化の方向を少数に絞ることをいう。例えば英文から次の語を予測し、予測の誤りを基にこの行列の組を更新する。学習の出力は、続きの語を予測できる言語モデルである。

一定のステップごとに行列の組を元の重みに足し、新しい組で学習を再開する。実験では注意機構とフィードフォワード層の線形部分に導入し、ランクを16、再開始間隔を2,000最適化ステップに設定した。著者らは性能に加え、重みが使う方向の広さを表す比例実効ランクと、小さな変化に対する計算の敏感さを表す条件数を調べた。

新規性

著者らの新規性はReLoRAの発明ではなく、11M・66Mパラメータの小規模言語モデルで性能と学習中の状態を併せて調べた点にある。通常学習との比較に加え、重みと更新の比例実効ランク、条件数を追跡した。

低ランクの更新を繰り返し統合すれば、累積した更新がより多くの方向を扱えるというReLoRAの狙いに対し、この設定ではランクの低下と学習初期の条件数の上昇を報告した。

従来手法との違い

比較対象は、同じ系統のデコーダーを通常の全ランク更新で事前学習するpico-decoderである。全ランク更新では、ReLoRAのように各回の更新を小さな行列の組に制限しない。著者らは各規模で両方式を1回ずつ学習した。

著者らによると、11M規模では性能差が小さく、66M規模ではReLoRA版の不利が目立つ。ただし、対応するモデル間でも学習可能なパラメータ数と総パラメータ数は異なる。この2規模と単一の設定から、別のモデル構造やReLoRA設定での優劣までは判断できない。

実験結果

著者らはDolmaの英語テキストで、11M・66M規模の通常版とReLoRA版を各1回、20,000ステップ学習した。各モデルが学習中に見た量は41.9Bトークンと記されている。学習損失、Palomaから作った検証用の部分集合でのパープレキシティー(次の語の予測しにくさ)、BLiMPで文法的に適切な英文を選ぶ割合を評価した。

著者らはReLoRA版の成績が概して低く、66M規模で差が広がったと報告する。比例実効ランクは重みと更新で概して低く、更新の条件数は学習初期に高かった。ただし、ReLoRA側の更新では元の重みに勾配がないため、小さな行列の重みを代理指標として使っている。確認できた本文の範囲では、各性能指標の最終値を一覧にした数表はない。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案:社内の英語文書を対象に、小規模な文章補完モデルの学習方式を選ぶ比較実験へ使う。利用許可を確認した社内文書を学習用と未使用の評価用に分け、文書の冒頭を入力、続きの語の予測と評価記録を出力とする。

次の語を予測する学習では文章自体が正解となり、追加の人手ラベルは不要である。文法選択も調べるなら、正解を付けた英文の対を別途用意する。センサーは不要。社内文書の語彙や書式に合わせてテキストの分割と語彙処理を調整する必要がある。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案:利用許可を確認した英語文書を学習用と未使用の評価用に分け、通常の全ランク更新版とReLoRA版を同じ文書量・ステップ数で比較する。文書の分割と語彙処理をそろえ、ReLoRAのランク、再開始間隔、開始方法を記録する。

次の語を予測する学習に追加の人手ラベルやセンサーは不要である。未使用文書で予測損失を測り、文法選択を用途に含める場合だけ、正解を付けた英文の対も用意する。学習時間、再開始直後の損失、比例実効ランクを併せて確認し、複数回の実行で差の再現性を調べる。

限界と課題

著者らが明記する制約は、各設定の学習が1回だけであること、モデルが2規模・単一のデコーダー構造に限られること、ReLoRAの設定を1組しか調べていないことである。元のReLoRAに含まれる全ランク学習からの開始も省いている。

編集上の検証事項は、乱数初期値を変えて差が再現するか、ランクや再開始間隔、開始方法で結果が変わるかである。英語のDolma、Paloma、BLiMPで見られた傾向を、社内文書や日本語にそのまま当てはめる根拠も、この研究にはない。

応用先の候補
研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

Investigating ReLoRA: Effects on the Learning Dynamics of Small Language Models

Parameter-efficient methods like LoRA have revolutionised large language model (LLM) fine-tuning. ReLoRA extends this idea to pretraining by repeatedly merging and reinitialising low-rank adapters, increasing cumulative rank while keeping updates cheap. This aligns well with observations that high-capacity models learn through locally low-rank trajectories that expand over time. By contrast, recent work suggests that small language models (SLMs) exhibit rank deficiencies and under-utilise their available dimensionality. This raises a natural question: can ReLoRA's rank-expanding update rule \textit{steer} SLMs toward healthier learning dynamics, mitigating rank bottlenecks in a capacity-constrained regime? We argue SLMs are an ideal testbed: they train quickly, enable controlled ablations, and make rank phenomena more measurable. We present the first systematic study of ReLoRA in SLMs (11M-66M parameters), evaluating both performance and learning dynamics. Across loss, Paloma perplexity, and BLiMP, we find that ReLoRA underperforms full-rank training, with gaps widening at larger scales. Analysis of proportional effective rank and condition numbers shows that ReLoRA amplifies existing rank deficiencies and induces ill-conditioned updates early in training. Our results suggest that while ReLoRA's merge-and-restart strategy can expand ranks in larger models, it does not straightforwardly translate to capacity-limited SLMs, motivating adaptive-rank or hybrid-rank approaches for low-compute pretraining.

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-10-01