
CausalWAM: 월드 액션 모델에서 행동의 원인 학습
CausalWAM: Learning the Causes of Actions in World Action Models
Juntao Cheng, Jinzhou Tang, Fan Feng, Henrik I. Christensen, Yujia Zheng
월드 액션 모델(World Action Model, WAM)은 관측에서 잠재 변수를 학습하여 장면 변화를 예측하고 행동을 생성하는 모델이다. 기존 WAM은 시연 데이터에 피팅될 때 행동 관련 잠재 변수와 불필요한 변수(누이선스 변수)가 얽히기 쉬워, 행동이 실제 인과 변수만으로 생성된다는 보장이 없다. 이 논문은 희소성(sparsity) 제약 조건 하에서 행동 생성에 쓰이는 잠재 변수가 행동 관련 잠재 변수만의 가역 함수임을 수학적으로 증명하며, 이 결과는 잠재 변수 간 통계적 의존성이 있어도 성립한다. 이를 구현한 CWAM(Causal WAM)은 사전 학습된 WAM의 행동 헤드에 희소 라우팅 경로를 추가하며, 월드 모델링 목적 함수는 그대로 유지한다. 합성 데이터와 실제 로봇 실험에서 이론을 검증하고 폐루프(closed-loop) 성공률 향상을 확인했다.

















