PoSafeNet: Safe Learning with Poset-Structured Neural Nets
Kiwan Wong, Wei Xiao, Daniela Rus
TL;DR
The paper tackles the challenge of enforcing safety in learning-based robotic controllers when safety constraints are heterogeneous and only partially ordered. It introduces PoSafeNet, a differentiable framework that enforces poset-respecting safety by performing sequential, closed-form halfspace projections corresponding to control barrier function (CBF) constraints, organized into linear extensions of a safety poset. The approach uses a multi-head architecture where each head embodies a linear extension; outputs are combined via convex mixtures or discrete head selection, with safety preserved by construction in hard selections and under certain geometric conditions in mixtures. Empirical results across 2D navigation, constrained manipulation, and vision-based driving demonstrate improved feasibility, robustness, and scalability compared to unstructured safety layers and differentiable QP-based baselines, highlighting the practical benefits of explicit poset-based safety for complex, safety-critical tasks.
Abstract
Safe learning is essential for deploying learningbased controllers in safety-critical robotic systems, yet existing approaches often enforce multiple safety constraints uniformly or via fixed priority orders, leading to infeasibility and brittle behavior. In practice, safety requirements are heterogeneous and admit only partial priority relations, where some constraints are comparable while others are inherently incomparable. We formalize this setting as poset-structured safety, modeling safety constraints as a partially ordered set and treating safety composition as a structural property of the policy class. Building on this formulation, we propose PoSafeNet, a differentiable neural safety layer that enforces safety via sequential closed-form projection under poset-consistent constraint orderings, enabling adaptive selection or mixing of valid safety executions while preserving priority semantics by construction. Experiments on multi-obstacle navigation, constrained robot manipulation, and vision-based autonomous driving demonstrate improved feasibility, robustness, and scalability over unstructured and differentiable quadratic program-based safety layers.
