{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/arxiv-2602-08245","title":"STEP: Warm-Started Visuomotor Policies with Spatiotemporal Consistency Prediction","arxiv_id":"2602.08245","date":"2026-02-09","proceeding":"ICML","authors":["Jinhao Li","Yuxuan Cong","Yingqiao Wang","Hao Xia","Shan Huang","Yijia Zhang","Ningyi Xu","Guohao Dai"],"abstract":"Diffusion policies have recently emerged as a powerful paradigm for visuomotor control in robotic manipulation due to their ability to model the distribution of action sequences and capture multimodality. However, iterative denoising leads to substantial inference latency, limiting control frequency in real-time closed-loop systems. Existing acceleration methods either reduce sampling steps, bypass diffusion through direct prediction, or reuse past actions, but often struggle to jointly preserve action quality and achieve consistently low latency. In this work, we propose STEP, a lightweight spatiotemporal consistency prediction mechanism to construct high-quality warm-start actions that are both distributionally close to the target action and temporally consistent, without compromising the generative capability of the original diffusion policy. Then, we propose a velocity-aware perturbation injection mechanism that adaptively modulates actuation excitation based on temporal action variation to prevent execution stall especially for real-world tasks. We further provide a theoretical analysis showing that the proposed prediction induces a locally contractive mapping, ensuring convergence of action errors during diffusion refinement. We conduct extensive evaluations on nine simulated benchmarks and two real-world tasks. Notably, STEP with 2 steps can achieve an average 21.6% and 27.5% higher success rate than BRIDGER and DDIM on the RoboMimic benchmark and real-world tasks, respectively. These results demonstrate that STEP consistently advances the Pareto frontier of inference latency and success rate over existing methods.The code is publicly available at https://github.com/Kimho666/STEP.","url_abs":"https://arxiv.org/abs/2602.08245","url_pdf":"https://arxiv.org/pdf/2602.08245","source":{"archive":null,"snapshot":"2025-07-28","note":"not in the Papers with Code archive (frozen at the snapshot)","row_kind":"graph","title_abstract_authors_date":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)"},"code_links":[],"tasks":[],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2602.08245","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2602.08245"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"mentioned_in_github":null,"is_official":null,"provenance":"deterministic:regex_extraction","mentioned_in_paper":null,"url":"https://github.com/Kimho666/STEP","reach":null}],"summary":{"unverified":8},"by_repo_kind":{"found_in_text":{"samples":8,"ran":0,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":0,"samples":[{"code_sha256_prefix":"3351ad91683faf7f","entry":"compute_metrics","repo":"Kimho666/STEP","repo_kind":"found_in_text","path":"multirun_metrics.py","file_url":"https://github.com/Kimho666/STEP/blob/HEAD/multirun_metrics.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"3351ad91683faf7f"}},{"code_sha256_prefix":"0cc739967c061073","entry":"compute_metrics_agg","repo":"Kimho666/STEP","repo_kind":"found_in_text","path":"multirun_metrics.py","file_url":"https://github.com/Kimho666/STEP/blob/HEAD/multirun_metrics.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"0cc739967c061073"}},{"code_sha256_prefix":"d24a3503e232501f","entry":"create_obs_key_mapper","repo":"Kimho666/STEP","repo_kind":"found_in_text","path":"eval_combined_inference.py","file_url":"https://github.com/Kimho666/STEP/blob/HEAD/eval_combined_inference.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"d24a3503e232501f"}},{"code_sha256_prefix":"01ac1bfcde3e63b3","entry":"detect_policy_type","repo":"Kimho666/STEP","repo_kind":"found_in_text","path":"eval_cfm.py","file_url":"https://github.com/Kimho666/STEP/blob/HEAD/eval_cfm.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"01ac1bfcde3e63b3"}},{"code_sha256_prefix":"338cabf4cb17d013","entry":"get_model_obs_keys","repo":"Kimho666/STEP","repo_kind":"found_in_text","path":"eval_combined_inference.py","file_url":"https://github.com/Kimho666/STEP/blob/HEAD/eval_combined_inference.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"338cabf4cb17d013"}},{"code_sha256_prefix":"9061a9fc934ec01d","entry":"patch_cfm_policy_solver","repo":"Kimho666/STEP","repo_kind":"found_in_text","path":"eval_cfm.py","file_url":"https://github.com/Kimho666/STEP/blob/HEAD/eval_cfm.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"9061a9fc934ec01d"}},{"code_sha256_prefix":"c416c26539f06ac1","entry":"run_single_evaluation","repo":"Kimho666/STEP","repo_kind":"found_in_text","path":"eval_cfm.py","file_url":"https://github.com/Kimho666/STEP/blob/HEAD/eval_cfm.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"c416c26539f06ac1"}},{"code_sha256_prefix":"4194cf80b56b4bde","entry":"worker_fn","repo":"Kimho666/STEP","repo_kind":"found_in_text","path":"ray_exec.py","file_url":"https://github.com/Kimho666/STEP/blob/HEAD/ray_exec.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"4194cf80b56b4bde"}}]},"arxiv_metadata":{"licence":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)","fields":["title","abstract","authors","date"],"primary_category":"cs.AI","source":"arxiv_2026.jsonl"},"syntology_extracted_results":null}