日本フィジカルAI新聞

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

週刊ニュースレター購読
両腕操作/エッジAI/量子化arXiv:2608.03938v1

8GB予算での両腕操作:エントリーレベルJetson上のゼロコピーセンシングと量子化ACT

Bimanual Manipulation Within an 8 GB Budget: Zero-Copy Sensing and Quantized ACT on an Entry-Level Jetson

シェア:XThreadsFacebookLINEはてブBluesky

エントリーレベルのJetson Orin Nano Super (8GB)上で、両腕操作ポリシーを実行するためのコストを評価し、ゼロコピーセンシングとTensorRT量子化により推論遅延を大幅に削減しつつタスク成功率を維持できることを示した。

詳しい要約

1. どんなもの?

本論文は、エントリーレベルの組み込みGPUであるNVIDIA Jetson Orin Nano Super (8 GB)上で、模倣学習によるバイマニュアル操作ポリシーを実行するシステムを提案している。デスクトップGPU (RTX 3070)はオフライン学習のみに使用し、変形可能なビーンバッグのピックアンドプレースを評価する。システムは、NVMMバッファを用いたGStreamerキャプチャパイプラインによるゼロコピーセンシングと、量子化されたACT (Action Chunking with Transformers)をTensorRTで実行する。

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

従来のバイマニュアル操作ポリシーはワークステーションやデータセンタークラスのGPUで評価されることが多く、組み込みハードウェアでの展開コストは未解明だった。本論文は、エントリーレベルのJetson上で完全に動作するシステムを実証し、ゼロコピーセンシングの効果(CPUヘッドルームとレイテンシの改善)や、ACTの量子化に関する新たな知見(INT8キャリブレーションがTransformer層を量子化しないことなど)を提供する点が新しい。

3. 技術・手法の肝は?

手法の肝は、(1) GStreamerとNVMMバッファによるゼロコピーキャプチャパイプラインで、ホスト-デバイス間のコピーを削減し、CPU負荷とレイテンシを低減する点、(2) ACTとDiffusion Policyを同一のデモンストレーションで学習し、それぞれの参照予算(ACT: 100kステップ、Diffusion Policy: 200kステップ)で比較する点、(3) ACTをTensorRTに変換し、FP16およびINT8精度で量子化する点。特に、INT8キャリブレーションがResNet18バックボーンのみを量子化し、Transformer層はゼロ個受け入れないことを発見し、その理由を分析している。

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

有効性は、ビーンバッグのピックアンドプレースタスクで検証した。ゼロコピーセンシングは、従来パイプラインと比較して、ピークシングルコア使用率が98.0%から77.0%に低下し、ワーストケースレイテンシが117.31 msから101.52 msに改善した。ACTは19/20試行成功し、Diffusion Policyは0/10で収束しなかった。TensorRT変換により、FP16で平均推論レイテンシが114.02 msから17.93 ms(6.4倍)、INT8で12.65 ms(9.0倍)に短縮され、タスク成功率は全精度で維持された(19/20, 18/20, 19/20)。

5. 議論はある?

議論として、Diffusion Policyが収束しなかった理由を、精度の限界ではなく収束コストの違いに帰属している。また、TensorRTのINT8キャリブレーションがTransformer層を量子化しないため、INT8のサイズ削減がFP16に対してわずか0.9%である一方、レイテンシはさらに28%改善するという発見を報告している。さらに、量子化の必要性はACTのアクションチャンキング構成に依存し、n_action_steps=100では全精度で実行可能だが、時間的アンサンブリングに必要なステップごとの再予測では不可能であると述べている。

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

要旨で参照されている研究は、ACT (Action Chunking with Transformers) と Diffusion Policy である。次に読むべき論文としては、これらの元論文(ACTの論文、Diffusion Policyの論文)が挙げられる。また、関連する模倣学習や組み込み展開の研究も参考になる。

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

著者: Ekansh Singh, Eva Samuel, Alessandra Reneau, Ryan Schmeelk, Yashvi Gandhi

分類: cs.RO, eess.SY

原文アブストラクト

Bimanual manipulation policies trained with imitation learning are typically evaluated on workstation or datacenter-class GPUs, leaving the cost of deploying them on embedded hardware largely uncharacterized. We present a bimanual SO-101 system running entirely on an NVIDIA Jetson Orin Nano Super (8 GB), the entry-level tier of NVIDIA's embedded line, using a desktop GPU (RTX 3070) only for offline training, evaluated on pick-and-place of a deformable beanbag. First, we build a GStreamer capture pipeline backed by NVMM buffers that removes redundant host-device copies from three-camera sensing. Contrary to expectation, the conventional path fit the memory budget and dropped no frames; what zero-copy sensing recovers is CPU headroom (peak single-core utilization 98.0% to 77.0%) and worst-case latency (117.31 ms to 101.52 ms). Second, we train ACT and Diffusion Policy on identical demonstrations, each at its own reference budget (100k gradient steps for ACT, 200k for Diffusion Policy). ACT converges to a task-competent policy (19/20 trials) while Diffusion Policy does not converge to a usable one (0/10) even at twice the step count, which we attribute to differing convergence costs rather than an accuracy ceiling. Third, we convert ACT to TensorRT. FP16 reduces mean inference latency from 114.02 ms to 17.93 ms (6.4x) and INT8 to 12.65 ms (9.0x), with task success preserved at all three precisions (19/20, 18/20, 19/20). We report two findings not previously documented for ACT: TensorRT's general-purpose INT8 calibration quantizes the ResNet18 backbone but accepts zero of 145 transformer layers, explaining INT8's negligible size reduction over FP16 (0.9%) despite a further 28% latency gain; and the need for quantization is conditional on ACT's action-chunking configuration, feasible in full precision at n_action_steps = 100 but not at the per-step re-prediction temporal ensembling requires.