日本フィジカルAI新聞

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

週刊ニュースレター購読
最適制御arXiv:2610.40427

JAXによるGPU加速確率制約付き確率的最適制御ソルバsocx

socx: GPU-Accelerated Chance-Constrained Stochastic Optimal Control using JAX

シェア:XThreadsFacebookLINEはてブBluesky

線形確率最適制御の確率制約付き凸緩和問題をGPU上で高速に解くJAX実装のソルバを提案し、分散最適化により並列化してMOSEK比最大98倍、SCS比最大2224倍の高速化を達成した。

詳しい要約

1. どんなもの?

- socx とは - 線形 stochastic optimal control の chance constraints 付き問題を、lifted convex reformulation として解く GPU 加速ソルバ - JAX で実装されている - 対象は linear stochastic optimal control の chance-constrained 版 - 提供するもの - GPU 上での高速求解 - 大規模 horizon への対応 - 分散最適化に基づく分解解法

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

- 比較対象 - 汎用凸最適化ソルバ MOSEK および SCS と比較 - 速度面 - MOSEK 比で最大 98× の speed-up - SCS 比で最大 2224× の speed-up - スケール面 - socx は horizon N=1120 まで求解可能 - MOSEK は N=320 で out of memory - 位置づけ - lifted convex reformulation を GPU 上で分散的に解く点が特徴

3. 技術・手法の肝は?

- 問題構造の活用 - 問題の sparse かつ independent な構造を利用 - 分散最適化による分解 - 元問題をより単純な部分問題へ分解 - 独立な linear-quadratic tracking (LQT) 問題 - 集合への Euclidean projections - 並列化と結合 - 部分問題を並列に解く - その後 constraints を用いて結合 - 実装基盤 - JAX による GPU 実装

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

- 検証対象 - hover-linearized 3D quadrotor - 比較実験 - MOSEK および SCS との速度比較 - 最大 98× (対 MOSEK)、最大 2224× (対 SCS) の speed-up を報告 - スケール検証 - socx は N=1120 まで求解 - MOSEK は N=320 で out of memory - 要因分析 - speed-up を可能にする主要コンポーネントを特定する ablation を実施

5. 議論はある?

- 報告されている議論 - sparse かつ independent な構造を利用した分散最適化の有効性 - GPU 加速と分解による計算効率・メモリ効率の改善 - ablation による speed-up 要因の切り分け - 要旨からは不明 - 分解の収束性や最適性保証に関する詳細な議論 - quadrotor 以外の系への一般化可能性 - chance constraints の違反率などの実証的評価

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

- 要旨で直接参照/比較されている研究 - MOSEK - SCS - 関連手法として挙げられるもの - linear-quadratic tracking (LQT) - lifted convex reformulation - chance-constrained stochastic optimal control - Euclidean projection - 同分野の定番として検討しうるもの - model predictive control (MPC) - sequential convex programming

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

著者: Tanmay Dokania, Yashwanth Kumar Nakka

分類: eess.SY, cs.RO

原文アブストラクト

In this article, we present socx, a GPU-accelerated solver for lifted convex reformulations of linear stochastic optimal control under chance constraints, implemented in JAX. We demonstrate that by exploiting the sparse and independent structure of the problem, distributed optimization can reduce it to simpler problems: independent linear-quadratic tracking (LQT) problems and Euclidean projections onto sets. These simpler problems can be solved in parallel and then coupled using the constraints. To validate our solver, we run it on a hover-linearized 3D quadrotor and show a speed-up of up to $98\times$ over MOSEK and up to $2224\times$ over SCS. socx also solves horizons up to $N=1120$, whereas MOSEK runs out of memory at $N=320$. We present ablations to isolate the key components that enable the speed-up.

関連論文

PR本紙発行元 EmplifAI