TraceFlow: Guiding Frozen Flow-Matching Robot Policies with Success and Failure Traces
On the map: Diffusion models → Diffusion and flow policies for robotics
Abstract (arXiv)
A vision-language-action (VLA) policy with a flow-matching action expert generates each action chunk (a short command sequence) by integrating a learned velocity field; once its weights are fixed, the success or failure of an earlier rollout cannot change the chunk generated now. Concurrent test-time methods give a frozen policy such an input from retrieved successes, a learned critic, a verifier, or a dynamics model, but none uses the robot's own failed rollouts as negative evidence with nothing but a terminal outcome bit. We introduce TraceFlow, a progress-aligned guidance field that turns the action densities of retrieved successful and failed rollouts into a bounded correction to a frozen flow-matching action expert, using one terminal outcome bit per rollout and no other label. Its TraceBank stores traces, time-ordered state-action records with a terminal label, starts from the target-task training traces, and later admits the deployed robot's own rollouts. On an ordered real-robot packing task the base completes 21 of 50 trials in order, TraceFlow 39, and one stacking round without any weight update 47, with wrong-sequence episodes falling from 20 to 0. In simulation the gain is selective: with per-suite selected settings, TraceFlow raises RoboMemArena Sequence from 78.92% to 91.50% task success and Transferring from 54.41% to 62.00% at stacking round 2, leaves the 26-task aggregate unchanged, lowers Counting and Occlusion by 1.12 and 1.42 points, and changes LIBERO-Plus (Long) by +1.27 points (p = 0.0733). Stacking gains are finite, every branch peaking before round ten, and the bank's success-to-failure ratio predicts no retrieval allocation. Project page: https://zhangjiaxuan-xuan.github.io/TraceFlow/
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 51 of its 57 references that are in the map. First posted on arXiv 17 Sep 2026.
Builds on
51 of 57 references.
Closest ancestors · 6
Same direction, newest first
- 2026Guided Action Flow: Q-Guided Inference for Flow-Matching Vision-Language-Action Policies2 Jul
- 2025Steering Vision-Language-Action Models as Anti-Exploration: A Test-Time Scaling Approach2 Dec
- 2025DynaGuide: Steering Diffusion Polices with Active Dynamic Guidance16 Jun
- 2025CCDP: Composition of Conditional Diffusion Policies with Guided Sampling19 Mar
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusion7 Mar
- 2022Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learning12 Aug
Other references · 45
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2022Classifier-Free Diffusion Guidancecited by 5,717
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,718
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
33 more
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2023RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Controlcited by 2,993
- 2020Supervised Contrastive Learningcited by 2,803
- 2024OpenVLA: An Open-Source Vision-Language-Action Modelcited by 2,777
- 2024π0: A Vision-Language-Action Flow Model for General Robot Controlcited by 2,297
- 2017Gradient Episodic Memory for Continual Learningcited by 1,871
- 2025π0.5: a Vision-Language-Action Model with Open-World Generalizationcited by 1,653
- 2024Octo: An Open-Source Generalist Robot Policycited by 1,427
- 2023LIBERO: Benchmarking Knowledge Transfer for Lifelong Robot Learningcited by 1,338
- 2018Experience Replay for Continual Learningcited by 1,074
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
- 2016HyperNetworkscited by 1,003
- 2020Dark Experience for General Continual Learning: a Strong, Simple Baselinecited by 895
- 2022Test-Time Prompt Tuning for Zero-Shot Generalization in Vision-Language Modelscited by 379
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 329
- 2025LIBERO-Plus: In-depth Robustness Analysis of Vision-Language-Action Modelscited by 278
- 2023REFLECT: Summarizing Robot Experiences for Failure Explanation and Correctioncited by 171
- 2017Neural Episodic Controlcited by 148
- 2021Continual World: A Robotic Benchmark For Continual Reinforcement Learningcited by 112
- 2016Model-Free Episodic Controlcited by 100
- 2025RoboMonkey: Scaling Test-Time Sampling and Verification for Vision-Language-Action Modelscited by 56
- 2025On the Guidance of Flow Matchingcited by 44
- 2024DynaMem: Online Dynamic Spatio-Semantic Memory for Open World Mobile Manipulationcited by 32
- 2025MAP-VLA: Memory-Augmented Prompting for Vision-Language-Action Model in Robotic Manipulationcited by 28
- 2026RoboMemArena: A Comprehensive and Challenging Robotic Memory Benchmarkcited by 21
- 2026Global Prior Meets Local Consistency: Dual-Memory Augmented Vision-Language-Action Model for Efficient Robotic Manipulationcited by 19
- 2026RoboTTT: Context Scaling for Robot Policiescited by 13
- 2024Learning Manipulation Skills through Robot Chain-of-Thought with Sparse Failure Guidancecited by 9
- 2026ProcVLM: Learning Procedure-Grounded Progress Rewards for Robotic Manipulationcited by 8
- 2026Memory Retrieval in Visuomotor Policies for Long-Horizon Robot Controlcited by 7
- 2026Retrieve-then-Steer: Online Success Memory for Test-Time Adaptation of Generative VLAscited by 5
- 2026Test-Time Gradient Guidance of Flow Policies in Reinforcement Learningcited by 5
- 2026Retrieve, Don't Retrain: Extending Vision Language Action Models to New Tasks at Test Timecited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026RedFlow: Redirect Failure into Action-Level Corrections for Flow-matching VLA Policy
- 2026CommitFlow: Semantic Commitment Verification and Local Correction for Long-Horizon Robot Manipulation VLA Execution
- 2026Fewer Steps, Better Actions: Rethinking Flow-Matching Inference for VLA Policies
- 2026CARF: Contrastive Attraction-Repulsion of Failure-Guided Flow Matching
- 2026Looking Back to Move Forward: Temporal Verification for Generative Robot Policies
- 2026PreferenceFlow: Test-Time Guidance of Flow-Matching Robot Policies from Human Interventions
- 2026SEAM: Smooth Execution of Action-Chunked Motion for Vision-Language-Action Policies
- 2026RecastVLA: From Past Interaction to Future Control with Adaptive Policy States ↗