Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces a differentiable reachability framework using Taylor model flowpipe construction and CROWN-style bound propagation for neural network dynamics and control. The framework enables certified training of dynamics models and controllers, as well as reachability-aware MPC. Experiments on manipulation and quadrotor tasks demonstrate the framework's ability to perform online planning with certified reachable set over-approximations, even in high-dimensional systems.
Differentiable reachability unlocks certified training and control of neural network dynamics, enabling robots to plan safely in high-dimensional, uncertain environments.
Neural network (NN) dynamics models and control policies achieve strong performance in robotics, but providing sound guarantees under uncertainty remains difficult, especially for closed-loop NN systems. Existing reachability tools provide formal over-approximations, yet are often non-differentiable, overly conservative, or too slow for modern learning and online planning pipelines. To address this, we present a parallelizable, differentiable reachability framework in JAX for continuous- and discrete-time systems with analytical and NN-based dynamics and controllers. Our framework combines Taylor-model flowpipe construction with CROWN-style linear bound propagation through a unified representation that preserves affine dependencies while supporting GPU-batched computation and automatic differentiation. Building on this reachability primitive, we develop (i) a certified training method that encourages reachability-friendly dynamics models and controllers, and (ii) a reachability-aware sampling-based MPC scheme with gradient-based refinement. Experiments on non-prehensile manipulation and quadrotor tasks, including hardware and higher-dimensional evaluations (up to 72D), demonstrate practical online planning while maintaining certified reachable-set over-approximations under bounded uncertainty.