Preprint
Computer Vision

UniMPA: A Unified Memory-Prediction-Action Model via Action-Grounded Transition Modeling

Wei Li, Rui Shao, Jie He, Lingsen Zhang, Ziwei Liu, Liqiang Nie
September 10, 2026

0

Citations

0

Influential Citations

—

Venue

2026

Year

Abstract

Recent advances in Vision-Language-Action (VLA) models have improved robotic manipulation, yet observation-to-action learning remains limited by a fundamental transition realizability gap, manifested in three tightly coupled problems: (i) Transition ambiguity. Visually similar current observations may correspond to different manipulation phases and imply different subsequent transitions. (ii) Prediction--execution mismatch. A visually plausible predicted future observation does not necessarily correspond to a physically realizable transition. (iii) Experience--realization mismatch. A historically executable action pattern may not necessarily realize the intended transition in the current scene and therefore requires context-aware adaptation. Accordingly, we propose UniMPA, a Unified Memory-Prediction-Action model that addresses these problems through a shared action-grounded transition interface. (i) UniMPA introduces Persistent-Selective Future Prediction to resolve transition ambiguity by modeling the intended future state evolution. A persistent latent stream continuously tracks task-level progress, while a transition-critical pixel stream selectively resolves fine-grained interaction changes through memory-grounded prediction. (ii) To assess the physical executability of the anticipated transition, the predicted transition queries a temporal Visual-Action Memory Bank. The bank retrieves historically realized visual-action experience, grounding future prediction in executable evidence. (iii) To adapt executable experience to the current scene, an Action-Visual Memory Bank retrieves visually grounded action prototypes from historical action evolution. Prototype-Biased Flow then shifts the flow source toward a historically supported action manifold for context-aware refinement.

Analysis

Why This Paper Matters

Vision-Language-Action (VLA) models have become a central paradigm for robotic manipulation, yet they remain constrained by a fundamental disconnect between what a model observes and what it can actually execute. UniMPA names this the "transition realizability gap" and argues it manifests in three coupled failures: visually similar observations can imply different manipulation phases (transition ambiguity), a plausible predicted future image need not be physically achievable (prediction-execution mismatch), and a historically successful action may not transfer to the current scene (experience-realization mismatch). Framing these as one unified problem rather than three separate ones is the paper's most valuable conceptual move.

The work matters because it targets a persistent weakness in embodied AI: models that generate convincing rollouts but fail on real hardware. By grounding prediction and action in memory of what has actually been realized, UniMPA points toward systems whose internal forecasts are constrained by executable evidence rather than visual plausibility alone.

Technical Contributions

  • Persistent-Selective Future Prediction: a dual-stream design where a persistent latent stream tracks task-level progress while a transition-critical pixel stream selectively resolves fine-grained interaction changes through memory-grounded prediction, addressing transition ambiguity.
  • Temporal Visual-Action Memory Bank: the predicted transition queries this bank to retrieve historically realized visual-action experience, grounding future prediction in executable evidence and assessing physical executability.
  • Action-Visual Memory Bank and Prototype-Biased Flow: retrieves visually grounded action prototypes from historical action evolution and shifts the flow source toward a historically supported action manifold for context-aware refinement.
  • Unified action-grounded transition interface: a shared representation that ties memory, prediction, and action together rather than treating them as separate modules.

Results

The abstract reports no quantitative metrics, benchmarks, or comparisons. It is a methods-focused contribution describing the architecture and its motivation. Any empirical claims would need to be verified in the full paper.

Significance

If validated, UniMPA's memory-grounded transition modeling could influence how VLA systems are designed, shifting emphasis from purely generative future prediction toward executability-aware forecasting. The dual memory-bank formulation and Prototype-Biased Flow are reusable ideas for other sequential decision-making domains where historical experience must be adapted to novel contexts. The paper's taxonomy of the transition realizability gap also provides a useful diagnostic framework for future work on observation-to-action learning.