Textual Planning with Explicit Latent Transitions
Not on the map. Why?
Abstract (arXiv)
Planning requires a transition model that predicts how each action changes the current state. When a large language model (LLM) plays this role, every next state is generated token by token, which makes searching over many possible futures slow and expensive. Existing alternatives either still query an LLM at every step or require a symbolic model of the domain. We propose EmbedPlan, a transition model built on frozen text embeddings: it embeds natural language descriptions of the state and the action with a frozen LLM, predicts the embedding of the next state with a lightweight learned network, and returns the closest real state. Because this network can be trained on top of any encoder, EmbedPlan also provides a controlled way to compare text representations for learning transitions. We evaluate it on 9 classical planning domains, under six settings that hold out progressively more of the data, from transitions to entire domains, and against baselines ranging from predicting no change to learning symbolic action rules. On planning problems seen during training, EmbedPlan almost always ranks the true next state among its top five guesses, still does so for most queries even when every observed state is a candidate, and retains 92-99% of its single-step accuracy when predicting several steps ahead from its own outputs. Given the same candidate states as GPT-5.4, it picks the true next state more often while taking about 0.17 ms per transition with cached embeddings. Accuracy is lower on unseen problems and near chance on unseen domains, and the controlled comparison traces this limit to the state representation rather than to the learned transition.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 41 of its 58 references that are in the map. First posted on arXiv 4 Feb 2026.
Builds on
41 of 58 references.
References · 41
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,276
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
- 2021SimCSE: Simple Contrastive Learning of Sentence Embeddingscited by 2,173
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
- 2018Relational inductive biases, deep learning, and graph networkscited by 1,360
- 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,353
29 more
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,264
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 1,039
- 2016HyperNetworkscited by 1,027
- 2023Reasoning with Language Model is Planning with World Modelcited by 861
- 2019Generalization through Memorization: Nearest Neighbor Language Modelscited by 800
- 2018Recurrent World Models Facilitate Policy Evolutioncited by 785
- 2020CURL: Contrastive Unsupervised Representations for Reinforcement Learningcited by 760
- 2022MTEB: Massive Text Embedding Benchmarkcited by 640
- 2023LLM+P: Empowering Large Language Models with Optimal Planning Proficiencycited by 501
- 2024DINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planningcited by 386
- 2022Transformers are Sample-Efficient World Modelscited by 328
- 2020Data-Efficient Reinforcement Learning with Self-Predictive Representationscited by 318
- 2020COGS: A Compositional Generalization Challenge Based on Semantic Interpretationcited by 263
- 2019Contrastive Learning of Structured World Modelscited by 209
- 2023Leveraging Pre-trained Large Language Models to Construct and Utilize World Models for Model-based Task Planningcited by 207
- 2023Learning to Model the World with Languagecited by 68
- 2017Classical Planning in Deep Latent Space: Bridging the Subsymbolic-Symbolic Boundarycited by 58
- 2024Can Language Models Serve as Text-Based World Simulators?cited by 52
- 2019Learning Domain-Independent Planning Heuristics with Hypergraph Networkscited by 50
- 2020Learning Dynamic Belief Graphs to Generalize on Text-Based Gamescited by 49
- 2024ACPBench: Reasoning about Action, Change, and Planningcited by 39
- 2021Learning General Optimal Policies with Graph Neural Networks: Expressive Power, Transparency, and Limitscited by 38
- 2025LLMs as Planning Formalizers: A Survey for Leveraging Large Language Models to Construct Automated Planning Modelscited by 38
- 2024Planetarium: A Rigorous Benchmark for Translating Text to Structured Planning Languagescited by 33
- 2017Action Schema Networks: Generalised Policies with Deep Learningcited by 32
- 2024Thought of Search: Planning with Language Models Through The Lens of Efficiencycited by 32
- 2022Large-Scale Retrieval for Reinforcement Learningcited by 24
- 2021Safe Learning of Lifted Action Modelscited by 16
- 2024SNAP: Semantic Stories for Next Activity Predictioncited by 8
Built on by
Not cited yet.