Le Générateur d'État du Monde (WSG) est un modèle qui maintient les plans des agents linguistiques alignés sur les règles de leur environnement d'exécution en réécrivant les états après des échecs. Il est entraîné sur 226K trajectoires extraites de sept domaines synthétiques où un programme applique des règles et vérifie l'atteignabilité de l'objectif.

  • Le WSG prend un échec comme une règle énoncée par le monde et réécrit les états restants pour s'y conformer.
  • Le modèle écrit les plans comme des états vérifiables du monde plutôt que de simples instructions linguistiques.
  • Sur sept benchmarks publics, le WSG augmente le succès de bout en bout pour deux modèles ouverts proches de 30B paramètres par rapport au prompting.
  • Cette performance porte les modèles ouverts au niveau des modèles propriétaires.

Le WSG résout le problème où les planificateurs avec prompts échouent en réécrivant les étapes refusées sans avancer, garantissant que les plans s'adaptent aux limites physiques du monde et aux dépendances logiques au fur et à mesure de l'exécution.