SAD-Flower: Flow Matching for Safe, Admissible, and Dynamically Consistent Planning

Anonymous Author(s)
Affiliation

We propose a novel control-augmented flow matching framework for generating Safe, Admissible, and Dynamically consistent trajectories.

Abstract

Flow matching (FM) has shown promising results in data-driven planning. However, it inherently lacks formal guarantees for ensuring state and action constraints, whose satisfaction is a fundamental and crucial requirement for the safety and admissibility of planned trajectories on various systems. Moreover, existing FM planners do not ensure the dynamical consistency, which potentially renders trajectories inexecutable. We address these shortcomings by proposing SAD-Flower, a novel framework for generating Safe, Admissible, and Dynamically consistent trajectories. Our approach relies on an augmentation of the flow with a virtual control input. Thereby, principled guidance can be derived using techniques from nonlinear control theory, providing formal guarantees for state constraints, action constraints, and dynamic consistency. Crucially, SAD-Flower operates without retraining, enabling test-time satisfaction of unseen constraints. Through extensive experiments across several tasks, we demonstrate that SAD-Flower outperforms various generative-model-based baselines in ensuring constraint satisfaction.

Prescribed-Time Scheduling Control

Given the test-time constraint (red ellipses and super-ellipses below), SAD-Flower generates safe, admissible, and dynamically consistent trajectories without retraining. The left figure shows the \( t \in [0, T_0) \) period without prescribed-time control, while the right figure shows the \( t \in [T_0, 1] \) period with prescribed-time control enforcing constraint satisfaction within the end of flow time.

Prescribed-Time Scheduling Control demo

Demo in Robot Locomotion

Given the test-time constraint (red roof below), SAD-Flower generates safe, admissible, and dynamically consistent trajectories withoput retraining. The left figure is the hopper system and the right figure shows the walker2d system. Both of the torso positions are constrained to be below the red roof, which demonstrating the effectiveness of SAD-Flower.

Robot Locomotion demo

Dexterous Grasping Demo for High-Dimensional System

Given the test-time constraint (joint limitations to avoid collision), SAD-Flower generates safe, admissible, and dynamically consistent trajectories for relocate tasks.

Dexterous Grasping demo