STMA: A Spatio-Temporal Memory Agent for Long-Horizon Embodied Task Planning

Mingcong Lei, Yiming Zhao, Ge Wang, Zhixin Mai, Shuguang Cui +2 more
2/14/2025
cs.AI

No confident code implementation found for this paper. 2 possible matches can be shown below.

Abstract

A key objective of embodied intelligence is enabling agents to perform long-horizon tasks in dynamic environments while maintaining robust decision-making and adaptability. To achieve this goal, we propose the Spatio-Temporal Memory Agent (STMA), a novel framework designed to enhance task planning and execution by integrating spatio-temporal memory. STMA is built upon three critical components: (1) a spatio-temporal memory module that captures historical and environmental changes in real time, (2) a dynamic knowledge graph that facilitates adaptive spatial reasoning, and (3) a planner-critic mechanism that iteratively refines task strategies. We evaluate STMA in the TextWorld environment on 32 tasks, involving multi-step planning and exploration under varying levels of complexity. Experimental results demonstrate that STMA achieves a 31.25% improvement in success rate and a 24.7% increase in average score compared to the state-of-the-art model. The results highlight the effectiveness of spatio-temporal memory in advancing the memory capabilities of embodied agents.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 2 weaker matches are hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{lei2025stma,
  title  = {STMA: A Spatio-Temporal Memory Agent for Long-Horizon Embodied Task Planning},
  author = {Mingcong Lei and Yiming Zhao and Ge Wang and Zhixin Mai and Shuguang Cui and Yatong Han and Jinke Ren},
  year   = {2025},
  eprint = {2502.10177},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2502.10177v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2502.10177.svg)](https://codeofpaper.com/paper/2502.10177)

Discussion