NVIDIAは、自動運転開発者向けの車載推論モデルAlpamayo 2 Edgeを公開した。周囲のカメラ映像とナビゲーション入力から走行の根拠と将来の経路を生成する一方、通常時は推論を減らして車載コンピューターの計算負荷を抑える。自動運転開発者は、閉ループ学習と展開ツールを併用して車載モデルを試験できる。
1. 推論を減らし、走行結果から再び学習
このモデルはNVIDIA Cosmos 3 Edgeを基盤とする20億パラメーターのモデルで、映像・言語の解釈を走行行動につなげる。複数方向のカメラ映像、過去の動き、ナビゲーション入力を受け取り、走行判断の理由と将来の経路を生成する。
日常的な走行状況では、判断の根拠を長く生成する代わりに<NOMINAL>トークンを出力する。必要な場面で推論を生成する適応型の方式により、不要なトークン生成を減らし、推論計算を節約する。
学習では、Alpamayo 2 Superが生成した推論を小型モデルに伝える蒸留に続き、AlpaGymシミュレーションで閉ループ強化学習を行う。蒸留とは、大型モデルの知識を小型モデルに移す学習である。閉ループ強化学習は、モデルが出力した行動の結果を再び経験し、報酬によって改善する方式である。
事後学習は、事前学習の後に性能を追加調整する過程である。同社発表によると、この閉ループ強化学習により、同じ強化学習を行っていない最高性能モデルに比べ、閉ループ走行性能が48.9%向上した。
2. シミュレーション比較と車載時の遅延
AlpaSimの閉ループ評価は、モデルが出力した行動をシミュレーションに反映し、走行結果を調べる。
同社発表によるAlpamayo 2 Edge(2B)とQwen-Drive-1.0(4B)の過失事故間の走行距離は、次のとおりである。NVIDIAは約40倍の差と説明した。
記録された走行データから将来の経路を予測する、開ループ評価の最小平均変位誤差も併せて示した。予測経路の誤差を表し、低いほど良い指標である。
表:2つのモデルの走行評価比較
条件:NVIDIA発表のAlpamayo 2 Edge(2B)とQwen-Drive-1.0(4B)の比較。閉ループと開ループは別々の評価であり、指標ごとに2つのモデルを比較する。
単位:過失事故間の走行距離はkm、最小平均変位誤差はm
| 評価方式 | 指標 | Alpamayo 2 Edge(2B) | Qwen-Drive-1.0(4B) | 優劣の方向 |
|---|---|---|---|---|
| AlpaSim閉ループ | 過失事故間の走行距離(km) | 9.87 | 0.25 | 高いほど良い |
| 開ループ | 最小平均変位誤差(m) | 0.53 | 1.68 | 低いほど良い |
チャート:過失事故間の走行距離 条件:上表のAlpaSim閉ループ評価。高いほど良く、棒の長さは値に比例する。単位:km
Alpamayo 2 Edge(2B) ███████████████████▊ 9.87
Qwen-Drive-1.0(4B) ▌ 0.25チャート:最小平均変位誤差 条件:上表の開ループ評価。低いほど良く、棒の長さは値に比例する。距離のチャートとは軸を共有しない。単位:m
Alpamayo 2 Edge(2B) █████▎ 0.53
Qwen-Drive-1.0(4B) ████████████████▊ 1.68車載性能は別の構成で測定した。同社発表によると、DRIVE AGX Thor-X上でTensorRT 11.3.1.9により最適化したモデルは、映像入力から経路出力まで、推論量に応じて160~190msを要した。測定にはDRIVE OS 7.2.7.0とCUDA 13.4を使用した。
Embedlの共同創業者兼最高経営責任者ハンス・サロモンソン氏は、AGX Thorで48時間以内に毎秒10回の走行計画を実行したと述べた。
3. 車載最適化は別途の展開段階
Hugging Faceから取得するモデルの重みは、基本的にBF16という数値精度形式で保存されており、対応するNVIDIA GPUでPyTorchモデルとして実行できる。車内でリアルタイムに使うには、さらに最適化が必要になる。
展開レシピは、DFlash2ドラフトモデルの準備とFP8キャリブレーションから始まる。補助モデルで推論生成を高速化し、数値精度を下げた演算に合わせて調整する過程である。
続いて、展開ツールに移せるよう変換したモデル表現であるONNXグラフを出力し、TensorRT実行エンジンを作成した後、展開プログラムを起動する。
展開レシピでは、同じ入力と同一に初期化した行動ノイズを用い、最適化前後の出力品質も比較する。
展開バンドルを実行する原文のコマンドは、次のとおりである。
python -m alpamayo2edge_edgellm.runtime.run \
--engines "$WORK/engines" \
--model "$MODEL" \
--reference "$INPUTS" \
--dflash2 \
--rolling \
--warmup 1 \
--repeat 5 \
--output "$WORK/results.json"4. 推論品質を学習データと報酬につなげる
NVIDIAのAlpaJudgeは、走行経路の誤差とは別に説明の品質を評価する、人間の判断に合わせて調整された評価・報酬モデルである。参照ありの方式は、質問、正解、候補回答を比較する。参照なしの方式は、前方の広角・望遠カメラのフレーム、過去・未来の動作文脈、候補推論を用い、評価項目別のスコアと総合スコアを出す。
このスコアは、データ選別と事後学習にもつながる。同社発表によると、スコアに基づく選別後、学習に有用な中間スコアのサンプルの割合は27%から81%に増えた。
公開された強化学習レシピは、モデルの推論と経路を評価し、経路誤差・乗り心地の指標に推論スコアを加え、1つの報酬にまとめる。同社発表によると、Alpamayo 2 Superベースの強化学習前モデルで測定した推論品質の向上は、評価モデルによる採点で10.2%、人間による評価で11.4%だった。
推論量を減らして車内で素早く経路を得る一方、開発チームには数値精度の調整、実行エンジンの作成、出力品質の比較という作業が加わる。モデルを入手する段階と、車内でリアルタイムに動かす段階の間に、最適化コストが存在することになる。
AlpaJudgeのスコアは説明が評価基準に合うかを示すため、衝突・道路逸脱などの走行指標と併せて確認すべきである。高スコアの事例は、モデルがすでに十分説明でき、学ぶ余地が少ない場合もある。低スコアには誤ラベルだけでなくまれな失敗も混じり、一括除去すると必要な学習事例が失われかねない。中間スコア帯には追加学習に有用な事例が多く、選別する価値がある。
要点は、必要な瞬間に推論を集中させ、車載展開と評価・再学習を1つの開発の流れにつなげることにある。
