IMLE Policy: Fast and Sample Efficient Visuomotor Policy Learning via Implicit Maximum Likelihood Estimation
Created by
Haebom
저자
Krishan Rana, Robert Lee, David Pershouse, Niko Suenderhauf
개요
본 논문은 생성 모델링 기법(예: 확산 모델)을 사용하는 모방 학습의 최근 발전에도 불구하고, 로봇 제어 분야에서 데이터 수집 비용이 높고 컴퓨팅 자원이 제한적인 문제점을 해결하기 위해, Implicit Maximum Likelihood Estimation (IMLE) 기반의 새로운 행동 복제 접근법인 IMLE Policy를 제안합니다. IMLE Policy는 적은 데이터로도 효과적으로 학습하며, 복잡한 다중 모드 행동 학습에서 기존 방법들에 비해 평균 38% 적은 데이터로 동일한 성능을 달성합니다. 간단한 생성자 기반 아키텍처 덕분에 단일 단계 행동 생성이 가능하여, 확산 정책에 비해 추론 속도가 97.3% 향상되며, 단일 단계 흐름 일치 방법보다 성능이 우수합니다. 시뮬레이션 및 실제 환경에서 다양한 조작 작업을 통해 데이터 제약 조건 하에서 복잡한 행동을 학습할 수 있는 능력을 검증했습니다. 프로젝트 페이지(https://imle-policy.github.io/)에서 비디오와 코드를 제공합니다.