Adversarially Trained Weighted Actor-Critic for Safe Offline Reinforcement Learning

We propose WSAC (Weighted Safe Actor-Critic), a novel algorithm for Safe Offline Reinforcement Learning (RL) under functional approximation, which can robustly optimize policies to improve upon an arbitrary reference policy with limited data coverage. WSAC is designed as a two-player Stackelberg game to optimize a refined objective function. The actor optimizes the policy against two adversarially trained value critics with small importance-weighted Bellman errors, which focus on scenarios where the actor's performance is inferior to the reference policy. In theory, we demonstrate that when the actor employs a no-regret optimization oracle, WSAC achieves a number of guarantees: (i) For the first time in the safe offline RL setting, we establish that WSAC can produce a policy that outperforms any reference policy while maintaining the same level of safety, which is critical to designing a safe algorithm for offline RL. (ii) WSAC achieves the optimal statistical convergence rate of $1/\sqrt{N}$ to the reference policy, where $N$ is the size of the offline dataset. (iii) We theoretically show that WSAC guarantees a safe policy improvement across a broad range of hyperparameters that control the degree of pessimism, indicating its practical robustness. Additionally, we offer a practical version of WSAC and compare it with existing state-of-the-art safe offline RL algorithms in several continuous control environments. WSAC outperforms all baselines across a range of tasks, supporting the theoretical results.

Paper

Similar papers

Peer review

Reviewer HDvF5/10 · confidence 2/52024-06-30

Summary

The Weighted Safe Actor-Critic (WSAC) is a new Safe Offline Reinforcement Learning algorithm designed to outperform any reference policy while ensuring safety with limited data. It uses a two-player Stackelberg game to achieve optimal convergence and safe policy improvements. In practical tests, WSAC surpasses baselines in continuous control environments.

Strengths

- The ability of WSAC to outperform the behavior policy over a wide range of hyperparameters is a crucial property for practical use. - The author provides theoretical proof.

Weaknesses

- Given that I haven't examined the mathematical details, I find that many of the assumptions and proofs of key theorems in the paper resemble those in ATAC [1]. The primary differences are the authors' focus on the safe offline RL setting and the inclusion of a cost value in their theory. However, the use of a primal-dual approach in the algorithm's implementation may introduce training stability issues [2]. From both theoretical and practical implementation perspectives, it is difficult to identify novel insights in the paper. - The author needs to compare more state-of-the-art baselines, such as CDT [3] and FISOR [2]. - Line 34 contains a duplicate citation. [1] Cheng, Ching-An, et al. "Adversarially trained actor critic for offline reinforcement learning." *International Conference on Machine Learning*. PMLR, 2022. [2] Zheng, Yinan, et al. "Safe offline reinforcement learning with feasibility-guided diffusion model". *International Conference on Learning Representations* (2024). [3] Liu, Zuxin, et al. "Constrained decision transformer for offline safe reinforcement learning." *International Conference on Machine Learning*. PMLR, 2023.

Questions

- What about the performance under different cost limits? The average cost in Table 2 does not adequately reveal the safety of the algorithm.

Rating

5

Confidence

2

Soundness

2

Presentation

2

Contribution

2

Limitations

n/a

Authorsrebuttal2024-08-11

