arXiv preprint ·
socx: GPU-Accelerated Chance-Constrained Stochastic Optimal Control using JAX
- Published
- Source
- arXiv preprint, posted before peer review
- Original
- Read it at arXiv
- Topics
Abstract
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.
The authors’ abstract, as arXiv lists it. arXiv’s listings, abstracts included, are free for anyone to reuse.
Read it on ReadSearch
Each opens after you create an account, and brings you back to this paper.
More in Systems & Control
Unlocking with the Heart: Investigating User Perceptions of ECG-Based Biometric Authentication
HIPC: Humanoid Iterative Planning and Control via Goal-Conditioned Motion Closure
REALM: Identifying Minimal Task Assets for LLM Agents
Quantile Propagation Planning for Budgeted Tail-Risk Evaluation
BEVLight: Generalizable Traffic Signal Control from UAV Bird's-Eye Views
Get the few papers worth your time in Systems & Control every morning.
ReadSearch reads the new papers in your field each day and brings you the few worth reading, each with its key finding in a sentence.