日本フィジカルAI新聞

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

ソフトロボティクスarXiv:2608.06650

SoRoMoX: 高速・微分可能・並列化可能なソフトロボットモデル

SoRoMoX: Fast, Differentiable, and Parallelizable Soft Robot Models

シェア:XThreadsFacebookLINEはてブBluesky

ソフトロボットの制御に必要な微分可能でGPU並列化可能なモデリングフレームワークをJAXで実装し、従来比で最大234倍の高速化を達成した。

詳しい要約

1. どんなもの?

SoRoMoXは、Cosserat-rod理論に基づくソフトロボットの低次元モデルを、JAXを用いて実装したフレームワークである。JITコンパイル可能で、GPU上で並列実行でき、状態・入力・パラメータに関してエンドツーエンドで微分可能な、制御指向の統一インターフェースを提供する。Articulated、Piecewise Constant Strain、Variable Strainの各モデルを実装し、慣性行列、重力・弾性力、ヤコビアン、およびそれらの導関数を計算できる。

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

既存のCosserat-rodモデル実装は、微分可能・GPU並列・制御指向のワークフローをサポートしていない点が課題であった。SoRoMoXは、ロッド/ひずみベースのソフトロボットモデリングフレームワークとして初めてGPU上で直接動作し、状態・入力・パラメータに関してエンドツーエンドで微分可能である。これにより、従来は困難だった静的最適化や強化学習などのワークフローを高速に実行できる。

3. 技術・手法の肝は?

手法の核は、JAXを用いた完全数値実装とJITコンパイル、GPU並列化、自動微分の活用である。統一された制御指向インターフェースにより、異なるモデル(Articulated, Piecewise Constant Strain, Variable Strain)を同じAPIで扱える。さらに、モデルを状態・入力・パラメータに関して微分可能にすることで、勾配ベースの最適化や学習を可能にしている。

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

有効性は、複数のタスクで検証された。静的平衡システム同定ではマーカーRMSEを66%削減、残差力学習ではさらに64%削減、計算トルク追従ではモデルフリーPDと比較してRMSEを約500倍削減、制御ゲイン最適化では未調整ゲインと比較して損失を最大62%削減、安全制約付き制御では高次制御バリア関数を用いてピーク接触力を5N以内に抑え(安全制約なしでは33.5N)、強化学習ポリシー訓練ではCPUベースのPyElastica離散ロッドベースラインと比較して最大7倍高速化された。また、シーケンシャルCPUロールアウトは最先端の代替手法と比較して最大18.1倍高速であり、GPU並列ロールアウトはスループットを最大234.6倍向上させた。

5. 議論はある?

要旨からは、SoRoMoXの限界や課題についての議論は不明である。ただし、提案手法がGPU並列と微分可能性を活用することで、従来のソフトロボットモデリングの実装が抱える計算上のボトルネックを解消し、制御や学習の高度なワークフローを可能にした点が強調されている。

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

要旨で参照されている関連研究として、PyElastica(離散ロッドモデル)が挙げられる。また、Cosserat-rod理論に基づくソフトロボットモデリングの基礎論文や、JAXを用いた微分可能物理シミュレーションの関連研究(例:Brax)も関連する。具体的な論文タイトルは要旨からは不明である。

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

著者: Maximilian Stölzle, Solange Gribonval, Daniel Feliu-Talegon, Vito Daniele Perfetta, Michele Martini, Chuhan Zhang, Kiwan Wong, Mohammed Tarnini, Anup Teejo Mathew, Federico Renda, Daniela Rus, Cosimo Della Santina

分類: cs.RO, cs.AI

原文アブストラクト

Reduced-order models based on Cosserat-rod theory are now well established, and modeling theory is no longer the primary bottleneck in soft-robot control. Their implementations, however, do not support the differentiable, GPU-parallel, and control-oriented workflows that underpin advanced rigid-robotics applications. Here, we fill this gap with SoRoMoX (Soft Robot Models in JAX), a fully numerical, JIT-compilable Python/JAX framework. SoRoMoX implements articulated, Piecewise Constant Strain, and Variable Strain models through a unified, control-ready interface that provides inertia matrices, gravitational and elastic forces, Jacobians, and their derivatives. To our knowledge, it is the first rod/strain-based soft-robot modeling framework that runs directly on GPUs and is end-to-end differentiable with respect to states, inputs, and parameters. Sequential CPU rollouts are up to 18.1x faster than state-of-the-art alternatives, while GPU-parallel rollouts increase throughput by up to 234.6x. This performance enables workflows that were previously impractical or impossible: static-equilibrium system identification with 66% lower marker RMSE; residual-force learning with a further 64% reduction; computed-torque tracking with RMSE reduced by a factor of approximately 500 relative to model-free PD; control-gain optimization with up to 62% lower loss than untuned gains; safety-constrained control using high-order control barrier functions to keep the peak contact force within a prescribed 5 N bound, compared with 33.5 N without the safety constraint; and reinforcement-learning policy training up to 7x faster than a CPU PyElastica discrete-rod baseline through massively parallel rollouts.

関連論文

  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