> Averaging the number of safety constraint violations **There seems to be some confusion.** In our paper, we used a single cost limit (thus, there was no point in taking an average) similar to what the baselines (CDT, COptiDICE, CPQ) did in their original papers. The average is taken over different random seeds, but we used **a single cost limit**. We believe that the results over different random seeds in Table 2 can correctly reflect the true performance of our approach. From Table 2, it is clear for the single cost limit, our approach is the *only one* that can satisfy the constraints across different benchmarks. During the rebuttal phase, per the reviewer's request, we ran the algorithm with different cost limits following the exact format used in the Offline Safe RL Benchmark (OSRL) paper, where they report the average across performance with different cost limits. It is clear that our algorithm performs well, as no other algorithms are consistently safe, in terms of average performance. To further address the reviewer’s concern, we report individual results with different cost limits under our algorithm in the following table, where our algorithm achieves very low costs and high safety rates and is nearly safe for all environments. Note that due to time constraints in the rebuttal phase, we did not perform any parameter tuning, and we believe we can further improve the performance (both in terms of reward and safety) if we do so. Note that, we can't compare with other baselines since the OSRL paper doesn't have the results for different cost limits (they only have the average results). We also believe that although one of our main contributions is [theoretical](https://openreview.net/forum?id=82Ndsr4OS6&noteId=fBIu70VHUw), our approach with theoretical support is quite general and has the potential to be incorporated into other practical Safe-RL algorithms. | | Reward ↑ | Cost ↓ | Reward ↑ | Cost ↓ | Reward ↑ | Cost ↓ | |-------------|----------|--------|----------|--------|----------|--------| | Cost Limit | 10 | | 20 | | 40 | | | BallCircle | 0.71 | 0.10 | 0.76 | 1.17 | 0.75 | 0.27 | | CarCircle | 0.60 | 0.07 | 0.67 | 0.99 | 0.68 | 0.59 | | | Reward ↑ | Cost ↓ | Reward ↑ | Cost ↓ | Reward ↑ | Cost ↓ | |-------------|----------|--------|----------|--------|----------|--------| | Cost Limit | 20 | | 40 | | 80 | | | PointButton | 0.01 | 0.47 | 0.13 | 0.67 | 0.18 | 0.51 | | PointPush | 0.10 | 1.11 | 0.07 | 0.52 | 0.05 | 0.21 | **Table 1: The normalized reward and cost of WSAC for different cost limits. Each value is averaged over 20 evaluation episodes, and 3 random seeds.** > limited subset of environments We focus on environments where most (if not all) baselines are unsafe (e.g., no baselines are safe in PointButton, and only BCQL is safe in PointPush). Yet, we show that our proposed approach can achieve safety while maintaining good reward which points towards its efficacy. We believe these environments are both challenging and representative, effectively justifying an algorithm's ability to guarantee safety. We would like to emphasize that our main contribution in this paper is to address the Safe Robust Policy Iteration (SRPI) and policy coverage limitations in theoretical offline safe RL which are quite important in the theoretical safe-RL community (see Table 1). For example, none of the existing approaches (including the baselines) have a **safe robust policy improvement guarantee** using only a *single policy coverage assumption*. In particular, our approach provides a way to achieve a safe policy using only offline data with bare minimum richness (single policy coverability). Existing approaches that are based on primal-dual concept require more richness in data (all policy concentrability which is not possible to achieve in practice). Please see Table 1 and discussion in Introduction. Furthermore, the baselines do not provide any theoretical guarantees, since they aim to design practical algorithms. To demonstrate the empirical efficiency of our approach, we included four challenging environments in our paper. We agree that running the algorithm on all 38 environments in the baseline Safe RL paper would surely have values, and we will try to evaluate our approach on more baselines in the final version. However, we believe that the environments we selected are sufficient to demonstrate the core ideas of our paper and validate the theoretical insights. Even the state-of-the-art algorithms (without theoretical guarantees) only include a limited number of representative environments in their papers; for instance, CDT has 5, CPQ has 3, and COptiDICE has 4. We sincerely hope that the reviewer will reevaluate the rating based on the novel contributions of our paper.

Reviewer HDvF2024-08-12

