日本フィジカルAI新聞

世界のフィジカルAIを、日本語で。

運動予測・計画arXiv:2506.08228

運動予測と計画のスケーリング則

Scaling Laws of Motion Forecasting and Planning -- Technical Report

シェア:XThreadsFacebookLINEはてブBluesky

自動運転における運動予測と計画タスクで、トランスフォーマーモデルの性能が計算量に対してべき乗則で向上することを示し、最適なモデルサイズとデータサイズの比率や推論時計算のスケーリングも分析した。

詳しい要約

1. どんなもの?

- 自動運転領域における joint motion forecasting and planning タスクを対象に、encoder-decoder autoregressive transformer モデル群の経験的 scaling law を調べた技術報告。 - 50万時間の走行データセットを用い、言語モデリングと同様に性能が総 compute budget の power-law 関数として改善することを示す。 - 学習 loss と評価指標の強い相関、closed-loop 指標も scaling で改善する点、training compute-optimal なモデルサイズとデータサイズの最適比率、inference-time compute scaling を扱う。

2. 先行研究と比べてどこがすごい?

- 言語モデリングで知られる scaling law を、自動運転の joint motion forecasting and planning に大規模データで適用した点が特徴。 - 学習 loss と評価指標の相関に加え、closed-loop 指標も scaling で改善することを示し、open-loop 指標のモデル開発・hill climbing への適合性に重要な示唆を与える。 - 先行研究との具体的な比較は要旨からは不明。

3. 技術・手法の肝は?

- encoder-decoder autoregressive transformer を用い、50万時間の走行データセットで学習。 - 総 compute budget に対する性能の power-law を経験的に推定。 - training compute-optimal な transformer parameters 数と training data size の最適 scaling を分析し、compute budget 増加時にモデルサイズをデータサイズの1.5倍の速さで増やす必要があると報告。 - inference-time compute scaling として、小モデルの出力を sampling と clustering する手法を検討。

4. どうやって有効だと検証した?

- 50万時間の走行データセットを用いた学習実験で、性能が総 compute budget の power-law に従うことを示す。 - 学習 loss と評価指標の強い相関を確認。 - closed-loop 指標も scaling で改善することを確認。 - sampling と clustering により小モデルが crossover point まで大モデルと競合することを観察。

5. 議論はある?

- closed-loop 指標も scaling で改善するため、open-loop 指標のモデル開発・hill climbing への適合性に重要な含意があると議論。 - inference-time compute scaling では crossover point を超えると大モデルの方が inference-compute efficient になる点を議論。 - 他エージェントの一般 logged driving data での学習が ego-agent 性能向上に有用かも簡潔に検討し、大容量モデル学習のための robotics data 不足への対処として重要と位置づけ。

6. 次に読むべき論文は?

- 要旨で参照/比較されている具体的な先行研究は明示されていない。 - 関連手法として、言語モデリングの scaling law、encoder-decoder autoregressive transformer、joint motion forecasting and planning、closed-loop 評価、inference-time compute scaling(sampling と clustering)が挙げられる。 - 同分野の定番として motion forecasting および planning の transformer ベース手法、自動運転の closed-loop 評価に関する研究を次に読む候補とする。

※ AIが要旨から生成した要約です。正確性は原文をご確認ください。

著者: Mustafa Baniodeh, Kratarth Goel, Scott Ettinger, Carlos Fuertes, Ari Seff, Tim Shen, Cole Gulino, Chenjie Yang, Ghassen Jerfel, Dokook Choe, Rui Wang, Benjamin Charrow, Vinutha Kallem, Sergio Casas, Rami Al-Rfou, Benjamin Sapp, Dragomir Anguelov

分類: cs.LG, cs.AI, cs.RO

原文アブストラクト

We study the empirical scaling laws of a family of encoder-decoder autoregressive transformer models on the task of joint motion forecasting and planning in the autonomous driving domain. Using a 500 thousand hours driving dataset, we demonstrate that, similar to language modeling, model performance improves as a power-law function of the total compute budget, and we observe a strong correlation between model training loss and model evaluation metrics. Most interestingly, closed-loop metrics also improve with scaling, which has important implications for the suitability of open-loop metrics for model development and hill climbing. We also study the optimal scaling of the number of transformer parameters and the training data size for a training compute-optimal model. We find that as the training compute budget grows, optimal scaling requires increasing the model size 1.5x as fast as the dataset size. We also study inference-time compute scaling, where we observe that sampling and clustering the output of smaller models makes them competitive with larger models, up to a crossover point beyond which a larger models becomes more inference-compute efficient. Overall, our experimental results demonstrate that optimizing the training and inference-time scaling properties of motion forecasting and planning models is a key lever for improving their performance to address a wide variety of driving scenarios. Finally, we briefly study the utility of training on general logged driving data of other agents to improve the performance of the ego-agent, an important research area to address the scarcity of robotics data for large capacity models training.

  1. 1Boston Dynamics、車輪付きAtlasの研究試作機を紹介研究・技術
  2. 2GXOとApptronik、Apolloの物流現場試験を拡大 最大1,000台の展開を構想量産・展開
  3. 3newmoとKDDIが資本業務提携 2028年に堺市で自動運転タクシー商用化目指す提携・買収
  4. 4NVIDIA、CoRLでGR00T 2とIsaac Lab 3.0を発表新製品・発表
  5. 5UC BerkeleyとNVIDIAの研究者、AIエージェントでロボット制御コードを仮想環境で改善研究・技術

PR・アフィリエイト広告

FRONTIER(GPU搭載BTOパソコン)

FRONTIER(GPU搭載BTOパソコン)の公式広告バナーFRONTIERでGPU搭載PCを見る ↗
PR本紙発行元 EmplifAI