월드 상태 생성기(WSG)는 실패 후 상태를 재작성하여 언어 에이전트의 계획이 실행 환경의 규칙과 일치하도록 유지하는 모델입니다. 이는 규칙을 강제하고 목표 도달 가능성을 검증하는 프로그램이 있는 7개의 합성 도메인에서 추출된 226K 트래젝토리에서 훈련되었습니다.

  • WSG는 실패를 세계가 정한 규칙으로 간주하고 이를 준수하기 위해 나머지 상태를 재작성합니다.
  • 이 모델은 계획을 단순 언어 지시사항이 아닌 검증 가능한 세계 상태로 작성합니다.
  • 7개의 공개 벤치마크 전반에 걸쳐 WSG는 약 30B 파라미터의 두 오픈 모델의 프롬프팅 대비 엔드투엔드 성공률을 높였습니다.
  • 이 성능은 오픈 모델을 프로프라이터리 모델 수준으로 끌어올립니다.

WSG는 프롬프트된 플래너가 거부된 단계에서 앞으로 나아가지 못하고 실패하는 문제를 해결하며, 진행 중인 실행에 따라 계획이 세계의 물리적 한계와 논리적 종속성에 맞춰지도록 보장합니다.