REVIEW 3 major objections 4 minor 14 references
Behavioral Entropy-Guided Dataset Generation for Offline Reinforcement Learning
T0 review · 3 major / 4 minor · reviewed 2026-08-08 · deepseek-v4-flash
Pith's one-line read The paper claims that offline RL algorithms trained on datasets collected by a behavioral-entropy-maximizing policy outperform those trained on datasets from Shannon entropy, Rényi entropy, SMM, and RND on all five tasks considered.
desk verdict Solid empirical comparison, but the central claim that the reward actually maximizes behavioral entropy rests on an unjustified approximation, so as written the experiments compare a heuristic. 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 object is differential behavioral entropy, $H_{B,\alpha,\beta}(f) = \beta \int e^{-\beta(-\log f(x))^\alpha} (-\log f(x))^\alpha dx$, which is Shannon entropy evaluated under Prelec's probability weighting $w(x)=e^{-\beta(-\log x)^\alpha}$. The estimator is the importance-sampling-corrected k-NN plug-in, and the load-bearing simplification is the reduction from the full estimator to the reward $r(s,a) = \|s - NN_k(s)\|_2 e^{-\beta(\log(\|s-NN_k(s)\|_2 + c))^\alpha} (\log(\|s-NN_k(s)\|_2 + c))^\alpha$ with $d=1$; maximizing this reward is claimed to approximately maximize BE of the state occupancy measure.
What would settle it
Compute the omitted constant $D_{k,n}$ for the experiment's settings ($d=1$, $n=500{,}000$, $k=12$) and compare it with the distribution of $\log \|s - NN_k(s)\|_2$ in a generated dataset; if the constant is comparable to or larger than the typical log-distance, the approximation in Eq. (22) is not negligible, and a direct comparison of Eq. (21) and Eq. (24) as policy-ranking objectives would show whether the experimental results hold for the true BE estimator.
Extended reading notes
Core claim
Behavioral entropy in continuous spaces is defined by applying Prelec's probability weighting $w(x)=e^{-\beta(-\log x)^\alpha}$ to the density and then computing Shannon entropy of the weighted density. The paper derives a k-NN plug-in estimator for this quantity, corrected by importance sampling, and proves it converges and has controlled bias and variance. From the estimator it extracts a per-step reward that is approximately proportional to BE improvement, so that any RL algorithm trained on that reward approximately maximizes BE of the state occupancy measure. In experiments, policies trained on the BE reward collected 500K-step datasets for Walker and Quadruped, and offline RL on those datasets outperformed baselines in the majority of task-algorithm combinations. The paper's stated conclusion is that BE-generated datasets lead to superior offline RL performance and better data efficiency compared with Shannon, Rényi, SMM, and RND objectives.
Load-bearing premise
The reward used in the experiments drops a constant $D_{k,n}$ that depends on $n$, $k$, and the dimension $d$, and the central claim depends on this drop being negligible, since otherwise the policies are maximizing a heuristic derived from behavioral entropy rather than behavioral entropy itself.
Editorial extensions
If this is right
- If the central claim holds, offline RL practitioners can replace hand-tuned exploration rewards with a single BE reward and get better downstream task performance on the same computational budget.
- The k-NN estimator's consistency and bias/variance guarantees mean BE can be used in high-dimensional state representations if the representation dimension is kept moderate.
- BE's smooth variation in $\alpha$ gives dataset designers a knob that interpolates between broad and focused coverage, something the paper shows Rényi entropy does not offer stably for $q > 1$.
- The paper shows comparable downstream performance with 500K-element datasets to that of prior work using 10M-element datasets, suggesting BE collection reduces both data and training requirements.
Reading between the lines
- Editorial inference: the approximation dropping $D_{k,n}$ deserves scrutiny; if it is not negligible, the paper's experiments should be read as evidence for a BE-inspired heuristic reward rather than for BE itself, and the gap between Eq. (21) and Eq. (24) is a direct way to test this.
- Editorial inference: because the paper notes the extension to state-action occupancy measures is straightforward, the same machinery could generate datasets targeting action diversity, which may matter for tasks like manipulation where multiple actions reach the same state.
- Editorial inference: the smooth dependence of coverage on $\alpha$ suggests an automatic schedule of $\alpha$ during data collection could produce a single dataset with both coarse and fine coverage, something the paper does not explore.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper extends behavioral entropy (BE), previously defined for discrete distributions, to continuous state spaces. It proposes k-nearest-neighbor estimators for differential BE, states consistency and bias-variance guarantees (Theorems 1 and 2), and derives a practical reward function intended to induce BE-maximizing policies. These policies are then used with the APT algorithm to generate datasets for offline RL in Walker and Quadruped MuJoCo environments, and the datasets are compared against those generated with Rényi entropy, Shannon entropy, RND, and SMM. The headline empirical result is that offline RL algorithms trained on BE-generated datasets outperform all baselines on five tasks, and outperform Rényi-generated datasets on four of the five tasks.
Significance. If the connection between the implemented reward and the BE objective were established, the paper would offer a principled and flexible exploration objective for dataset generation, with substantial empirical support. The experimental effort is a real strength: the authors train 1,275 offline RL policies, report clear aggregate results in Table 1, and include ablations and coverage visualizations. The theoretical results for the importance-sampling k-NN estimator in Theorems 1 and 2 are also valuable as a first step for BE estimation in continuous spaces. However, the central derivation linking the practical reward in Eq. (24) to behavioral-entropy maximization contains an unverified approximation and a dimension mismatch. Because that link is the stated reason the generated datasets should reflect BE coverage, the significance of the empirical comparison is currently conditional on a point that the paper does not validate.
major comments (3)
- [§4, Eqs. (21)–(22)] The step from Eq. (21) to Eq. (22) replaces d log R_{i,k,n} + D_{k,n} with d log R_{i,k,n} in both the exponent and the power, on the assertion that the contribution of D_{k,n} is negligible. This step is load-bearing: it is the only derivation connecting the trained policies to the stated objective H^{B,α,β}(d_π). In the experimental setup n = 500,000 and k = 12, and with the d = 1 used in Eq. (24), D_{k,n} = log(nπ^{1/2}/(kΓ(3/2))) ≈ 11.9. This is not small compared with the per-sample log-distance terms, which are of order zero or negative in typical normalized feature spaces. Dropping D can change the sign of the argument of the α-power, and for non-integer α the expression in Eq. (21) is not even defined when d log R_{i,k,n} + D_{k,n} < 0. Theorems 1 and 2 are asymptotic consistency and bias results for Eq. (13), not for the approximation in Eq. (22), and no numerical check is reported to show that Eq. (22) tracks Eq. (21) on actual state samples. Without such a check, Eq. (24) is a heuristic exploration bonus, and the abstract's causal attribution of the Table 1 results to behavioral-entropy maximization is not established.
- [§4, Eqs. (23)–(24), and Table 2] Eq. (23) carries the dimension d explicitly in the exponent and in the log-power, but Eq. (24) sets d = 1, while Table 2 reports that the APT representation dimension is 512. This is not a cosmetic numerical-stability change: moving from a d-dimensional log distance to a one-dimensional one changes the reward surface by a factor of d inside the exponent and inside the log-power, and the additive constant c in Eq. (24) introduces a further modification that has no counterpart in the estimator of Eq. (21). The paper should either derive Eq. (24) from the d-dimensional k-NN estimator without this substitution, or provide empirical evidence that the ranking of policies induced by Eq. (24) matches the ranking induced by Eq. (21) on samples from the relevant feature distribution. As it stands, the implemented reward is not shown to be approximately proportional to the BE estimator it is claimed to encode.
- [Appendix A.1, Lemma 1] Lemma 1 as stated bounds |E[H_n^{B,w}(f)] − H^{B,w}(f)| by O((k/n)^{ξ/d}) for the estimator H_n^{B,w} defined in Eq. (26), which is a sample average of g(f(X_i)) and is exactly unbiased for H^{B,w}(f). The stated bound is therefore vacuous, and the sentence attributing it to Singh & Póczos, Theorem 5 is misleading because Eq. (26) contains no k-NN density estimate at all. The proof of Theorem 2 does not actually need this bound—the first term in Eq. (33) is controlled through Lemma 2 and the second term is zero—but the supporting lemma should be corrected, redefined in terms of the plug-in estimator, or removed so that the theoretical contribution is stated accurately.
minor comments (4)
- [Appendix A.1, Eqs. (25)–(27)] The displayed equations are garbled: Eq. (25) appears to combine two separate equalities, and Eq. (26) uses "1/f(x)" where the argument should be X_i. Please rewrite these definitions so that the integrand and the sample average are unambiguous.
- [§5, Dataset Generation and Visualization] The sentence "Altogether we considered 17 datasets: eight BE, five BE, two SE, and one each for RND and SMM" should read "eight BE, five RE, two SE, and one each for RND and SMM"; as written it mislabels the Rényi datasets.
- [§5, Experimental Setup] The comparison with the 10M-element ExORL datasets is reported as evidence of data- and sample-efficiency, but the evaluation protocols, environment versions, and offline training budgets may differ from those used to produce the ExORL numbers. A short paragraph listing the exact source of each reference number and any differences in evaluation would make this comparison reproducible.
- [Appendix A.6, Figure 13] The reward-function visualization would be more informative if it showed the behavior of Eq. (24) as a function of the raw distance for the actual values of α and β used in the experiments, rather than only for the α range shown; this would also help readers gauge the effect of the additive constant c.
Circularity Check
No significant circularity: the BE reward is derived from a BE estimator and tested against external baselines; the Eq. 21 to Eq. 22 approximation is a validity concern, not a circular reduction.
full rationale
The derivation chain is not circular. Behavioral entropy is defined independently in Eq. 8 from Prelec's weighting function, and the k-NN estimator in Eq. 13 is a plug-in estimator of that definition whose consistency is anchored to external results (Devroye and Wagner 1977; Singh and Poczos 2016; Zhao and Lai 2022). The practical reward in Eq. 24 is derived from the estimator, not fitted to the experimental outcomes, and the experimental comparisons are against external baselines (Shannon, Renyi, SMM, RND) in standard MuJoCo environments. The claim that BE-generated datasets outperform these baselines is an empirical downstream result, not a quantity defined in terms of the input. The self-citation to Suresh et al. (2024) for the beta conditioning is load-bearing for the theoretical admissibility claim, but that prior work is a published, independently checkable result rather than an unverified premise imported solely to force the present conclusion. The main weakness is the approximation in Eq. 22 and the implementation choice d=1 in Eq. 24: these threaten the fidelity of the reward to the stated BE objective, as does the paper's own admission that the volumetric-coverage relationships are not conclusive. However, a broken or heuristic approximation is a correctness risk, not a circular reduction, because the claimed 'prediction' is not equivalent to any fitted parameter or to the estimator input by construction. No step in the paper defines the output in terms of the claimed input or renames a known result as a derivation. The score is therefore 0.
Assumptions & free parameters
free parameters (5)
- alpha (α) =
grid {0.2, 0.5, 0.7, 0.9, 1.5, 2.0, 3.0, 5.0}; best per task selected for headline
- M (representation dimension for beta conditioning) =
512
- k (k-NN neighbors) =
12
- c (additive constant in Eq 24) =
not reported
- q (Rényi order) for baseline =
grid {0.2, ..., 1.1} for offline RL; q greater than 1 excluded
assumptions (4)
- standard math k-NN density estimator consistency (Devroye-Wagner; Zhao-Lai sup-norm bounds)
- domain assumption The state occupancy measure d_pi has a density that is bounded away from zero on a compact support
- ad hoc to paper D_{k,n} is negligible in the reward derivation (Eq 21 to Eq 22)
- domain assumption Beta conditioning from Suresh et al. 2024 makes BE an admissible entropy
Cite this review
Pith. "Pith review of Behavioral Entropy-Guided Dataset Generation for Offline Reinforcement Learning." pith.science (2026). https://pith.science/paper/2IEMHEDG
@misc{pith2026250204141,
author = {Pith},
title = {Pith review of: Behavioral Entropy-Guided Dataset Generation for Offline Reinforcement Learning},
year = {2026},
howpublished = {\url{https://pith.science/paper/2IEMHEDG}},
note = {Machine review of arXiv:2502.04141}
}
abstract
Entropy-based objectives are widely used to perform state space exploration in reinforcement learning (RL) and dataset generation for offline RL. Behavioral entropy (BE), a rigorous generalization of classical entropies that incorporates cognitive and perceptual biases of agents, was recently proposed for discrete settings and shown to be a promising metric for robotic exploration problems. In this work, we propose using BE as a principled exploration objective for systematically generating datasets that provide diverse state space coverage in complex, continuous, potentially high-dimensional domains. To achieve this, we extend the notion of BE to continuous settings, derive tractable $k$-nearest neighbor estimators, provide theoretical guarantees for these estimators, and develop practical reward functions that can be used with standard RL methods to learn BE-maximizing policies. Using standard MuJoCo environments, we experimentally compare the performance of offline RL algorithms for a variety of downstream tasks on datasets generated using BE, R\'{e}nyi, and Shannon entropy-maximizing policies, as well as the SMM and RND algorithms. We find that offline RL algorithms trained on datasets collected using BE outperform those trained on datasets collected using Shannon entropy, SMM, and RND on all tasks considered, and on 80% of the tasks compared to datasets collected using R\'{e}nyi entropy.
Figures
Figures from the paper (10 more)
Reference graph
Works this paper leans on
-
[2]
(31) 13 Published as a conference paper at ICLR 2025 We are now in a position to prove our main result. Theorem
work page 2025
-
[7]
Don’t change the algorithm, change the data: Exploratory data for offline reinforcement learning
Denis Yarats, David Brandfonbrener, Hao Liu, Michael Laskin, Pieter Abbeel, Alessandro Lazaric, and Lerrel Pinto. Don’t change the algorithm, change the data: Exploratory data for offline reinforcement learning. In ICLR 2022 Workshop on Generalizable Policy Learning in Physical World,
work page 2022
-
[9]
Let µ denote the Lebesgue measure and Br(x) = {x′ ∈ Rd | ∥x′ − x∥2 < r}. Define H B,w (f ) = − Z X log(w(f (x)))w(f (x))dx Z X g(f (x))f (x)dx, (25) H B,w n (f ) = − nX i=1 1 f (x) log(w(f (Xi)))w(f (Xi)) = 1 n nX i=1 g(f (Xi)), (26) bH B,w k,n (f ) = − nX i=1 1 ˆf (x) log(w( ˆf (Xi)))w( ˆf (Xi)) = 1 n nX i=1 g( ˆf (Xi)). (27) Our goal is to establish a b...
work page 2003
-
[11]
We first recall another useful result from the literature. For a given set S ⊂ X, radius r, and m > 0, let N (S, r) denote the covering number, the minimum number of balls of radius r needed to cover S. Let ∥·∥op denote the operator norm. Lemma 2 ((Zhao & Lai, 2022)). Suppose there exist C1, C2, C3, N0 > 0 and β ∈ (0, 1] such that the following conditions...
work page 2022
-
[13]
(32) Proof. First notice that E h bH B,w k,n (f ) i − H B,w (f ) ≤ E h bH B,w k,n (f ) − H B,w n (f ) i + E H B,w n (f ) − H B,w (f ) . (33) The second term can be bounded using Lemma 1, so it just remains to bound the first term. Recall that X is compact, f is bounded strictly away from 0 on X , and w is Lipschitz. We therefore have that g is the product...
work page 2025
-
[28]
Lemma 1 ((Singh & P ´oczos, 2016))
We first recall the following result. Lemma 1 ((Singh & P ´oczos, 2016)). Suppose that, for some ξ ∈ (0, 2], f is ξ-H¨older continuous and strictly positive on X . Suppose furthermore that there exists a function f∗ : X →R+ and a constant f ∗ such that 0 < f∗(x) ≤ R Br(x) f (y)dy/µ(Br(x)) ≤ f ∗ < ∞, for all x ∈ X, r∈ (0, √ d], and assume that R ∞ 0 e−xxkf...
work page 2016
-
[512]
→ 1024 → dim(A) ReLU MLP representation dimension 512 k in NN approximator 12 average top k in NN True RND hyperparameter representation dimension 512 predictor, target network architecture dim(S) → 1024 → 1024 → 512 ReLU MLP normalized observation clipping 5 SMM hyperparameter skill dimension 4 skill discriminator learning rate 10−3 V AE learning rate 10...
work page 2025
-
[2008]
Offline reinforcement learning: Tuto- rial, review, and perspectives on open problems
Sergey Levine, Aviral Kumar, George Tucker, and Justin Fu. Offline reinforcement learning: Tuto- rial, review, and perspectives on open problems. arXiv preprint arXiv:2005.01643,
arXiv 2005
Show all 14 references
-
[2016]
Diversity is all you need: Learning skills without a reward function
Benjamin Eysenbach, Abhishek Gupta, Julian Ibarz, and Sergey Levine. Diversity is all you need: Learning skills without a reward function. arXiv preprint arXiv:1802.06070,
-
[2018]
nearest neighbor
Kevin Beyer, Jonathan Goldstein, Raghu Ramakrishnan, and Uri Shaft. When is “nearest neighbor” meaningful? In Database Theory—ICDT’99: 7th International Conference Jerusalem, Israel, January 10–12, 1999 Proceedings 7, pp. 217–235. Springer,
1999
-
[2019]
Curiosity-driven exploration by self-supervised prediction
11 Published as a conference paper at ICLR 2025 Deepak Pathak, Pulkit Agrawal, Alexei A Efros, and Trevor Darrell. Curiosity-driven exploration by self-supervised prediction. In International conference on machine learning, pp. 2778–2787. PMLR,
2025
-
[2020]
Urlb: Unsupervised reinforcement learning benchmark
Michael Laskin, Denis Yarats, Hao Liu, Kimin Lee, Albert Zhan, Kevin Lu, Catherine Cang, Lerrel Pinto, and Pieter Abbeel. Urlb: Unsupervised reinforcement learning benchmark. arXiv preprint arXiv:2110.15191,
-
[2021]
Efficient exploration via state marginal matching
Lisa Lee, Benjamin Eysenbach, Emilio Parisotto, Eric Xing, Sergey Levine, and Ruslan Salakhutdi- nov. Efficient exploration via state marginal matching. arXiv preprint arXiv:1906.05274,
1906 arXiv
-
[2022]
Fix a p.d.f
12 Published as a conference paper at ICLR 2025 A A PPENDIX A.1 P ROOFS Fix a probability weighting function w and let g(y) = − 1 y log(w(y))w(y). Fix a p.d.f. f ∈ ∆(X ), where X ⊂Rd is compact. Fix n, k∈ N, and let X1, . . . , Xn ∼ f (·). Recall the definition of ˆf from equation
2025
Reviewed August 8, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.