本文にスキップ
AI News HubLIVE
サイト内リライト2 分で読了

Rustで言語モデルをエンドツーエンド訓練:経験報告

記事の要約

著者はチームもPyTorchも訓練経路のPythonも使わず、純粋なRustだけで約0.4Bパラメータのベンガル語優先言語モデルをエンドツーエンド事前訓練し、レンタルGPU代は164ドルだった。論文の主な貢献は、Rustの訓練バックエンドとしてのCandleとBurnの欠陥分類(それぞれ5件と3件、無勾配の融合カーネルや数十億パラメータ規模での訓練中セグフォルトを含む)と、6件の静かな失敗を検出した勾配フロー検証テストである。著者はRustは訓練にはまだ競争力がないが、サービングには適する可能性があると結論づけている。

ソースarXiv Computational Linguistics著者: Arif Adito
Rustで言語モデルをエンドツーエンド訓練:経験報告
誤りを報告

訂正窓口はまだ利用できません。記事情報をコピーして保存できます。

訂正案内
本文へ

このarXiv論文はArif Adito氏によって2026年7月27日に提出され、arXiv:2609.25008、cs.CL(計算と言語)に分類されている。著者は、チームを組まず、PyTorchも使わず、訓練経路にPythonも一切挟まず、純粋なRustだけで言語モデルをエンドツーエンドで事前訓練した経験を報告している。レンタルGPUの費用は合計164ドルだった。著者はこれを推奨ではなく一つの達成として位置づけ、より有用な貢献は、2026年時点で主要なRust機械学習フレームワークであるCandleとBurnを「推論」ではなく「訓練」バックエンドとして使った場合の、計測に基づく失敗分類だとする。

論文ではCandleの5つの欠陥が記録されている。その一つは、融合カーネルが静かに勾配をまったく生成しないというものだ。Burnについては3つの欠陥が報告され、逆伝播が理論上のGPUスループットの約3%しか出ないことや、カーネル融合経路が数十億パラメータ規模の訓練中にセグメンテーション違反を起こすことが含まれる。これらの欠陥はどれも自ら名乗り出ることはなく、通常の損失曲線の検査をすべて通過してしまった。これに対処するため、著者は検証規律を記述している。中心となるのは「勾配フローアービター」と呼ばれるテストで、1回の順伝播と逆伝播を実行し、すべての訓練可能パラメータが有限で非ゼロの勾配を受け取ることを表明する。このテストは任意のフレームワークに一般化でき、6件の同種の静かな失敗を捕まえた。

訓練されたモデルは約0.4Bパラメータで、ベンガル語を優先している。ベンガル語の言語モデリングでは明確な信号を示し、トークンあたり負の対数尤度は0.93だった。ランダムに初期化した双子モデルでは12.60である。一方、英語の常識多肢選択ではチャンスレベルにとどまる。これは意図的に小さくベンガル語に重みを置いた予算、すなわち約20億トークン、54.6時間、レンタルしたH100 1枚という条件から予想される結果である。著者はさらに、ベンガル文字における「トークナイザ肥沃性トラップ」を報告している。素朴なバイトレベルトークナイズでは、ベンガル語は約1.4文字/トークンに圧縮され、英語の3.9文字/トークンと比べてコーパスの言語バランスを静かに逆転させてしまう。修正後は約4.1文字/トークンに達した。

著者の知る限り、これは純粋なRustによるエンドツーエンド言語モデル事前訓練の最初期の記録の一つである。ただし今回の実行後、著者は訓練をPyTorchに移し、Rustはオンデバイスサービングのために残した。著者の手元では、Rustはまだ言語モデルを訓練する競争力のある場所ではないが、言語モデルを提供する場所としては良い可能性がある、というのが結論である。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • レンタルGPU代164ドルで、チームなし、PyTorchなし、訓練経路にPythonなしの純粋なRustによるエンドツーエンド事前訓練を実施。
  • Candleの5つの欠陥とBurnの3つの欠陥を文書化。融合カーネルが静かに勾配を生まない問題や、数十億パラメータ規模で訓練中にセグフォルトする経路を含む。
  • 「勾配フローアービター」を提案:1回の順伝播・逆伝播で全ての訓練可能パラメータが有限かつ非ゼロの勾配を受け取ることを検証し、任意のフレームワークに一般化可能。
  • 約0.4Bのベンガル語優先モデルはトークンあたりNLL 0.93(ランダム初期化の双子は12.60)、英語常識選択はチャンスレベル。ベンガル文字のトークナイザ肥沃性トラップ(1.4→4.1文字/トークン)も報告。

要点と分析は自動生成され、誤りを含む場合があります。原典をご確認ください。