Atari Pong의 World Model을 위한 개념 기반 공간 정규화 (Concept-Guided Spatial Regularization)

요약

본 논문은 Atari Pong의 5가지 시각적 World Model 에이전트(DreamerV3 등)를 분석하고, 이들의 성능 저하 원인을 진단했습니다. 기존 모델들은 공의 사라짐이나 잘못된 움직임 같은 명확한 실패 사례를 보였습니다. 이에 따라 작업에 중요한 개념을 모델링하는 '개념 기반 공간 정규화(CGSReg)' 기법을 제안하여 World Model의 안정성을 개선함을 입증했습니다.

핵심 포인트

• World model은 MBRL 시스템의 핵심 구성 요소로 평가됨.
• 기존 5가지 모델 모두에서 공의 사라짐 등 명확한 실패 사례가 발견됨.
• CGSReg는 DreamerV3, DIAMOND, TWISTER의 성능을 개선함.
• 개념(task-critical concepts) 모델링 부족이 World Model 실패의 주요 원인으로 추정됨.