REVIEW 2 major objections 6 minor 21 references
CANDOR: Counterfactual ANnotated DOubly Robust Off-Policy Evaluation
T0 review · 2 major / 6 minor · reviewed 2026-08-11 · deepseek-v4-flash
Pith's one-line read The paper proves that imperfect counterfactual annotations should enter a doubly robust estimator only through its reward model: that estimator stays unbiased under biased annotations, and it performs best when the reward model is…
desk verdict The bias analysis and practical recommendation are solid and new; the variance theorem has a real algebra error and the sepsis experiments run outside the theory's coverage assumption. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The central mechanism is the placement of the counterfactual-annotated dataset $D_+$ inside the doubly robust decomposition $\hat{V} = (1/N)\sum_i [\hat{R}(s_i,\pi_e) + \rho_{s_i}(a_i)(r_i - \hat{R}(s_i,a_i))]$. The three estimators are the three nontrivial placements: DM+-IS fits the reward model $\hat{R}_+$ on $D_+$ and keeps the IS correction on factual rewards; DM-IS+ keeps the standard reward model and pushes annotations into the IS term through the augmented behavior policy $\pi_b^+(a|s) = \bar{W}(a|s,a)\pi_b(a|s) + \sum_{\tilde{a}\neq a} \bar{W}(a|s,\tilde{a})\pi_b(\tilde{a}|s)$, with per-sample weights that sum to one; DM+-IS+ does both. The augmented policy and weights preserve the factual context distribution when annotations are appended, and the naive alternative that skips them distorts the context distribution and is arbitrarily biased (Appendix D). The load-bearing identity is the telescoping in the proof of Proposition 1: $E_{a\sim\pi_b}[\rho_s(a)(r - \hat{R}_+(s,a))] = E_{a\sim\pi_e}[\bar{R}(s,a) - \hat{R}_+(s,a)]$, which exactly cancels the $E[\hat{R}_+(s,\pi_e)]$ term and leaves $v(\pi_e)$ with no dependence on the annotation bias $\epsilon_G$.
What would settle it
Run the DM+-IS estimator in a two-dataset regime where Assumptions 2 and 4 provably hold, with annotations biased by a known amount $\epsilon_G$; Proposition 1 predicts the estimate's bias is exactly zero for every $\epsilon_G$, so a statistically significant non-zero bias would refute it. A second practical check is to repeat the experiment in the single-dataset regime (reward model and OPE estimate on the same behavior data): if a systematic bias appears that grows with $\epsilon_G$, the two-dataset independence assumption is load-bearing and the guarantee does not transfer to that regime.
Extended reading notes
Core claim
Under Assumption 2 (annotations with expectation $\bar{R}(s,a) + \epsilon_G(s,a)$) and common support between behavior and target policies, Proposition 1 establishes that $E[\hat{V}^{\mathrm{DM}+\text{-IS}}] = v(\pi_e)$: the estimator that fits its reward model on the annotation-augmented dataset and applies standard importance sampling to factual rewards is unbiased, no matter how biased the annotations are. Theorem 2 shows the two estimators that route annotations through the augmented importance-sampling term, DM-IS+ and DM+-IS+, both carry the same bias term, $E_{s\sim d_0, a\sim\pi_e}[(1 - \bar{W}(a|s,a)\pi_b(a|s)/\pi_b^+(a|s))\epsilon_G(s,a)]$, which is the bias term that already affected the earlier IS+ estimator. The mechanism is a cancellation: the reward model's annotation-induced bias is exactly offset by the inverse-propensity correction $\rho(r_i - \hat{R}_+(s_i,a_i))$ evaluated on factual rewards, which remain unbiased. Empirically, across a two-context bandit, a Heartsteps step-count environment, and a sepsis treatment simulator with misspecified reward models, DM+-IS achieves the lowest RMSE across all tested magnitudes of annotation bias, and the paper concludes that any further use of imperfect annotations beyond the reward model can worsen the estimate. A lookup-table guide and a sensitivity analysis in the sepsis setting support the recommendation that DM+-IS is the safe default when annotation and reward-model quality are unknown.
Load-bearing premise
The unbiasedness proof assumes the reward model $\hat{R}_+$ is fitted on a behavior dataset separate from the one used to compute the OPE estimate; if a practitioner fits and evaluates on the same data, the cancellation that removes the annotation bias can break, and the paper does not analyze that single-dataset regime.
Editorial extensions
If this is right
- When counterfactual annotations are available, they should be used to fit the reward model only; feeding them into the importance-sampling term adds a bias proportional to the annotation bias.
- If annotation quality and reward-model misspecification are unknown, the empirical comparison supports DM+-IS as the default estimator, with sepsis-setting estimates within a small margin of the best method across all annotation qualities.
- The naive strategy of plugging the annotated dataset directly into a standard doubly robust estimator is unsafe because it silently changes the context distribution and can be arbitrarily biased.
- With perfect annotations and equal sample weights, IS+, DM-IS+, and DM+-IS+ coincide, so the choice among those three matters only when annotations are imperfect or weights are unequal.
- Under a misspecified reward model, pure DM and DM+ degrade while all doubly robust variants keep the protection of their IS term, and DM+-IS gets that protection without the annotation-bias cost of the IS+ variants.
Reading between the lines
- A general design rule suggested by Proposition 1, though not stated in the paper: in a doubly robust estimator, imperfect auxiliary data should enter through the model term that the unbiased factual correction can cancel, never through the reweighted term that inherits the auxiliary data's bias.
- The two-dataset assumption implies a practical requirement the paper leaves implicit: practitioners need two separate behavior datasets or sample splitting to realize the unbiasedness guarantee, since the single-dataset regime is not analyzed.
- A testable extension the paper does not run: with a fixed annotation budget, the theory suggests annotations should be spent where they most improve the reward model (under-covered contexts and actions), because placement does not affect bias, only the quality of the reward model.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper studies off-policy evaluation (OPE) in contextual bandits when counterfactual annotations are available but possibly imperfect. It proposes three doubly robust inspired estimators that incorporate annotations in different places: DM+-IS uses annotations only in the reward model, DM-IS+ uses them only in the importance sampling part, and DM+-IS+ uses them in both. The main theoretical result is Proposition 1, which states that under biased annotations and common support, DM+-IS is unbiased, while Theorem 2 shows that DM-IS+ and DM+-IS+ carry an annotation-bias term. Theorem 3 gives a variance decomposition for DM+-IS. Experiments on a two-context bandit, Heartsteps, and a Sepsis treatment simulator compare the estimators under misspecified reward models and imperfect annotations, leading the authors to recommend DM+-IS as the most robust choice. The appendix contains proofs, additional experiments, and a discussion of weighted versus unweighted reward models.
Significance. If the main claims hold, the paper provides clear practical guidance: when counterfactual annotations are imperfect, they should be used only in the direct-method component of a doubly robust estimator. Proposition 1 is a clean and useful result, and the paper is a rare OPE contribution that both proves a property and ships code and detailed empirical comparisons. However, the proof of Theorem 3 contains an algebraic error that invalidates the stated variance formula, and the Sepsis experiments violate Assumption 4 on which Proposition 1 depends. These issues affect the formal and empirical support for the 'most robust across all datasets' claim, so the manuscript needs revision before the recommendations can be accepted as stated.
major comments (2)
- [Appendix H.1.2 (Theorem 3)] The displayed derivation does not establish the stated variance formula. After the unbiased-DM substitution, the derivation reaches an expression of the form E_s[ Σ_a πe(a|s)^2 (1/πb(a|s) - 1) V_{D̂R}[R̂+(s,a)] ]. The next displayed line rewrites this as E_s E_{a∼πb}[(ρs(a)^2 - 1/πb(a|s)) Δ_{R̂+}(s,a)], which equals Σ_a (πe(a|s)^2/πb(a|s) - 1) Δ_{R̂+}(s,a), not the previous expression; the factor πe(a|s)^2 on the second term has been replaced by 1. The valid simplification would keep E_s[ E_{a∼πb}[ρs(a)^2 Δ_{R̂+}(s,a)] - Σ_a πe(a|s)^2 Δ_{R̂+}(s,a) ]. Since Theorem 3 is the variance result listed in Table 3 and is invoked in Section 6.1 to interpret the RMSE behavior of DM+-IS, the theorem must be corrected or explicitly qualified. The same passage also contains an apparent algebraic slip in the biased-DM case, where a term 2vπe(s)^2 R̃(s,πe) appears instead of 2vπe(s)R̃(s,πe).
- [Appendix B.3 and Section 6.2] The Sepsis experiments include behavior policies that violate Assumption 4, which is required by Proposition 1. For example, πb1 = [0.1,0.1,0.4,0.3,0.1,0,0,0] and πb2 = [0.1,0.1,0.4,0.2,0.1,0.1,0,0] place zero mass on actions that the target policy πe = [0.3,0.2,0,0,0.2,0.1,0.1,0.1] supports. The proof of Proposition 1 uses the identity E_{a∼πb}[ρ(a)f(a)] = E_{a∼πe}[f(a)], which fails when πb does not cover πe. For unsupported actions, the estimator carries an additional, unmodeled bias of order Σ_{a:πb(a|s)=0} πe(a|s)(R̄(s,a) - E[R̂+(s,a)]), which is not analyzed. Nevertheless, Figure 3c averages RMSE over these behavior policies and Section 6.2 concludes that DM+-IS is most robust 'across all datasets.' Please either restrict the Sepsis evaluation to behavior policies satisfying Assumption 4 or extend the theoretical analysis to partial coverage and report the corresponding bias term.
minor comments (6)
- [Section 4, Theorem 3 display] The displayed variance formula has a bracket mismatch and an unclosed expectation expression; please correct the typesetting so that the formula is parseable.
- [Appendix H.1.1, Propositions 12 and 13] Propositions 12 and 13 are stated under Assumptions 1 and 5, but their proofs use the standard IS ratio and rely on Assumption 4 (common support without annotations). Please align the stated assumptions with the proofs.
- [Appendix C and Acknowledgements] The reported total compute time differs between the acknowledgements (approximately 30 hours) and Appendix C (approximately 100 hours); please reconcile these numbers.
- [Appendix I, Corollary 18] The corollary title lists DM+-IS, but the statement and proof concern DM-IS+ and DM+-IS+; please correct the title.
- [Appendix B.1] Please clarify whether the dataset used to fit R̂ and R̂+ is independent of the dataset used to form the OPE estimate, as required by Section 4; the phrase 'the same dataset size and distribution is used to learn a reward function estimate' is ambiguous.
- [Section 7] In the paragraph on unknown reward model and annotation quality, 'annotation equality' should read 'annotation quality.'
Circularity Check
No circular derivation: DM+-IS unbiasedness is a standard DR cancellation; self-citations to prior IS+ work are not load-bearing.
full rationale
The paper's central theoretical claim (Proposition 1) is not circular. The proof in Appendix H.1.1 shows that E[V_hat^{DM+-IS}] = v(pi_e) by a standard doubly robust cancellation: under Assumption 4, the IS correction term E_{a~pi_b}[rho(s,a)(Rbar(s,a) - E[R_hat+(s,a)])] is converted to E_{a~pi_e}[Rbar(s,a) - E[R_hat+(s,a)]], which cancels the DM term E_{a~pi_e}[E[R_hat+(s,a)]]. This cancellation holds for any reward model estimate, including one whose expectation is biased by imperfect annotations, because the reward model is fitted on an independent dataset, as explicitly assumed in Section 4. Thus the unbiasedness result follows from the estimator's algebraic structure, not from assuming the conclusion. Theorem 2's bias expression for DM-IS+ and DM+-IS+ is derived in Appendix H.2.1; the observation that the bias term is identical to that of IS+ is a comparison, not a load-bearing citation. The self-citations to Tang and Wiens supply the IS+ estimator and the augmented behavior policy pi_b+ construction, but the paper re-derives the needed identities and the recommended estimator, DM+-IS, does not rely on IS+ being correct. There are no fitted parameters renamed as predictions, no imported uniqueness theorems, and no ansatz smuggled in solely by citation. The Sepsis support violation is an internal-validity concern about the empirical claim, not a circularity of the derivation.
Assumptions & free parameters
free parameters (1)
- Annotation weights w_i,a =
equal weights (1/|A|) in all experiments
assumptions (6)
- domain assumption Behavior policy pi_b and target policy pi_e are known, so propensity ratios rho are known.
- domain assumption The reward-model training dataset D_Rhat is independent of the OPE evaluation dataset D0.
- domain assumption Common support: pi_e(a|s) > 0 implies pi_b(a|s) > 0 (Assumption 4) or pi_b_plus(a|s) > 0 (Assumption 5).
- domain assumption Annotations are drawn from a distribution G(s,a) with expectation R_bar + eps_G (Assumption 2) and variance sigma_R^2 + Delta_G (Assumption 3).
- domain assumption Misspecified reward model: for the function class F, min_f ||f - R*||_D > 0 (Assumption 6).
- domain assumption Finite discrete context and action spaces with rewards of finite variance.
Cite this review
Pith. "Pith review of CANDOR: Counterfactual ANnotated DOubly Robust Off-Policy Evaluation." pith.science (2026). https://pith.science/paper/IUQZOT57
@misc{pith2026241208052,
author = {Pith},
title = {Pith review of: CANDOR: Counterfactual ANnotated DOubly Robust Off-Policy Evaluation},
year = {2026},
howpublished = {\url{https://pith.science/paper/IUQZOT57}},
note = {Machine review of arXiv:2412.08052}
}
read the original abstract
Off-policy evaluation (OPE) is critical for applying contextual bandit algorithms to high-stakes decision-making settings such as healthcare, where new treatment policies must be evaluated prior to deployment. Unfortunately, OPE techniques are inherently limited by the breadth of the available data, which may not be sufficient to evaluate the performance of a new policy. Recent work attempts to improve dataset coverage by adding expert-annotated counterfactual samples. However, such annotations are often imperfect and can lead to worse estimator performance than using no annotations at all. To better leverage imperfect annotations, we propose a family of OPE estimators grounded in the doubly robust (DR) framework, which combines importance sampling (IS) with a reward model (direct method, DM) for better statistical guarantees. We study three ways of incorporating counterfactual annotations. Under mild assumptions, we prove that using annotations within just the DM component yields the most desirable theoretical results. Experiments on multiple healthcare tasks, including real-world electronic health records (EHR) data, show that this strategy is most robust under misspecified reward models and inaccurate annotations. By addressing the challenges posed by imperfect annotations, this work broadens the applicability of OPE methods and facilitates safer deployment of decision-making policies in healthcare.
Figures
Figures from the paper (5 more)
Reference graph
Works this paper leans on
-
[1]
A. Beygelzimer and J. Langford. The offset tree for learning with partial labels. In Proceedings of the 15th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, KDD '09, page 129–138, New York, NY, USA, 2009. Association for Computing Machinery. ISBN 9781605584959. doi:10.1145/1557019.1557040. URL https://doi.org/10.1145/1557019.1557040
arXiv 2009
- [2]
-
[3]
M. Dudík, D. Erhan, J. Langford, and L. Li. Doubly robust policy evaluation and optimization. Statistical Science, 29 0 (4), Nov. 2014. ISSN 0883-4237. doi:10.1214/14-sts500. URL http://dx.doi.org/10.1214/14-STS500
-
[4]
M. Farajtabar, Y. Chow, and M. Ghavamzadeh. More robust doubly robust off-policy evaluation, 2018
work page 2018
-
[5]
Q($\lambda$) with Off-Policy Corrections
A. Harutyunyan, M. G. Bellemare, T. Stepleton, and R. Munos. Q( ) with off-policy corrections, 2016. URL https://arxiv.org/abs/1602.04951
work page Pith review arXiv 2016
-
[6]
D. G. Horvitz and D. J. Thompson. A generalization of sampling without replacement from a finite universe. Journal of the American Statistical Association, 47 0 (260): 0 663--685, 1952. ISSN 01621459. URL http://www.jstor.org/stable/2280784
arXiv 1952
-
[7]
N. Jiang and L. Li. Doubly robust off-policy value evaluation for reinforcement learning, 2016
work page 2016
-
[8]
A. S. Lan and R. Baraniuk. A contextual bandits framework for personalized learning action selection. In Educational Data Mining, 2016. URL https://api.semanticscholar.org/CorpusID:15394680
work page 2016
Show all 21 references
-
[9]
H. M. Le, C. Voloshin, and Y. Yue. Batch policy learning under constraints, 2019. URL https://arxiv.org/abs/1903.08738
2019 arXiv
-
[10]
L. Li, W. Chu, J. Langford, and R. E. Schapire. A contextual-bandit approach to personalized news article recommendation. In Proceedings of the 19th international conference on World wide web, WWW ’10. ACM, Apr. 2010. doi:10.1145/1772690.1772758. URL http://dx.doi.org/10.1145/...
2010
-
[11]
V. Li. Pre/dicta takes a radically different approach to predictive analytics than others, 05 2023
2023
-
[12]
Liu and S
S. Liu and S. Zhang. Efficient policy evaluation with offline data informed behavior policy design, 2024. URL https://arxiv.org/abs/2301.13734
2024 arXiv
-
[13]
Mandyam, M
A. Mandyam, M. Jörke, W. Denton, B. E. Engelhardt, and E. Brunskill. Adaptive interventions with user-defined goals for health behavior change, 2024
2024
-
[14]
Oberst and D
M. Oberst and D. Sontag. Counterfactual off-policy evaluation with gumbel-max structural causal models, 2019. URL https://arxiv.org/abs/1905.05824
2019 arXiv
-
[15]
Precup, R
D. Precup, R. Sutton, and S. Singh. Eligibility traces for off-policy policy evaluation. Computer Science Department Faculty Publication Series, 06 2000
2000
-
[16]
R. S. Sutton and A. G. Barto. Reinforcement learning: an introduction. Adaptive computation and machine learning series. The MIT Press, second edition edition, 2018. ISBN 9780262039246
2018
-
[17]
Tang and J
S. Tang and J. Wiens. Counterfactual-augmented importance sampling for semi-offline policy evaluation. In Thirty-seventh Conference on Neural Information Processing Systems, 2023. URL https://openreview.net/forum?id=dsH244r9fA
2023
-
[18]
P. S. Thomas and E. Brunskill. Data-efficient off-policy policy evaluation for reinforcement learning, 2016
2016
-
[19]
van Seijen, H
H. van Seijen, H. V. Hasselt, S. Whiteson, and M. A. Wiering. A theoretical and empirical analysis of expected sarsa. 2009 IEEE Symposium on Adaptive Dynamic Programming and Reinforcement Learning, pages 177--184, 2009. URL https://api.semanticscholar.org/CorpusID:6230754
2009
-
[20]
Voloshin, H
C. Voloshin, H. M. Le, N. Jiang, and Y. Yue. Empirical study of off-policy policy evaluation for reinforcement learning, 2021
2021
-
[21]
J. Yao, E. Brunskill, W. Pan, S. Murphy, and F. Doshi-Velez. Power constrained bandits. In Proceedings of the 6th Machine Learning for Healthcare Conference, pages 209--259, 2021
2021
Reviewed August 11, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.