SafeDiffuser: Safe Planning with Diffusion Probabilistic Models

Wei Xiao,Tsun-Hsuan Wang,Chuang Gan,Daniela Rus
2023-06-01
Abstract:Diffusion model-based approaches have shown promise in data-driven planning, but there are no safety guarantees, thus making it hard to be applied for safety-critical applications. To address these challenges, we propose a new method, called SafeDiffuser, to ensure diffusion probabilistic models satisfy specifications by using a class of control barrier functions. The key idea of our approach is to embed the proposed finite-time diffusion invariance into the denoising diffusion procedure, which enables trustworthy diffusion data generation. Moreover, we demonstrate that our finite-time diffusion invariance method through generative models not only maintains generalization performance but also creates robustness in safe data generation. We test our method on a series of safe planning tasks, including maze path generation, legged robot locomotion, and 3D space manipulation, with results showing the advantages of robustness and guarantees over vanilla diffusion models.
Machine Learning,Robotics,Systems and Control
What problem does this paper attempt to address?
The paper aims to address the safety issues of diffusion models in data-driven planning. Specifically, although diffusion model-based methods show potential in data-driven planning, they lack safety guarantees, making it difficult to apply these methods in scenarios with high safety requirements. To tackle this challenge, the paper proposes a new method—SafeDiffuser, which ensures that the diffusion probabilistic model meets specific safety specifications by using a class of Control Barrier Functions (CBFs). The core idea of SafeDiffuser is to embed finite-time diffusion invariance in the denoising diffusion process, thereby achieving trustworthy data generation. This method not only maintains generalization performance but also enhances robustness in safe data generation. Experimental results show that SafeDiffuser performs excellently in a series of safety planning tasks, including maze path generation, legged robot walking, and 3D space manipulation.