Certified Safety Curation: Distribution-Free Guarantees for Safe Offline Reinforcement Learning
Introduces Certified Safety Curation, a filter-then-clone pipeline for safe offline reinforcement learning under weak safety supervision (only segment comparisons and occasional episode-level budget labels) Uses a state-only value function trained from pairwise segment comparisons to score entire trajectories, enabling trajectory-level selection rather than transition-level reweighting Applies Learn-then-Test calibration to certify a distribution-free $(\alpha, \delta)$ bound on the unsafe fract
Analysis
TL;DR
- Introduces Certified Safety Curation, a filter-then-clone pipeline for safe offline reinforcement learning under weak safety supervision (only segment comparisons and occasional episode-level budget labels)
- Uses a state-only value function trained from pairwise segment comparisons to score entire trajectories, enabling trajectory-level selection rather than transition-level reweighting
- Applies Learn-then-Test calibration to certify a distribution-free $(\alpha, \delta)$ bound on the unsafe fraction of selected data, with a closed-form refusal probability dependent on pool purity
- Achieves safety on 11 of 15 DSRL benchmark tasks, approaching the theoretical limit of cloning the ground-truth safe subset (which requires per-trajectory labels)
- Demonstrates that retraining strong full-label methods on the certified selection yields safe policies where no cost-target tuning could previously achieve safety
Why It Matters
This work addresses a critical practical gap: real-world safe RL systems often cannot afford transition-level cost labels, yet must still provide verifiable safety guarantees. By delivering distribution-free certification without assuming a parametric cost model, it enables deployment in safety-critical domains where both data efficiency and rigorous safety bounds are non-negotiable.
Technical Details
- Filter-then-clone pipeline: A state-only value function is trained from pairwise comparisons of short trajectory segments; this scorer then evaluates whole trajectories, which are filtered before behavior cloning
- Learn-then-Test calibration: A distribution-free statistical framework certifies a selection threshold with an $(\alpha, \delta)$ bound on the fraction of unsafe trajectories in the curated dataset, without requiring assumptions about the underlying data distribution
- Oracle justification: The authors prove that transition-level reweighting fails even with an exact value function, motivating trajectory-level selection as the correct abstraction
- Closed-form refusal probability: The certificate's failure probability has an explicit formula in terms of pool purity, estimated from a calibration sample and entered by the scorer only through this scalar
- Empirical evaluation: Tested on 15 DSRL (Deep Safety RL) tasks; the certified method satisfies cost budgets on 11 tasks, while the uncertified variant reaches 12, with the gap attributable to the cost of certification
Industry Insight
- The trajectory-level selection insight (vs. transition-level) should reshape how practitioners approach offline RL safety when cost labels are sparse, suggesting that coarser supervision can still yield strong guarantees if the aggregation unit is chosen correctly
- Distribution-free certification is a significant step toward deployable safe RL in regulated industries (autonomous vehicles, healthcare, robotics), where unverifiable "best-effort" safety is insufficient for compliance
- The closed-form refusal probability offers a practical calibration tool: practitioners can predictably tune the safety-certainty tradeoff without expensive cross-validation, making certified curation viable for production pipelines
Disclaimer: The above content is generated by AI and is for reference only.