GPU 가속 로봇 학습을 위한 PyTorch의 배치형 미분 가능 강체 역학
Batched Differentiable Rigid Body Dynamics in PyTorch for GPU-Accelerated Robot Learning
TL;DR AI
1분핵심 요약
연구진은 배치 GPU 평가와 자동 미분에 맞춘 자가 포함형 PyTorch 기반 강체 동역학 엔진 BARD를 소개했다.
BARD는 캐싱, 사전 계산된 관절 변환, 레벨 병렬 전파를 활용해 수치 정확도를 유지하면서 처리량을 크게 높였다.
여러 로봇 모델 벤치마크에서 Pinocchio보다 큰 속도 향상을 보였으며, 특히 NVIDIA H200에서 두드러졌다.
또한 기울기 기반 시스템 식별과 Isaac Lab 학습 파이프라인에 적용돼 로봇 학습의 핵심 병목을 줄이는 가능성을 보여줬다.