Contractive Diffusion Policies: Robust Action Diffusion Via Contractive Score-based Sampling With Differential Equations

Abstract

Diffusion policies have emerged as powerful generative models for offline policy learning, whose sampling process can be rigorously characterized by a score function guiding a stochastic differential equation (SDE). However, the same score-based SDE modeling that grants diffusion policies the flexibility to learn diverse behavior also incurs solver and score-matching errors, large data requirements, and inconsistencies in action generation. While less critical in image generation, these inaccuracies compound and lead to failure in continuous control settings. We introduce contractive diffusion policies (CDPs) to induce contractive behavior in the diffusion sampling dynamics. Contraction pulls nearby flows closer to enhance robustness against solver and score-matching errors while reducing unwanted action variance. We develop an in-depth theoretical analysis along with a practical implementation recipe to incorporate CDPs into existing diffusion policy architectures with minimal modific

Stats

• citations0
• S2 citations—
• github stars0
• HF likes0
• heat score0.00
• arxiv keyabyaneh2026contractive

Related papers

• Streaming Diffusion Policy: Fast Policy Synthesis With Variable Noise Diffusion Models (2024)0.00
• Learning A Diffusion Model Policy From Rewards Via Q-score Matching (2023)0.00
• Diffusion Actor-critic: Formulating Constrained Policy Iteration As Diffusion Noise Regression For Offline Reinforcement Learning (2024)2.92
• Diffusion Policies With Value-conditional Optimization For Offline Reinforcement Learning (2025)0.00
• Policy-guided Diffusion (2024)0.00
• Diffcps: Diffusion Model Based Constrained Policy Search For Offline Reinforcement Learning (2023)1.91
• Enhanced DACER Algorithm With High Diffusion Efficiency (2025)0.00
• Fine-tuning Diffusion Policies With Backpropagation Through Diffusion Timesteps (2025)0.00