Papers
Topics
Authors
Recent
Search
2000 character limit reached

parallelcbf: A composable safety-filter and auditability framework for tensor-parallel reinforcement learning

Published 15 May 2026 in cs.LG and cs.RO | (2605.15509v1)

Abstract: While Isaac Lab provides massive parallel UAV simulation, OmniSafe and safe-control-gym provide constrained-RL benchmarks, and CBFKit provides control-barrier-function synthesis tooling, no existing framework unifies these capabilities for end-to-end safety-constrained training. ParallelCBF is the first framework to unify (i)~tensor-parallel UAV environments, (ii)~hard-gate CBF safety filters, (iii)~sharded BC-to-RL pipelines, and (iv)~first-class operational auditability -- pre-registration, watchdog registries, failure forensics, and dataset audits as composable APIs rather than user-implemented scripts. We release ParallelCBF v0.1.0 under Apache~2.0 with a four-layer composable API, a CPU PyTorch reference implementation of a dual-barrier (squared / linear-predictive) CBF, property-based safety invariance tests across vectorized batch sizes that complete in 1.67~s for the full 39-test suite, and a 31{,}415-episode behavior-cloning collection campaign whose curriculum mix, per-bucket yields, and dataset SHA-256 are auditable through the framework's own \texttt{ops} primitives. We report a representative end-to-end pipeline execution in which the framework's auditability layer halted a downstream training stage that did not meet pre-registered convergence criteria, preventing silent propagation of a degraded checkpoint -- an architectural property we argue is necessary, not merely useful, for reproducible empirical robotics research. The framework is installable via \texttt{pip install parallelcbf}; source and release artifacts are available at https://github.com/xiaoyang-123-cell/ParallelCBF.

Authors (3)

Summary

  • The paper introduces a composable framework that unifies safety filtering and auditability for tensor-parallel reinforcement learning.
  • It employs a dual-barrier control mechanism with vectorized safety filters to ensure invariant safety under adversarial conditions.
  • The operational auditability layer enforces reproducibility through pre-registration, halt logic, and diagnostic traceability across training pipelines.

Summary of "parallelcbf: A composable safety-filter and auditability framework for tensor-parallel reinforcement learning" (2605.15509)

Motivation and Problem Statement

Safety-critical reinforcement learning (RL) in robotics presents significant challenges, most notably the fragmentation across the capabilities required for scalable, safety-constrained training. While simulation throughput, constrained-RL algorithm infrastructure, control-theoretic safety filters, and reproducibility tools exist, these components lack a unified, compositional API surface. The absence of a consolidated safety-filter abstraction, standardized warmstarts from behavior cloning (BC) to RL, and framework-level support for operational discipline (pre-registration, auditability, forensics) impedes reliable and reproducible empirical research. Existing frameworks (Isaac Lab, OmniSafe, safe-control-gym, CBFKit) only partially address these aspects and do not provide the contract-level compositional guarantees necessary for trust and reproducibility.

Framework Architecture and Key Contributions

The ParallelCBF framework addresses these limitations through a type-checked, four-layer compositional design:

Layer 1: Environments

Extends standard gymnasium environments with safety state introspection, structured safety metrics, and constraint violation reporting. The reference implementation is a 2D toy avoidance environment with vectorized rollouts.

Layer 2: Safety Filters

The SafetyFilter abstraction is instantiated by a dual-barrier Control Barrier Function (CBF) implementation (DualBarrierCBF), with closed-form quadratic programming for single-obstacle scenarios and batched execution across parallel environments. The separation of nominal action and safety-projected action is enforced, with the filter comprising both a hard non-collision barrier and a linear predictive soft barrier that absorbs dynamics lag.

Layer 3: Algorithms

Defines Stable-Baselines3-compatible interfaces for learning, prediction, saving/loading, and sequence-aware policies. While v0.1 features a trivial random action algorithm, substantive RL primitives (KL-anchored PPO, sharded BC) are scheduled for future releases.

Layer 4: Operational Auditability

This layer's auditability primitives are architecturally novel: pre-registration with SHA-256 hash commitments, an extensible watchdog registry to enforce halt conditions, failure forensics for post-hoc diagnostics, atomic checkpointing, and dataset audits as composable, type-checked APIs rather than user-level scripts. This ensures empirical runs are reproducible and contractually consistent, with operational discipline enforced structurally rather than by ad-hoc user conventions.

