한 줄 요약
InternW0-Δ는 20K시간 이상의 이질적 데이터로 학습한 세계 행동 모델로, 시각 역학과 행동 생성을 통합하여 로봇 제어 성능을 향상시킨다.
핵심 기여도
- **Causal Imprint** 모듈을 도입하여 미래 관측 없이 행동 전문가에게 예측적 표현을 제공.
- **4D-aware distillation**을 통해 Track4World 모델로부터 기하학적 및 운동학적 사전 지식을 주입.
- **이질적 데이터 코퍼스** (20K시간 이상)를 구축하고 공개.
- **Mixture-of-Transformers (MoT)** 구조로 시각 전문가와 행동 전문가를 결합하여 통합된 행동 생성 가능.
핵심 아이디어
InternW0-Δ는 대규모 사전 학습 모델의 다양한 사전 지식을 통합하여 로봇 행동 생성을 최적화하는 새로운 접근법을 제시한다. 기존 방법에서는 시각 역동성 예측과 행동 생성을 별도로 처리하거나, 미래 영상 생성을 필요로 하는 경우가 많았지만, InternW0-Δ는 Causal Imprint를 통해 미래 영상 없이도 예측적 표현을 학습한다. 이는 추론 시 미래 영상 롤아웃 없이 행동을 직접 생성할 수 있게 한다. 또한, Track4World 모델을 활용한 4D-aware distillation을 통해 기하학적 및 운동학적 사전 지식을 주입함으로써, 행동 전문가가 더 정확한 제어를 수행할 수 있도록 지원한다. MoT 아키텍처는 이러한 전문가 간의 정보 흐름을 효율적으로 관리하며, VLM을 통해 작업 조건에 맞는 의미 정보를 제공한다.
기술적 접근법
- **모델 아키텍처**: Mixture-of-Transformers (MoT)를 사용하여 시각 전문가와 행동 전문가를 결합.
- **Causal Imprint**: 미래 관측 없이 미래 관련 장면 변화를 학습하는 모듈. 훈련 시 미래 정보를 사용하지만 추론 시에는 미래 영상이 필요하지 않다.
- **4D-aware distillation**: Track4World 모델로부터 기하학적 및 운동학적 사전 지식을 주입.
- **데이터셋**: UMI, Ego2Robot, 로봇 시연 등 다양한 출처의 데이터를 통합하여 20K시간 이상의 이질적 데이터 코퍼스를 구축.
- **훈련 전략**: 두 단계 훈련 (1단계: 시각 역동성과 행동 생성 학습, 2단계: 특정 로봇 및 작업에 맞춘 후 훈련).
- **최적화 기법**: 레이어별 컴파일, 액티베이션 체크포인팅, 비디오 오토인코더 레이턴트 캐싱 등을 통해 훈련 및 추론 효율성 향상.
주요 결과
- **LIBERO-Plus**: 92.8% 성공률 (Clean2Random: 71.9%, Clean2Clean: 90.0%)
- **EBench**: 66.0점 (전체 점수 기준 최고)
- **RoboDojo**: 평균 성공률 23.9% (기존 최고 WAM 대비 약 2배 향상)
- **실물 로봇 배포**: 두 개의 그리퍼 기반 및 두 개의 다exterous-hand 플랫폼에서 동일한 사전 훈련 체크포인트를 사용한 후 훈련으로 성공적 배포.
의의 및 한계
InternW0-Δ는 대규모 이질적 데이터를 기반으로 한 통합된 행동 생성 모델로서, 다양한 로봇 플랫폼과 작업 환경에서 뛰어난 성능을 보인다. 특히, 미래 영상 없이도 행동을 생성할 수 있는 Causal Imprint는 실시간 제어에 유리하며, 4D-aware distillation은 기하학적 이해를 강화하여 복잡한 작업에서도 안정적인 성능을 제공한다. 그러나 모델의 복잡성은 훈련 및 배포 시 높은 계산 자원을 요구하며, 특정 작업에 대한 세부 조정이 필요할 수 있다. 또한, 데이터셋의 이질성은 모델의 일반화 능력을 향상시키지만, 일부 작업에서는 데이터 부족으로 인한 성능 저하가 발생할 수 있다.
실용적 활용
InternW0-Δ는 다양한 로봇 플랫폼 (예: 그리퍼, 다exterous-hand)에서의 일반적 제어, 복잡한 작업 수행 (예: 이동형 이중 조작), 분산 환경에서의 실시간 제어 등에 적용 가능하다. 특히, 미래 영상 생성 없이 행동을 예측할 수 있는 구조는 실시간 성능 향상에 유리하며, 대규모 데이터셋과 공개된 인프라를 통해 연구 및 산업 현장에서의 활용이 확대될 수 있다.