Paper: 2607.05375 Authors: Lars van der Laan, Nathan Kallus Categories: stat.ML, cs.LG
The Gap
Offline policy evaluation (OPE) answers a deceptively simple question: if we had run a different policy, what would the total reward be? The dominant toolkit relies on occupancy ratios—the density ratio between the target policy’s state-action distribution and the behavior policy’s distribution. Primal-dual and minimax methods (e.g., dual doubly robust estimators, minimax weight learning) estimate these ratios by enforcing occupancy-balance moments over a critic class. The catch: their convergence guarantees require Bellman completeness—the assumption that applying the Bellman operator to any function in your value-function class stays inside that class. This is a strong structural condition on the hypothesis space that rarely holds in practice and is essentially impossible to verify.
The authors observe that existing fitted Q-evaluation analyses also lean on Bellman completeness or its cousin, projected-operator stability. The gap is clean: can we do offline policy evaluation with just realizability—the assumption that our hypothesis class contains the thing we’re looking for—without demanding closure under Bellman operators?
Problem: Offline policy evaluation needs occupancy ratios
|
v
Prior methods: primal-dual / minimax -> enforce balance moments
|
v
Requirement: Bellman completeness on value-function class
| \
| \ (rarely holds, hard to verify)
v v
Gap: Can we drop Bellman completeness?
|
v
FORE approach: adjoint Bellman recursion on ratios
|
v
Only need: realizability of the occupancy ratio itself
|
v
Evidence: KL-contraction + finite-sample regret bounds
|
v
Conclusion: Ratio realizability suffices for OPE
The Increment
One sentence: Before this paper, offline policy evaluation methods needed Bellman completeness (a structural closure condition on function classes); after this paper, we know that merely assuming the ratio is in your hypothesis class is sufficient.
Core Mechanism
FORE works with the discounted occupancy ratio (w(s,a)), which measures how much more (or less) likely the target policy visits a state-action pair compared to the behavior policy, accounting for temporal discounting. Instead of estimating this ratio indirectly through value functions and balance constraints, FORE directly characterizes (w) through an adjoint Bellman recursion: the ratio at the current timestep equals the immediate importance weight plus the discounted expected ratio at the next timestep, averaged over the behavior policy’s transitions.
At each iteration, FORE solves a single-level KL-projection problem: given the current ratio estimate, apply the adjoint Bellman operator to get a target, then project that target back onto the log-ratio hypothesis class by minimizing KL divergence. This is a simple supervised regression on one-step transition data—no inner optimization loop, no adversarial training. The key mathematical insight is that the adjoint Bellman operator is a contraction in KL divergence (relative entropy), so repeated projection converges to the fixed point, which is the true ratio.
The fitted ratio then supports three downstream estimation strategies: direct reward reweighting (weight observed rewards by the ratio), occupancy-weighted fitted Q-evaluation (use the ratio to reweight a fitted Q-function), and doubly robust estimation (combine the ratio with a Q-function for variance reduction).
Iteration t:
w_t(s,a) -----> Adjoint Bellman Operator -----> T[w_t](s,a)
(current ratio) (one-step transition) (target)
|
v
KL Projection onto
log-ratio class H
|
v
w_{t+1}(s,a) = argmin_{h in H} KL(T[w_t] || h)
(next ratio estimate)
|
v
Repeat until convergence
(KL contraction guarantees fixed point)
Structural metaphor: Calibrating a bathroom scale with known weights.
Imagine you have a bathroom scale that might be miscalibrated. You want to find the correction factor (how much to multiply every reading by to get the true weight) for every possible weight on the scale. The adjoint Bellman recursion is like this: if you know the correction factor at 150 lbs and you know the physical relationship between adjacent readings, you can infer what the correction factor should be at 149 lbs and 151 lbs. Each iteration is like placing a known weight on the scale, computing what the corrected reading *should be, then turning the calibration knob to minimize the gap between your current correction curve and the target. The KL projection is your “turning the knob” step—it adjusts your correction function to be as close as possible to the target in a relative-entropy sense. The contraction property means each calibration step gets you strictly closer to the true correction curve. You don’t need to know anything about the scale’s internal mechanism (no Bellman completeness); you just need your correction curve to be flexible enough to represent the true one (realizability). After enough rounds, the scale is calibrated.
Key Concepts
-
Occupancy ratio: Suppose you’re a doctor who treated patients under guideline A, but you want to know what would have happened under guideline B. The occupancy ratio is, for each patient profile, how much more likely guideline B would have produced that profile compared to guideline A. If the ratio is 3 for young diabetic patients, it means guideline B “targets” that population three times more. Multiplying observed rewards by this ratio reweights your observed data to look like it came from guideline B—without actually running guideline B.
-
Bellman completeness: This is the assumption that if you take any value function from your hypothesis class and apply one step of the Bellman equation (compute the one-step backup), the result is still representable in the same class. It’s like saying: if your spreadsheet can represent any polynomial of degree (\leq 3), and the operation “multiply by (x) and add 1” always produces another polynomial of degree (\leq 3), you’re complete. In practice, this almost never holds—the backup pushes functions outside the class, creating approximation errors that compound. This paper avoids needing this entirely.
-
Adjoint Bellman operator: The standard Bellman operator propagates *values forward in time (what’s the value of this state given future values?). The adjoint propagates importance weights backward—how much does the current state-action pair matter for the overall distribution mismatch? Think of it as the difference between tracking “where does the river flow?” (forward/standard) versus “where did the water in this bucket come from?” (backward/adjoint).
Framework Shift
Before (mainstream approach): After (this paper):
Value class V Ratio class W
| |
v v
Bellman operator T_V Adjoint Bellman T_W
| |
v v
Balance constraints KL projection
(enforce T_V[V] in V) (project T_W[w] onto W)
| |
v v
Requires: V closed under T_V Requires: W contains true w
(Bellman completeness) (realizability only)
| |
v v
Optimization: minimax / Optimization: single-level
primal-dual (hard) regression (easy)
From value-function completeness to ratio-function realizability, the core shift is moving the locus of approximation from the space where closure is demanded to a space where mere containment suffices.
Expert Assessment
Problem choice: This is a genuine and well-posed gap. Bellman completeness has been the Achilles’ heel of offline RL theory for years—it’s the assumption everyone makes and nobody likes. The occupancy-ratio angle is not entirely new (dual importance-weight estimators exist), but framing the problem as a fixed-point iteration on ratios rather than values is a clean conceptual move. This sits at a productive intersection of density ratio estimation, operator theory, and offline RL.
Method maturity: The approach is clever rather than brute force. The adjoint Bellman recursion as a KL-contraction is elegant and the single-level projection avoids the nested optimization headaches of primal-dual methods. However, the gap between the population-level contraction argument and the finite-sample bound involves approximation error terms that could be large in practice. The method inherits the usual curse of horizon through the discount structure—long horizons mean the ratio can become extreme, and KL projection on extreme ratios is numerically fragile.
Experimental integrity: This is a theory paper—the contribution is the theoretical framework and the finite-sample bounds, not empirical benchmarks. The absence of experiments is appropriate for the contribution, though it does mean we lack evidence on whether the KL projection step is practical at scale or whether the ratio class realizability assumption is more or less restrictive than Bellman completeness in practice.
Writing quality: The paper is technically dense but well-structured. The connection between the adjoint Bellman operator and KL-contraction is the conceptual core, and the authors explain it clearly. Where they cut corners: the discussion of *why ratio realizability is easier to satisfy than Bellman completeness is thin—more concrete examples (even toy MDPs) showing the difference would strengthen the narrative considerably. Section 4 on downstream estimators could also benefit from a running example.
Verdict: weak accept — The theoretical contribution is clean and fills a real gap, but the absence of any empirical grounding or concrete guidance on choosing ratio classes leaves practitioners without a clear path to adoption.
Takeaways
-
Reframe value estimation as ratio estimation. If you’re working on off-policy evaluation and hitting completeness walls, consider whether the object you’re estimating can be switched from a value function to a density ratio. Ratios often have simpler structure and weaker approximation requirements.
-
KL-contraction as a design principle. When designing iterative procedures for distribution-related quantities, check whether your operator contracts in KL (or another f-divergence). If it does, you can use single-level projection instead of adversarial training—the analysis becomes cleaner and the optimization easier.
-
The adjoint perspective on Bellman equations. The idea of studying the *adjoint of a forward operator—propagating weights backward rather than values forward—is a general technique that applies beyond RL. Anytime you have a Markov chain and want to understand distribution shift, the adjoint recursion gives you a direct handle on the density ratio without going through the value function.
论文: 2607.05375 作者: Lars van der Laan, Nathan Kallus 分类: stat.ML, cs.LG
缺口
离线策略评估(OPE)要回答一个看起来简单的问题:如果我们执行的是另一条策略,总回报会是多少?主流工具箱依赖占用率比——目标策略的状态-动作分布与行为策略分布之间的密度比。原始对偶方法和极小极大方法(比如对偶双重稳健估计器、极小加权学习)通过在评论器类上强制占用率平衡矩来估计这些比率。问题在于:它们的收敛保证需要贝尔曼完备性——即假设对值函数类中的任何函数施加贝尔曼算子后,结果仍然留在该类中。这是一个对假设空间很强的结构性条件,实践中几乎不成立,而且基本无法验证。
作者注意到,现有的拟合 Q 评估分析同样依赖贝尔曼完备性或其近亲——投影算子稳定性。缺口很清晰:能否在只保留可实现性假设(假设假设类包含我们寻找的目标函数)的情况下完成离线策略评估,而不要求在贝尔曼算子下封闭?
问题:离线策略评估需要占用率比
|
v
此前方法:原始对偶 / 极小极大 -> 强制平衡矩
|
v
要求:值函数类上的贝尔曼完备性
| \
| \ (实践中很少成立,难以验证)
v v
缺口:能否去掉贝尔曼完备性?
|
v
FORE 方案:对比占用率的伴随贝尔曼递推
|
v
仅需:占用率比本身的可实现性
|
v
证据:KL 收缩 + 有限样本遗憾界
|
v
结论:比的可实现性足以支撑离线策略评估
增量
一句话: 在这篇论文之前,离线策略评估需要贝尔曼完备性(函数类的结构性封闭条件);之后我们知道,只假设比率在假设类中就已足够。
核心机制
FORE 直接操作折扣占用率比 (w(s,a)),它衡量目标策略访问某个状态-动作对的概率相对于行为策略有多大差异(考虑时间折扣)。与通过值函数和平衡约束间接估计不同,FORE 通过伴随贝尔曼递推直接刻画 (w):当前时间步的比率等于即时重要性权重加上下一时间步比率的折扣期望(在行为策略的转移分布上取平均)。
每次迭代中,FORE 求解一个单层 KL 投影问题:给定当前比率估计,施加伴随贝尔曼算子得到目标,然后通过最小化 KL 散度将该目标投影回对数比率假设类。这是在单步转移数据上的简单监督回归——没有内层优化循环,没有对抗训练。关键的数学洞察是:伴随贝尔曼算子在 KL 散度(相对熵)意义下是收缩的,因此反复投影会收敛到不动点,即真实比率。
拟得的比率随后支持三种下游估计策略:直接奖励重加权(用比率加权观测奖励)、占用率加权拟合 Q 评估(用比率加权拟合 Q 函数)、双重稳健估计(结合比率与 Q 函数以降低方差)。
第 t 次迭代:
w_t(s,a) -----> 伴随贝尔曼算子 -----> T[w_t](s,a)
(当前比率) (单步转移) (目标)
|
v
KL 投影到
对数比率类 H
|
v
w_{t+1}(s,a) = argmin_{h in H} KL(T[w_t] || h)
(下一步比率估计)
|
v
重复直到收敛
(KL 收缩保证到达不动点)
核喻:用已知重量校准体重秤。
想象你有一台可能不准的体重秤。你想找到校正因子——对秤上每个读数乘以多少才能得到真实重量。伴随贝尔曼递推就像这样:如果你知道 150 斤读数的校正因子,并且你知道相邻读数之间的物理关系,你就能推断出 149 斤和 151 斤的校正因子应该是多少。每次迭代相当于放一个已知重量在秤上,算出校正后的读数**应该*是多少,然后转动校准旋钮,最小化当前校正曲线和目标之间的差距。KL 投影就是”转旋钮”这一步——它在相对熵意义下把你的校正函数调得尽可能接近目标。收缩性质意味着每次校准都让你严格更接近真实校正曲线。你不需要知道秤的内部结构(无需贝尔曼完备性);你只需要你的校正曲线足够灵活,能表示出真实的那条(可实现性)。经过足够多轮,秤就校准好了。
关键概念
-
占用率比: 假设你是一名医生,按指南 A 治疗了病人,但你想知道按指南 B 会怎样。占用率比就是,对每种病人特征组合,指南 B 产生该组合的概率是指南 A 的多少倍。如果对年轻糖尿病患者这个比值是 3,说明指南 B 对这个人群的”覆盖量”是三倍。将观测到的奖励乘以这个比值,就能把你的观测数据重加权成看起来像指南 B 产生的数据——而无需真正执行指南 B。
-
贝尔曼完备性: 这个假设是说,如果你从假设类中取任意一个值函数,施加一步贝尔曼方程(计算单步备份),结果仍然可以在这个类中表示。打个比方:如果你的模型能表示所有次数 (\leq 3) 的多项式,而”乘以 (x) 再加 1”这个操作总是产生另一个次数 (\leq 3) 的多项式,那你就是完备的。实践中这几乎不成立——备份会把函数推出类外,产生会累积的近似误差。本文完全绕开了这个要求。
-
伴随贝尔曼算子: 标准贝尔曼算子在时间上向前传播**值*(这个状态的值是多少,给定未来值?)。伴随算子向后传播重要性权重——当前状态-动作对对整体分布失配有多大影响?可以这样理解区别:正向/标准算子追踪”河水流向哪里?“,伴随算子追踪”这桶水从哪里来的?“。
框架转变
之前(主流方法): 之后(本文方法):
值函数类 V 比率函数类 W
| |
v v
贝尔曼算子 T_V 伴随贝尔曼算子 T_W
| |
v v
平衡约束 KL 投影
(强制 T_V[V] 包含于 V) (将 T_W[w] 投影到 W 上)
| |
v v
要求:V 在 T_V 下封闭 要求:W 包含真实 w
(贝尔曼完备性) (仅需可实现性)
| |
v v
优化:极小极大 / 优化:单层
原始对偶(难) 回归(易)
从值函数完备性到比率函数可实现性,核心转变是把近似误差的重担从一个要求封闭的空间转移到了一个只要求包含的空间。
专家评审
选题眼光: 这是一个真实且定义清晰的缺口。贝尔曼完备性一直是离线 RL 理论的阿喀琉斯之踵——所有人都在用它,没人喜欢它。占用率比这个角度不算全新(对偶重要性权重估计器早已有之),但把问题框架从值上的不动点迭代转到比率上的不动点迭代,是一个干净的概念性推进。这处在密度比估计、算子理论和离线 RL 的交汇处,位置很理想。
方法成熟度: 方法靠的是巧劲而非蛮力。伴随贝尔曼算子作为 KL 收缩算子的论证很优雅,单层投影避免了原始对偶方法的嵌套优化之痛。但人口级别的收缩论证和有限样本界之间的近似误差项在实践中可能很大。方法通过折扣结构继承了标准的视野诅咒——长视野意味着比率可能变得极端,而在极端比率上做 KL 投影在数值上很脆弱。
实验诚意: 这是一篇理论论文——贡献在于理论框架和有限样本界,不在于实证基准。没有实验对这个贡献来说是合适的,但这也意味着我们缺乏证据来判断 KL 投影步骤在大规模下是否实用,或者比率类可实现性假设在实践中是否比贝尔曼完备性更容易满足。
写作功力: 论文技术密度高但结构清晰。伴随贝尔曼算子和 KL 收缩之间的联系是概念核心,作者解释得很清楚。偷懒的地方:对”为什么比率可实现性比贝尔曼完备性更容易满足”的讨论太少——更多的具体例子(哪怕是玩具 MDP)能大幅强化叙事。第四节关于下游估计器的部分也需要一个贯穿始终的示例。
判决: 弱接收 — 理论贡献干净且填补了真实缺口,但缺乏任何实证基础或关于如何选择比率类的具体指导,实践者缺乏清晰的落地路径。
要点总结
-
把值估计重构为比率估计。 如果你在做离线策略评估时碰到了完备性的天花板,考虑一下你估计的对象能否从值函数切换到密度比。比率通常具有更简单的结构和更弱的近似要求。
-
KL 收缩作为设计原则。 设计与分布相关的迭代过程时,检查你的算子是否在 KL(或其他 f 散度)意义下收缩。如果是,你就可以用单层投影代替对抗训练——分析更干净,优化更简单。
-
贝尔曼方程的伴随视角。 研究正向算子的**伴随*——向后传播权重而非向前传播值——是一个通用技术,超越 RL 本身。任何时候你有马尔可夫链且想理解分布偏移,伴随递推都能让你直接掌握密度比,无需绕道值函数。