Empirical Validation and Numerical Results

Empirical validation of ParallelCBF demonstrates several key claims:

  • Behavior-cloning Data Collection:

A 50,000-attempt campaign with a pre-registered target mix for scene complexity resulted in 31,415 accepted episodes (aggregate yield 62.83%), exceeding the pre-registered prediction by +2.13 percentage points. Statistical analysis indicates that dry-run yield estimates systematically underpredict full-campaign yields, especially in higher-complexity buckets, and recommends broader initial samples for accurate allocation.

  • Test Suite and Safety Invariance:

The 39-case test suite including Hypothesis-based randomized property tests and vectorization invariants completes in 1.67 s, providing a practical reproducibility budget. Positive invariance of the hard-barrier safety filter under randomized initial states is empirically validated, ensuring that all distributed environments maintain provable safety contracts.

  • Auditability Demonstration:

The auditability layer successfully halted training when pre-registered convergence criteria were unmet, blocking downstream stages from operating on a degraded checkpoint. All diagnostic context was preserved for subsequent analysis. The halt logic was derived from the pre-committed spec, avoiding heuristic or experimenter-driven overrides.

  • Controlled Algorithmic Substitution:

Upon a failed pretrain due to representational bottleneck in a GRU policy, substitution with a causal Transformer backbone reduced validation loss from 0.0595 (GRU plateau) to 0.00603 over 49 epochs, maintaining all safety and operational contracts. The change required zero modifications outside the Layer 3 algorithm module, demonstrating strict composability and diagnostic traceability.

  • Safety Filter Robustness:

In adversarial rollouts with a poorly-behaving policy (covariate shift from behavior cloning), 0% hard-barrier collisions were observed across 200 episodes, with all failures being out-of-arena. This constitutes a strong empirical validation of the safety filter as an invariant layer, even when policy quality degrades.

Theoretical and Practical Implications

The architectural elevation of auditability primitives to first-class APIs directly advances ML reproducibilityโ€”a field-wide concern highlighted by recent meta-research. By anchoring operational discipline (pre-registration, halt logic, forensics) in type-checked, test-covered contracts, ParallelCBF reduces variability introduced by user conventions and scripting idiosyncrasies, providing a more robust platform for empirical validation and cross-lab reproducibility in robotics RL.

The strict layer isolation enables controlled experimentation, error localization, and post-mortem analysis. This composability is particularly critical as research progresses to more complex simulators and real-robot platforms, where silent propagation of training artifacts and empirical drift historically cause unreproducible results and unsafe deployments.

On the theoretical side, the dual-barrier safety formulation and its soundness under highly vectorized execution support the feasibility of scaling safety-constrained RL to large, parallelized workloads. The empirical validation against adversarially failing policies further strengthens the invariance claims.

Limitations and Future Directions

The v0.1.0 release constrains empirical scope to CPU-only, 2D toy environments and omits performance-intensive elements such as Isaac Lab integration and GPU execution. Substantive RL algorithms (e.g., KL-anchored PPO), advanced policy architectures (Mamba, 3D perception), and real-UAV case studies are slated for v0.2.0 and beyond. The frameworkโ€™s current surface does not yet encompass chance-constrained CBFs or complex multi-obstacle environments. The auditability layer exhibited a non-blocking error due to a configuration regression, subsequently corrected, highlighting the ongoing need for rigorous continuous integration.

Conclusion

ParallelCBF (2605.15509) introduces a structurally composable, type-checked, and operationally auditable framework for safety-critical, tensor-parallel RL. By unifying environment simulation, CBF-based safety filtering, modular algorithm contracts, and first-class auditability APIs, ParallelCBF establishes verifiable safety and reproducibility properties absent from prior toolkits. The empirical results substantiate both its numerical fidelity and architectural composability. This elevation of operational discipline to framework-level primitives is poised to impact reproducible research and safe RL pipeline deployment across simulated and real-world applications, with clear extensibility for future algorithmic and systemic advances in robotics AI.

Paper to Video (Beta)

No one has generated a video about this paper yet.

Whiteboard

No one has generated a whiteboard explanation for this paper yet.

Open Problems

We haven't generated a list of open problems mentioned in this paper yet.

Collections

Sign up for free to add this paper to one or more collections.