After considering the positive feedback provided by other reviewers on the theoretical aspects, I will increase the score from 3 to 5. However, I still think that the main proof core and the concept of safe policy improvement primarily originate from ATAC. I suggest that other reviewers might want to further compare the theoretical aspects of ATAC with those presented in this paper. Cheng, Ching-An, et al. "Adversarially trained actor critic for offline reinforcement learning." International Conference on Machine Learning. PMLR, 2022. Additionally, concerning the algorithm's performance on benchmarks and the selection of baselines, I think it would be prudent to include more advanced state-of-the-art approaches. After all, CPQ and Copitidice are articles from 2022, and bcql and bearl come from earlier offline RL algorithms. Regarding the recently well-performing algorithms like CDT and FISOR, it would be beneficial for the author to explore or discuss the feasibility of transitioning the WSAC theoretical framework to SOTA. Regarding the issue of non-comparison due to different settings mentioned by the author, it's noteworthy that while CDT requires the introduction of additional information such as target reward and target cost, the baseline in the paper, including WSAC, also necessitates a human-defined cost limit. For FISOR, since safety is a key goal in safe RL research, it doesn't make sense to say that stricter safety constraints stop us from comparing safety between different algorithms.

Authorsrebuttal2024-08-13

Thanks for increasing your score and engaging with us. > Technical Differences with ATAC ATAC is the first paper in the literature to investigate the property of RPI. While we certainly draw insights from their results, our work has the following significant differences. - First, we focus on the **constrained Markov decision process (CMDP)** setup rather than an unconstrained setup. The CMDP setup is fundamentally different from the unconstrained one. For example, in CMDP, the optimal policy can be stochastic, unlike in the unconstrained MDP. In the constrained setup, it is essential to bound both the sub-optimality of the reward and the constraint violation, whereas, in the unconstrained setup, only the sub-optimality of the reward needs to be bounded. Naturally, the analytical results and algorithms differ significantly from those in the unconstrained ATAC setup. - Furthermore, our approach to training the critics is different from that of ATAC. ATAC uses a squared Bellman error, while we utilize the average Bellman error to train the critic. Consequently, we achieve a $1/\sqrt{N}$ sample complexity error, while ATAC achieves a $1/N^{1/3}$ sample complexity error. The key difference is that we use an importance-weighted Bellman error to obtain an unbiased estimator of the critic for both the reward and cost, tuning the weight parameter to achieve a better rate, unlike ATAC. - Moreover, while primal-dual-based methods exist for solving the offline CMDP, achieving robust safe policy improvement and relaxing the assumption of all-policy concentrability remained open challenges (Table 1 in our paper). We resolved this open question. Our approach guarantees robust policy improvement, so if a safe reference policy (e.g., a safe behavioral policy) is provided, our algorithm will yield a policy that remains safe without sacrificing reward. Such a guarantee was previously missing from the literature. As pointed out in the introduction, **all-policy** concentrability is difficult to satisfy in practice, especially in a safe setting where the dataset may not cover state-action pairs from an unsafe policy. Instead, we only require single-policy concentrability, making our theoretical results highly impactful for the safe RL community. - We consider a policy improvement over **any reference policy** not only the behavior policy. - It is worth noting that to provide such a guarantee, we developed a rectified penalty-based approach rather than a primal-dual-based one. As a result, our analysis differs from existing primal-dual approaches. In fact, the existing primal-dual-based approaches only guarantee all-policy concentrability, so our analytical insights and proposed approach open new avenues for finding a safe policy from an offline dataset. > compare with the SOTA baselines it is crucial to compare with the state-of-the-art (SOTA) baselines to demonstrate the strength of our proposed approach. Therefore, we compare our practical version with existing approaches on selected benchmarks. As requested by the reviewer, we have included results from CDT in our rebuttal. We apologize for not making this clearer earlier. Notably, ours is the only approach that achieves safety, underscoring the efficacy of our method. Additionally, CDT uses a transformer architecture, which naturally results in a longer computational time compared to our [approach](https://openreview.net/forum?id=82Ndsr4OS6&noteId=VR6S2gv84s). Finally, we would like to mention that in safe RL, there are two types of constraints: soft constraints (in the long-term average sense) and hard constraints (step-wise). It is difficult to say which one is more important because, in the long-term average case, taking some risk is necessary; otherwise, the problem would be no different from an unconstrained problem. Moreover, the existing solutions in theoretical safe RL for addressing these two types of constraints are significantly different. We agree that the cost limits are chosen by humans, and we appreciate the reviewer pointing out that a fairer comparison should consider different sets of cost limits. We observe that there is a trend: the reward increases when the cost limit is higher. To understand the differences between two settings, we can also observe that as reported in the FISOR paper, some environments (e.g., SwimmerVel, CarButton1, CarGoal2) exhibit very low or even negative rewards because they aim to learn very safe policies. We will definitely add more discussion in the final revision. We are also happy to learn more about the reviewer's opinion on selecting the cost limits. What we typically do is to make sure the problem itself is feasible and the optimal solution is stochastic in synthetic CMDPs and follow what people use (like OSRL and other baselines) in complicated environments. **We hope that our response addresses the concerns of the reviewer and is open to further discussion. Thank you again for raising the score!**

Reviewer Tiai7/10 · confidence 3/52024-07-13

Summary

For safe RL methods, a desired property is Safe Robust Policy Improvement(SRPI), which means the learned policy is always at least as good and safe as the baseline behavior policies. But it's not achieved yet. Also, the traditional Actor-Critic framework may suffer from insufficient data coverage, which may fail to provide an accurate estimation of the policy for unseen states and actions. To address the issue, [45] and [11] use absolute pessimism or relative pessimism. However, his kind of approach fails to achieve the optimal statistical rate of $\sqrt{N}$. For addressing efficient policy improvement, the most commonly used approach for addressing safe RL problems is primal-dual optimization, but this method requires all policy concentrability, that is, the dataset must cover all possible strategies, which is impractical for the safe-related dataset. In contrast, the authors propose an aggression-limited objective function, the high-level intuition behind it is that by appropriately selecting a 𝜆, all unsafe policies are penalized. As a result, the policy that maximizes the objective function is the optimal safe policy. This formulation is fundamentally different from the traditional primal-dual approach as it does not require dual-variable tuning, and thus, does not require all policy concentrability. Beyond that, the proposed method also proved to achieve SRPI. [45] Tengyang Xie, Ching-An Cheng, Nan Jiang, Paul Mineiro, and Alekh Agarwal. Bellman-consistent pessimism for offline reinforcement learning. [11]Ching-An Cheng, Tengyang Xie, Nan Jiang, and Alekh Agarwal. Adversarially trained actor critic for offline reinforcement learning

Strengths

1. The writing logic is clear and reasonable. This paper is written with solid insights and emphasizes research gaps and innovations. 2. The authors conducted wide-range experiments and comparisons with other methods. And in terms of safety, the proposed method achieved SOTA performance. 3. There are no obvious red flags or drawbacks in this paper.

Weaknesses

1. In this paper's setting, safety is measured purely by cost, which is not always practical. e.g. in the real world, the cost function could be implicit or impossible to get. 2. The authors could try to combine the proposed method with other RL methods for a certain application to further justify its effectiveness.

Questions

The paper looks good. And there is a suggestion for future directions. Bridge the gap between theory and practice. Current RL methods have some bottlenecks like long-horizon tasks, safety, sample efficiency, etc.. Most of the methods to solve these bottlenecks come out of intuition instead of theory deduction. And some methods leverage external tools like foundation models(e.g. LLM), and control theories. It might be interesting to explain why these intuition or external tools work in view of theories.

Rating

7

Confidence

3

Soundness

3

Presentation

3

Contribution

3

Limitations

The authors did not provide limitations. A possible limitation may come from the experiment on PointPush, in which the simple BC's reward outperformance proposed method without sacrificing too much safety. And as stated in weakness, safety is expressed purely by the cost function, which sometimes is hard to get in the real world.

Reviewer U4TL6/10 · confidence 3/52024-07-13

Summary

This paper proposes weighted safe actor-critic, and provides corresponding theoretical analysis on its optimal statistical rate. Some interesting technical tools were introduced. The authors also implement a practical version of WSAC and evaluate it against SOTA offline safe RL baselines in continuous control tasks.

Strengths

(1) This paper addresses offline safe RL with adversarial trained weighted AC framework, showing its optimal statistical convergence rate. (2) Under the perfect function approximation assumption, the authors show WSAC outperforms any reference policy while maintaining the same level of safety. Besides, the theoretical finding on safe robust policy improvements bring insight to the offline safe RL methods. (3) Empirically, the authors provide a comparison to a set of baselines in OSRL benchmarks.

Weaknesses

See more discussion in the question parts.

Questions

(1) **Clarification of "Adversarial"**: I'd like to have a clarification of the term "adversarial" in this offline safe RL problem. Do the authors mean "adversarial" in that the cost critic always update cost critic via optimism and reward critic via pessimism? Since there are other formats of adversarial robustness in other components of safe RL [1], I may be helpful if the authors could clarify it in the early stage of this paper. (2) **Finite selection of W**: the current WSAC algorithm prototype only considers a discrete selection of $w$, can they be arbitrarily assigned for different offline datasets and environments? Intuitively if an arbitrary $w$ is close enough to its neighbor $w$ in set $\mathcal{W}$, we can still provide analytical bound of performance under slightly different $w$. (3) **Lack of discussion of assumption gap in the pratical version**: the authors may describe how certain assumptions may not hold in practice, as they are some seemingly relatively strong assumptions in the theoretical analysis, like approximate realizability, though some of them have already got loosened. Also, some experiment details (e.g. the behavior policy or the oracle policy) can be discussed in the appendix to provide more contexts in how WSAC can help practically. (4) **Capability under sparse-cost setting**: in many real-world applications, safety violations will occur only in long-tail events. Can the reweighting scheme also address such long-tail cases in the cost critic learning? Can the current WSAC framework can be extended for this kind of analysis based on the weighting technique? (5) **Extension of the current framework to multi-constraint settings**: in real-world applications, there might be multiple objectives and constraints, can the WSAC framework adapt to similar settings? (6) **Selection of weight W in the experiments**: Compared to algorithm 1, the practical implementation of WSAC seems to miss $\mathcal{W}$, how is this importance weight computed in practice? > [1] Liu, Zuxin, et al. "On the robustness of safe reinforcement learning under observational perturbations." *arXiv preprint arXiv:2205.14691* (2022).

Rating

6

Confidence

3

Soundness

3

Presentation

3

Contribution

3

Limitations

The authors have clearly defined the scope and discussed the limitations of this paper.

Authorsrebuttal2024-08-12

We greatly thank the reviewer for considering reevaluating our paper and for raising these two interesting questions. >Extracting the Behavior Policy In the standard offline RL setting, we assume that the dataset is generated by some behavior policy $\mu(s,a)$ and is i.i.d. Therefore, as long as the state-action space is finite, the method we provided guarantees that we can accurately estimate the behavior policy. This holds true whether we are considering safe RL or regular RL, as the only difference is the additional information regarding the cost function, while the process of generating the dataset remains the same. For more complicated cases, such as when the state space is continuous, we can use a neural network to approximate the policy by minimizing the distance between the learned policy and the behavior policy. Alternatively, we could use DAgger (Dataset Aggregation) to achieve an even better policy. However, in such cases, the assumptions made in offline RL (not only in our paper but in the field generally) may no longer hold, particularly the data/policy coverage assumption. This is why we argue in our paper that single-policy coverage is crucial since it is much weaker than the full-policy coverage assumption. >extended to the non-tabular cases There seems to be some confusion. We consider the function approximation setting, not a tabular setting, and the theoretical results are independent of the size of the state and action space under the given assumptions. The practical version of our algorithm is designed to develop a deep neural network approach for more complex environments. In order to handle the continuous state and action spaces, we use an actor-critic approach to solve the optimization problem (2), which aligns with the objective in our theoretical version. Specifically, we can use the aggression-limited objective to train the actor-network, which is feasible by considering two Q-value neural networks (one for reward and one for cost). The method used in our practical version for the weighted Bellman regularizer is a very simplified version. However, we believe it is possible to approximate $w(s,a)$ with another neural network such that the weighted Bellman error is minimized when the critic network is fixed. We will add more discussions and possibly some results in the revision. Please let us know if you have further questions and comments, we are glad to have more discussions.

Reviewer U4TL2024-08-13

I thank the authors for their extensive replies. Also, I read the discussion between the authors and reviewer HDvF. I think most of my questions have been well-addressed with the clarification of their theoretical contribution and the new empirical evidence. I will raise my score to 6 in favor of the acceptance.

Authorsrebuttal2024-08-13

Thank you very much again for your great comments and for taking the time to engage with us. We sincerely appreciate your acknowledgment and positive feedback on our work!

Reviewer 1bfE6/10 · confidence 3/52024-07-29

Summary

This paper introduces a principled approach for safe offline reinforcement learning (RL), aimed at robustly optimizing policies beyond a given reference policy, particularly when constrained by the limited data coverage of offline datasets. The traditional constrained actor-critic methods face challenges including (1) coping with insufficient data coverage, (2) ensuring robust policy improvement, and (3) facilitating computationally efficient actor optimization. To address these limitations, this study presents the Weighted Safe Actor-Critic (WSAC) framework. WSAC incorporates (1) a pessimistic bias through a weighted average Bellman error, (2) theoretical assurances for robust policy improvement, and (3) an efficiency advantage over traditional primal-dual optimization methods. WSAC employs an aggression-limited objective function, which discourages unsafe policies, relying on less stringent assumptions compared to prior methodologies. Furthermore, WSAC leverages the reference policy as a no-regret policy optimization oracle, allowing for safe policy training. The efficacy of WSAC is demonstrated across four benchmark environments: BallCircle, CarCircle, PointButton, and PointPush. The results indicate that WSAC effectively optimizes policies to maximize cumulative rewards while maintaining cumulative costs below predefined thresholds.

Strengths

This work is well-grounded in rigorous theoretical principles that effectively support the proposed WSAC method. The method's foundation on pessimistic value estimation and robust policy improvement is both mathematically sound and appropriate for the challenges of safe offline RL. Furthermore, the integration of an adversarial training component within the actor-critic architecture introduces a novel strategy for mitigating common issues such as insufficient data coverage in offline RL. This significantly bolsters the robustness of the resulting policies against shifts in data distribution, a crucial factor for applications in real-world scenarios. By addressing the critical issue of safety in policy optimization with a computationally efficient approach, I believe that the paper makes a substantial contribution to moving the field towards practical, deployable reinforcement learning systems capable of addressing real-world challenges.

Weaknesses

The proposed Weighted Safe Actor-Critic (WSAC) method in this submission is contingent upon the availability of an explicit reference policy, such as a behavior policy derived from the offline dataset. This requirement could make training difficult in scenarios where extracting a reliable reference policy from the offline data is challenging, particularly for algorithms aiming to be behavior-agnostic in offline RL settings. Additionally, the authors claim in line 239 that "Our approach is very computationally efficient and tractable compared with existing approaches." However, the absence of empirical evidence, such as wall-clock time comparisons, to substantiate this claim weakens their argument. Providing such comparative data would significantly strengthen their case for computational efficiency. Moreover, the paper does not include ablation studies to elucidate the contributions of the three key components of WSAC: (1) weighted Bellman error, (2) aggression-limited objective, and (3) no-regret policy optimization using a single reference policy. Identifying which of these components is most critical to performance enhancement would provide clearer insights into the framework's effectiveness and areas for potential improvement.

Questions

Q1. Could you elaborate on the sensitivity of the hyperparameters such as $\beta_c$, $\beta_r$, $\lambda$? Understanding their influence on the model's performance and robustness would be beneficial, especially in varying training conditions. (Minor Comments) 1. In obj 2, the Bellman error coefficients $\beta$ used in the reward and cost constraints appear to have different values, indicated by $\beta_c, \beta_r$. To avoid confusion, it would be prudent to denote these coefficients separately throughout the manuscript to reflect their distinct roles and values. 2. The typo in line 231 : "WSAC sovles" → "WSAC solves."

Rating

6

Confidence

3

Soundness

3

Presentation

3

Contribution

3

Limitations

The authors addressed their limitations in Section 4 and Conclusion and the broader societal impact in Checklist #10: Broader Impacts.

Reviewer 1bfE2024-08-11

Thank you for addressing my concerns. It would be beneficial if the authors could report the error bars (e.g., standard errors or confidence intervals) of the computational cost and results of the ablation study in the revision. I still believe this paper makes sufficient contributions to the offline RL community, so I will be maintaining my rating.

Authorsrebuttal2024-08-12

We thank the reviewer again for the positive evaluation of our paper. We provide the error bars in the following tables. | | BEARL | CPQ | CDT | COptiDICE | WSAC | |--------------------|--------|--------|--------|-----------|--------| | **Time (seconds)** | 120.0 | 113.8 | 464.6 | 112.0 | 116.40 | | **STD** | 1.41 | 2.13 | 1.62 | 5.05 | 1.85 | | **Confidence Interval** | (116.07, 123.93) | (107.88, 119.37) | (460.09, 469.11) | (97.95, 126.05) | (111.25, 121.55) | **Table 1:** Training Time (seconds) for 200 steps over 5 repeat experiments | Components | Cost | Reward | Cost STD | Reward STD | Cost Interval | Reward Interval | |-------------------------------------------------------------------|-------|--------|----------|------------|------------------|--------------------| | **ALL** | 0.014 | 0.788 | 0.006 | 0.004 | (0.00, 0.03) | (0.78, 0.80) | | **W/O no-regret policy optimization** | 0.014 | 0.788 | 0.006 | 0.004 | (0.000, 0.028) | (0.779, 0.798) | | **W/O Aggression-limited objective** | 0.014 | 0.788 | 0.006 | 0.005 | (0.000, 0.028) | (0.778, 0.798) | | **W/O Weighted Bellman regularizer** | 0.323 | 0.684 | 0.061 | 0.017 | (0.185, 0.462) | (0.645, 0.724) | **Table 2:** Ablation study under tabular case (cost limit is 0.1) over 10 repeat experiments

Reviewer HDvF2024-08-11

Thank you for your response. The evaluation of safety should focus on whether the algorithm can still ensure safety under a single cost limit. Averaging the number of safety constraint violations across multiple cost limits does not accurately represent the policy's safety, as exceeding the cost limit in practical applications can result in unsafe outcomes. It would be better if the author could separately present the model's performance under different cost limits to demonstrate the safety guarantees provided by their theoretical approach. Furthermore, the author has selected only a very limited subset of environments in OSRL for comparison (4 out of 38 environments), which does not sufficiently demonstrate the algorithm's advantages over the baseline. Based on these points, I maintain my current score.

Reviewer U4TL2024-08-11

I thank the authors for their detailed response, especially their justification of the assumption. One clarification question on the statement: " The behavior policy is easy to achieve, even if it is not given to us, since extracting the behavior policy from an offline dataset is not difficult with behavior cloning (BC)." Since the example you provided is using pure tabular case, which is even weaker than your assumption (discrete action space + complex state space). How do you justify the difficulty of behavior policy extraction given the safety constraints in offline safe RL? Another follow-up question about the gap between theoretical assumptions and their empirical practicability is that most OSRL baselines have continuous action and continuous state space, and the authors propose a practical version of WSAC in the appendix. I'm curious about whether the authors can provide some insights on how the current theoretical guarantees can be extended to the non-tabular cases, if applicable. Besides the above two questions, most of the other questions are well-addressed by the author's response. I thank the authors again for their dedicated efforts and will determine my final score after this round of discussion.

Reviewer Tiai2024-08-12

Thank you for the response! And it helps me keep a positive opinion of this work, so I would maintain my score.

Program Chairsdecision2024-09-25

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC