
Riemannian MeanFlow: 액션 매니폴드에서의 빠른 시각운동 정책 학습
Faster Visuomotor Policy Learning on Action Manifolds via Riemannian MeanFlow
S. Talha Bukhari, Austin Garrett, Yi Wei, Ruiqi Ni, Zachary Kingston 외 1명
확산(diffusion) 및 플로우 매칭(flow matching) 기반 시각운동 정책(visuomotor policy)은 행동 시퀀스의 다중 모드 분포를 표현할 수 있으나, 행동 생성 시 벡터 필드의 다단계 수치 적분이 필요해 실시간 제어에 병목이 된다. 로봇 행동 시퀀스는 매끄럽고 미분 가능한 다양체(manifold) 위에 정의되므로, 학습된 정책이 행동 공간의 내재적 기하 구조를 반영해야 한다는 과제도 있다. 이 논문은 리만 조건부 플로우 매칭(Riemannian Conditional Flow Matching) 앵커에 기반한 플로우 맵 일관성 목적함수를 통해 RMFP를 제안한다. RMFP는 단 한 번의 신경망 함수 평가(NFE)만으로 다양체 위에서 행동 시퀀스를 생성할 수 있다. 구형 LASA, Push-T, Robomimic의 Tool Hang·Transport 과제, Franka Kitchen 과제 및 실제 로봇 조작 실험에서 기존 방법과 경쟁적인 성능을 낮은 샘플링 비용으로 달성하였다.









