Response to the major-revision decision (continuous quotient learning)
Everything below is backed by files in experiments/revision/results/ (new) or the earlier result files; code is at https://github.com/npow/quotient-learning-papers (cql/), artifacts at tag v1 (to be created). Section names refer to the revised main paper; "App." to the appendix (supplement).
Summary of what changed
- Positioning vs. RKHS epsilon-machines, causal-state learning and bisimulation metrics rewritten (Intro, Sec. 2); contribution narrowed to a test-based formulation with a stated finite-sample target; empirical comparison added (Sec. 5, "Head-to-head comparison").
- Theorem 1 scope stated precisely (Sec. 4, "Which guarantee applies to which procedure"); Section 5 experiments re-run under the analysed variant (T) (Sec. 5, "Experiments under the analysed variant (T)").
- Proposition 1 corrected and its proof re-derived (Sec. 4, App. B).
- Table 2 Tiger typo fixed; GaussMix alpha reconciled; a second unsupported claim (silhouette picks B=K on K-cycle) found and withdrawn.
- Partition-level ARI under B-rules that do not use n; faithful W1 and DBC baselines; RKHS / causal-state / Ferns baselines.
- HalfCheetah "multi-step information" claim withdrawn after a held-out, nonlinear-baseline analysis; result files named; repo URL added.
- Several earlier positive claims weakened (listed at the end).
Concern 1
In Section 2 and the Introduction, discuss and empirically compare against RKHS epsilon-machines (MMD-based predictive-state clustering), causal-state learning (arXiv:1906.10437) and continuous bisimulation metrics, and restate the contribution relative to them.
Changed. Intro ("The gap ... and what is not new", "Our contribution (restated)") and Sec. 2 (new paragraphs "Predictive-state clustering with kernels" and "Causal-state learning with recurrent networks"; extended "Bisimulation" paragraph). Cited from sources read for this revision: Brodu & Crutchfield, Chaos 32(2):023103 (2022), arXiv:2011.14821 (their text states that clustering algorithms for conditional distributions map to RKHS "by using the MMD test"; method = regularised conditional mean embeddings, similarity matrix G^S, diffusion-map coordinates, DBSCAN); Jurgens & Brodu arXiv:2410.01076; Zhang et al. arXiv:1906.10437 (RNN trained to predict next observation/reward, discretised by k-means or learned discretiser); Goerg & Shalizi (LICORS, arXiv:1206.2398; Mixed LICORS, AISTATS 2013); Boots, Gordon, Gretton (UAI 2013); Ferns, Panangaden, Precup, SIAM J. Comput. 40(6) (2011). The paper now says explicitly that MMD-based predictive-state clustering is not new; the contribution is a test-based estimator with a stated finite-sample target (bin-level quotient), an analysed variant with a sample bound, and an empirical characterisation, not accuracy superiority.
Empirical comparison (Sec. 5, Table "baselines"; App. C.7; partition_baselines.csv). RKHS epsilon-machine (our implementation of the estimator equations, past = present observation; DBSCAN and k-means variants), GRU causal-state learner (full history), Ferns metric on the binned chain, DBC, multivariate W1. Not compared: the authors' released RKHS code, longer pasts for the RKHS method, LICORS / HSE-PSR (discussed only), the continuous-space Ferns metric (not computable from samples without an abstraction; we use the finite metric on the quantile-bin chain). Result: no method is uniformly best; the GRU baseline is best on K-cycle, the RKHS k-means variant is competitive on well-separated cases, Ferns/W1-silhouette beat ours at B=8 on K4 and lose at K8. The earlier claim that ours is "the only method correct on both" is withdrawn.
Concern 2
In Section 4, either extend Theorem 1 to the implemented procedure or run the Section 5 experiments with variant (T); ... permutation calibration, a data-selected median-heuristic bandwidth, BH dependence across shared-sample tests, and overlapping tau-mixing windows.
Changed (route: run under (T) and state precisely what is covered). Sec. 3 ("Variant (T)") defines (T1)-(T4): bins/bandwidth fitted on a design half, one window per trajectory and bin, independent sides, analytic threshold with alpha0 = delta/K^2, no BH. Implemented in continuous_quotient/variant_t.py (unit tests in tests/unit/test_variant_t.py). Theorem 1 is now stated for (T) only. New Sec. 4 subsection states: what carries over to the implemented procedure (P) (permutation validity gives a no-spurious-split statement if the bandwidth is permutation-invariant; this is only approximately true for the implementation's 500-point fixed-index subsample); that we have no BH/PRDS argument, no treatment of overlapping windows and no power guarantee for (P); hence all positive claims about (P) are empirical. Experiments (Sec. 5, "Experiments under the analysed variant (T)", Table "variantT"; partition_accuracy.csv, tiger_T_scaling.csv, dec_tiger_variant_t.csv): (T) matches (P) at B=n on K-cycle and most GaussMix, but is conservative: n_hat=1 on Tiger at N=3000 (resolved at N=10^4, 5/5 seeds), merges classes at B=8; on Dec-Tiger (T) does not over-split agent A (symmetric N=20000: R_crit 0.264 vs 0.262). An intermediate (P^fix = T's data pipeline with permutation+BH) reproduces (P)'s n_hat on 37/40 comparable runs, attributing (P)'s over-splitting to the calibration. Not done: no extension of Theorem 1 to (P); PO-MuJoCo and the 4-state chain were not re-run under (T) (no ground truth / not needed for the conclusion); N=100000 Tiger scaling was dropped for time.
Concern 3
In Proposition 1, add the conditions B>=n and per-class mass >=2/B (or >=1/B) to the statement, and reconcile the claim of exact purity with the vanishing contamination the Appendix B proof establishes.
Changed. Proposition 1 (Sec. 4) is restated in three parts: (i) under (a) only, |Q_B| <= min{B, 2n-1}, and n <= |Q_B| <= 2n-1 when B >= n and p_i >= 2/B (hence B >= 2n; the 1/B version needs cell alignment (b)); B < n is the counterexample to the old statement; (ii) under (a)+(b), p_i are positive multiples of 1/B (so B >= n follows) and Q*_B is the class partition; (iii) empirical bins are not pure: contamination fraction O_P(B/sqrt N) -> 0, and the estimator returns the partition because the induced MMD is below the threshold. App. B re-derives all parts and states what is not proved; the earlier "every non-empty bin is pure w.p. -> 1" is withdrawn (empirical edges miss the class gap unless F_N(sup I_i) = j_i/B exactly). Note: the (iii) recovery argument is for variant (T) with m_N = Theta(N) i.i.d. samples, as in the proof; it is not claimed for (P).
Concern 4
Resolve the Table 2 Tiger entry n_hat=20 at N=10000 ... and reconcile the GaussMix alpha (0.01 in Section 5 versus 0.05 in Appendix C).
Tiger. Traced: experiments/layer1_ground_truth/results/tiger_N_sensitivity.csv has n_classes = 2 on all five seeds at N=10000 (B=8, so 20 is impossible); "2, 2, 2, 2, 20" was a typo. Table 2 now reads 2, 2, 2, 2, 2 (consistent with App. D). Noted in App. "Artifacts ... Corrections".
GaussMix alpha. The shipped CSV has no alpha column; the script default and App. C say 0.05, Sec. 5 said 0.01. Re-ran the manual-B configuration at both alpha (B=N_s, R=99, N=2000, seeds 0-2; gaussmix_alpha_check.csv): n_hat = N_s on all 12 runs at both levels, so the file cannot discriminate and the numbers are unaffected. The paper states alpha = 0.05, R = 99 throughout and says this explicitly.
Concern 5
report partition-level accuracy (e.g. adjusted Rand against the latent classes) under a B-selection rule that does not use n, and implement the Wasserstein and DBC baselines faithfully (multivariate W1, standard DBC objective).
Changed. Sec. 5, "Partition-level accuracy without knowledge of n" (Table "ari"; partition_accuracy.csv): ARI vs the latent state per step, B by silhouette or fixed B=8 (B=n as reference), 10 environments x 5 seeds, plus the best ARI attainable by any merge of the bins (bin-readout ceiling). Findings: silhouette recovers Tiger, K<=3 and GaussMix but picks B=2 on K-cycle K>=4 (ARI 0.19-0.41); fixed B=8 recovers only Tiger/K=2. Baselines (App. C.7): W1 is now multivariate on the full continuation vector with exact OT; DBC uses the standard bisimulation loss with a learned Gaussian latent dynamics model (reward proxy documented for reward-free envs). Found while doing this: the previous statement "silhouette chose B=K on K-cycle" had no result file and is contradicted by the run above; withdrawn.
Concern 6
support the HalfCheetah 'multi-step information' claim against a nonlinear speed baseline with trajectory-level train/test splits, or weaken the claim; provide ... result-file names and a repository URL.
Claim withdrawn. The classes are a function of the binned speed norm, so they cannot carry information beyond speed. New held-out analysis (experiments/layer2_mujoco/halfcheetah_groupsplit_value.py, halfcheetah_groupsplit_value.csv; 5-fold GroupKFold over trajectories, quotient refit on train folds, bootstrap over trajectories, 3 collection seeds): out-of-fold R^2 for the classes is below gradient-boosted trees on speed under both policies and both B rules (e.g. sinusoidal: 0.103 auto-B / 0.256 B=8 vs 0.302), classes+speed = speed alone, and trees on all 8 velocities reach 0.89 (sinusoidal). The Kruskal-Wallis p-values are demoted (dependent time steps). Result-file names are listed in App. "Artifacts, result files and corrections" with the repo URL and tag v1; the legacy first-coordinate-W1 and custom-DBC files are marked superseded.
Other claims weakened or withdrawn (found while checking)
- Abstract: "90/90 at natural resolution or silhouette" -> 90/90 only at B=n; silhouette/fixed-B results reported.
- "Permutation calibration ... only method correct on both": withdrawn.
- Dec-Tiger over-splitting mechanism: the (P^fix, T) runs show over-splitting grows with N under permutation+BH; the main-text mechanism is left as is, with the new evidence reported.
- Integration test
tests/integration/test_kcycle_continuous.py failed before this revision because it used the package default B=8; it now uses B=K, R=199 (matching Table 1).
Not addressed / limitations
- No theorem for (P); BH under dependence unproved.
- "Pow 2026" remains unpublished; nothing new relies on it.
- The RKHS/causal-state/DBC baselines are our re-implementations in the one-observation-past setting; DBSCAN threshold rescaled by us.
- Dec-Tiger perm+BH at N=20000: 3 seeds only (runtime).