Vision-language-action models must enable agents to execute long-horizon tasks under partial observability. However, most existing approaches remain observation-driven, relying on short context windows or repeated queries to vision-language models (VLMs). This leads to loss of task progress, action repetition under perceptual aliasing, and high inference latency. While semantic grounding is important, long-horizon manipulation fundamentally requires persistent, action-conditioned state representations. Current VLAs lack such representations and exhibit limited temporal and physical reasoning, making them ill-suited for multi-stage control. This paper introduces RB-VLA, a belief-centric architecture trained with self-supervised world-model objectives that maintains a compact latent state encoding task-relevant history, dynamics, and object interactions. Queried once per task, the VLM provides high-level intent, while the belief tracks task progress and enables phase-aware, causally grounded control under partial observability without storing raw observations or scaling memory with time. The belief and intent jointly condition a diffusion policy for robust closed-loop execution. RB-VLA outperforms prior VLAs on long-horizon benchmarks, achieving 52.5 percent and 37.5 percent higher success rates on multi-stage pick-and-place and stacking tasks, respectively, compared to pi_0. It also reduces inference latency by up to five times relative to baselines and eliminates memory growth across timesteps observed in existing VLAs. Ablations show the belief module is the primary driver of performance, increasing success rates from 32.5 percent without belief to 77.5 percent with belief.
Figures & tables
Fig. 1 : Overall Architecture of RB-VLA . a) A VLM generates high-level intent from the current observation and instruction at t=0, implicitly grounding task semantics and relevant scene entities. b) Recursive belief estimator maintains a compact belief encoding task-relevant history. c) A DiT-based action head generates action chunks for closed-loop control
Fig. 2 : Environment with similar objects
Method
Single P&P
Multi P&P
Single Stack
Multi Stack
GROOT-N1
52.5
20.0
65.0
32.5
OpenVLA
22.5
×
×
×
π0
55.0
25.0
67.5
42.5
RB-VLA (ours)
82.5
77.5
85.0
80.0
TABLE I : Long-horizon success rates (%)
Fig. 3 : Long-horizon tasks under partial observability . After the first object is placed and becomes occluded, the model correctly selects the subsequent target despite visually similar objects, avoiding re-grasping the same object.
Fig. 4 : Average inference latency per episode for RB-VLA and baseline VLA methods.
Temporal Context
Qwen2-VL-7B
GR00T-N1
RB-VLA (Ours)
1 frame
1.00×
1.00×
1.0×
2 frames
1.81×
2.21×
1.0×
4 frames
3.64×
3.85×
1.0×
8 frames
6.82×
7.12×
1.0×
TABLE II : Relative memory usage as a function of temporal context length, comparing RB-VLA with baseline models.
Fig. 5 : (a) Future ℓ2 action divergence versus belief cosine similarity. (b) Future observation cosine similarity versus belief cosine similarity. Both plots exhibit strong monotonic correlation, indicating that proximity in belief space predicts similarity in future perception and control behavior.
Fig. 6 : (i) Future divergence using cosine distance with multiple sampled latent variables ( n=16 ) from a fixed belief state bt , illustrating stochastic rollout diversity. (ii) KL divergence loss across training, showing stable convergence of the belief model’s latent distribution.
Fig. 7 : Normalized self attention weights of the temporal transformer over the previous recursive belief token and short history of input frame encoder tokens, indicating which tokens contribute to the belief state at time t
Frame Encoder
Stoch. z
Belief → Policy
Success
×
×
×
32.5
×
×
✓
57.5
×
✓
✓
62.5
✓
✓
✓
77.5
TABLE III : Component Ablation on Two-Object Pick-and-Place task. Average success rate over 40 episodes under partial observability.