Skip to yearly menu bar Skip to main content


Poster

SafeDiffuser: Safe Planning with Diffusion Probabilistic Models

Wei Xiao · Johnson (Tsun-Hsuan) Wang · Chuang Gan · Ramin Hasani · Mathias Lechner · Daniela Rus

Hall 3 + Hall 2B #163
[ ] [ Project Page ]
Thu 24 Apr 7 p.m. PDT — 9:30 p.m. PDT

Abstract:

Diffusion models have shown promise in data-driven planning. While these planners are commonly employed in applications where decisions are critical, they still lack established safety guarantees. In this paper, we address this limitation by introducing SafeDiffuser, a method to equip diffusion models with safety guarantees via control barrier functions. The key idea of our approach is to embed finite-time diffusion invariance, i.e., a form of specification consisting of safety constraints, into the denoising diffusion procedure. This way we enable data generation under safety constraints. We show that SafeDiffusers maintain the generative performance of diffusion models while also providing robustness in safe data generation. We evaluate our method on a series of tasks, including maze path generation, legged robot locomotion, and 3D space manipulation, and demonstrate the advantages of robustness over vanilla diffusion models.

Live content is unavailable. Log in and register to view live content