SeedPolicy: Reversing the Horizon Scaling Paradox in Robotic Diffusion Policies

SeedPolicy: Horizon Scaling via Self-Evolving Diffusion Policy for Robot Manipulation

Summary
Problem
Method
Results
Takeaways
Abstract

SeedPolicy is a novel imitation learning framework that introduces the Self-Evolving Gated Attention (SEGA) module to extend the observation horizon of Diffusion Policies. By maintaining a recursively updated latent state and utilizing a cross-attention gating mechanism, it achieves state-of-the-art performance on the RoboTwin 2.0 benchmark, effectively overcoming the performance degradation typically seen in long-horizon robot manipulation.

Executive Summary

TL;DR: SeedPolicy introduces a breakthrough temporal modeling module called Self-Evolving Gated Attention (SEGA). It transforms the standard Diffusion Policy (DP) from a fixed-window observer into a recurrent, context-aware agent. By utilizing a "Self-Evolving Gate" derived from attention maps, the model filters out visual noise and maintains a "memory" of task progress, allowing it to excel in long-horizon tasks where traditional DPs fail.

Background Positioning: This work targets a known but unsolved "glitch" in the Diffusion Policy literature: the fact that adding more history can sometimes make the robot perform worse. SeedPolicy is a structural SOTA upgrade that solves this bottleneck without the massive compute overhead of long-window Transformers.

Problem: The Horizon Scaling Paradox

In robotic manipulation, more information should lead to better decisions. However, standard Diffusion Policies treat history as a simple stack of frames. As the number of frames grows:

  1. Quadratic Complexity: Attention costs explode.
  2. State Aliasing: Different stages of a task (e.g., before vs. after picking a block) look visually identical, causing the robot to "freeze" or "loop."
  3. Temporal Noise: Background shifts and small movements dilute the important semantic cues needed for the task.

Methodology: Self-Evolving Gated Attention (SEGA)

The core innovation is SEGA, which replaces frame-stacking with a recurrent-style latent state update. Instead of looking at frames, the model maintains a fixed-size Latent State that evolves over time.

1. Dual-Stream Interaction

As shown in the architecture, SEGA operates two parallel paths:

  • State Update (Upper Stream): Takes the current observation, extracts new info via Multi-Head Self-Attention (MSA), and merges it into the historical state.
  • State Retrieval (Lower Stream): Uses the "memory" in the state to enhance the current image features, effectively providing the robot with "contextual eyeglasses."

Overall Architecture

2. The Self-Evolving Gate (SEG)

How does the model know which frames are important? The authors use the Cross-Attention maps themselves as a gating signal. If the attention score is low, the gate closes, preventing noisy or irrelevant background shifts from "polluting" the robot's memory.

SEGA Detail and Gating

Experiments: Breaking the Bottleneck

Tested on the RoboTwin 2.0 benchmark (50 tasks), SeedPolicy showed its true strength as tasks became longer.

  • Short Tasks: Gains are modest because the fixed window is "enough."
  • Long Tasks: SeedPolicy widens the gap significantly (+21.9%), while the baseline DP collapses because it "forgets" where it is in the multi-step process.

Success Rate vs Task Length

Real-World Robustness

In real-world tests (Dexmal DOS-W1 robot), SeedPolicy fixed the "execution stagnation" issue. In a "Bottle Handover" task, standard DP often froze because the visual state mid-task looked too much like the start state. SeedPolicy used its evolving latent state to realize the task was already in progress, achieving a 56% success rate vs 15% for the baseline.

Critical Analysis & Conclusion

Takeaway: SeedPolicy proves that temporal modeling isn't just about "stacking more frames." It's about efficiently compressing that history into a meaningful state while dynamically ignoring noise.

Limitations: While SeedPolicy is efficient, it still struggles in "Hard" randomized settings compared to massive Foundation Models (VLAs) because it lacks the internet-scale visual prior. However, replacing the temporal backbone of those VLAs with SEGA could be a very promising future direction.

Future Outlook: We expect this "gated recurrent attention" approach to become a standard design pattern for lightweight but "smart" edge robotics where 1.2B parameter models are too heavy to run in real-time.

Find Similar Papers

Try Our Examples

  • Search for recent papers that utilize gating mechanisms within Transformer-based imitation learning to handle temporal noise in robotic manipulation.
  • Which study first identified the performance degradation of Diffusion Policy as the observation horizon increases, and how did they attempt to mitigate it?
  • Explore how Self-Evolving Gated Attention could be integrated into Vision-Language-Action (VLA) models to improve their long-context reasoning capabilities.
Contents
SeedPolicy: Reversing the Horizon Scaling Paradox in Robotic Diffusion Policies
1. Executive Summary
2. Problem: The Horizon Scaling Paradox
3. Methodology: Self-Evolving Gated Attention (SEGA)
3.1. 1. Dual-Stream Interaction
3.2. 2. The Self-Evolving Gate (SEG)
4. Experiments: Breaking the Bottleneck
4.1. Real-World Robustness
5. Critical Analysis & Conclusion