REVIEW 2 major objections 5 minor 24 references
Supervised fine-tuning of LLMs pushes action diversity below the minimum that accuracy requires, and training on every optimal action per state preserves that diversity.
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
T0 review · deepseek-v4-flash
2026-08-01 12:30 UTC pith:XXUUTPXD
load-bearing objection Solid empirical study of SFT-induced diversity collapse; the qualitative findings hold, but the headline quantitative claim is undercut by 8-sample entropy estimates. the 2 major comments →
When Reasoning Narrows the Move: Diversity Collapse in LLM Game Play
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
The paper's central claim is that SFT and reasoning-mode generation cause LLM policies to concentrate on single moves beyond what optimal play demands. The authors measure action entropy per state in four solved board games and compare model trajectories against an analytic upper bound — the iso-accuracy entropy ceiling of Eq. 2 — which is the maximum entropy any policy of a given accuracy can achieve. They find that plain SFT often loses more entropy than needed to reach its accuracy, a condition they call premature diversity collapse; reasoning mode amplifies this without reliably improving accuracy; and training on all optimal actions per state (action augmentation) keeps the policy much
What carries the argument
The load-bearing object is the iso-accuracy entropy ceiling — the per-state maximum entropy formula H^max_s(p_s) = h_2(p_s) + p_s log2 k_s + (1-p_s) log2(n_s - k_s), where p_s is the probability mass on optimal actions, k_s the number of optimal actions, n_s legal actions, and h_2 the binary entropy. It gives the highest mean entropy attainable at any mean accuracy on a state pool, so the gap between a model's observed trajectory and this curve isolates diversity loss beyond what accuracy mechanically requires. The second mechanism is action augmentation: including every optimal action per state in SFT data, rather than a single demonstrated action, which directly widens the support the mode
Load-bearing premise
The claim that diversity loss 'exceeds what is minimally required' depends on the measured per-state action entropy from only 8 samples per state faithfully reflecting the model's true entropy — with 8 samples the estimate is biased downward and capped at 3 bits, so some of the measured collapse could be an artifact of the sampling budget.
What would settle it
Recompute the empirical action entropy with a much larger sample budget per state (say 64 or 256) or with a bias-corrected estimator (e.g., Chao-Shen or Miller-Madow) on the same evaluation pools, and compare the resulting trajectories against the iso-accuracy ceiling. If the gap between plain SFT and the ceiling largely disappears, or the early-epoch collapse no longer appears, the 'premature collapse' result fails.
If this is right
- Standard SFT can quietly replace exploratory, multi-move behavior with near-deterministic play even on simple solved games.
- Reasoning traces, when trained as single demonstrations, concentrate the policy further; reasoning alone is not a substitute for diverse training data.
- Training on all optimal actions per state preserves substantially more action entropy at comparable accuracy, offering a concrete data-level intervention.
- Evaluations of LLM decision-making should measure action diversity alongside accuracy, since the two can decouple.
Where Pith is reading between the lines
- The entropy measurement at N=8 samples per state caps estimated diversity at log2(8)=3 bits; bias-corrected or larger-N estimates could shrink the reported gap to the iso-accuracy ceiling, so the magnitude (not necessarily the ordering) of the collapse may be overstated.
- The iso-accuracy ceiling presupposes that accuracy is the only driver of optimal concentration; in games with strategic value beyond win/loss (e.g., depth-to-mate), the minimum entropy could be even lower, which would make the ceiling conservative — a direction the authors did not explore.
- The action-augmentation intervention could be tested as a general recipe for other sequential decision tasks (e.g., coding or tool use) where multiple correct actions exist per state; if it transfers, it offers a simple alternative to loss-function modifications for preserving diversity.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. This paper studies how supervised fine-tuning (SFT) affects action diversity in LLM decision-making, using four deterministic tic-tac-toe variants with exactly computable optimal actions. The authors evaluate Qwen3-8B and LoRA fine-tuned variants in direct and reasoning modes, with either single-action ('plain') or multi-action ('augmented') supervision. Across state-level accuracy/entropy evaluations, arena Elo/trajectory-entropy measurements, and checkpoint trajectories, they report that reasoning compresses diversity without consistent accuracy gains, that plain SFT induces an early diversity collapse that they claim exceeds the accuracy–diversity tradeoff (by comparison to an analytic 'iso-accuracy entropy ceiling', Eq. 2), and that action-augmented SFT partially mitigates the collapse. The paper interprets these results as evidence of narrow-support imitation as a failure mode.
Significance. If the central quantitative claim is established, the paper would provide a clean, controlled demonstration that standard SFT can push LLM policies below the accuracy–diversity frontier, with a simple data-level intervention (multi-action augmentation) that partially restores diversity. The study's strengths are its exact game solvers, three independent measurement modes (state-level, arena, checkpoints), bootstrap uncertainty quantification, and an analytic ceiling derived from first principles rather than fitted to data. These features make the qualitative ordering of conditions (plain < augmented; reasoning < direct) credible. The quantitative 'exceeds the tradeoff' claim, however, is load-bearing and currently depends on measurement and ceiling-construction details that need to be addressed before the headline conclusion can be accepted.
major comments (2)
- [§4.1, §5, Figure 4] The central claim that plain SFT 'exceeds what is minimally required' is read off the gap between measured checkpoint entropies and the Eq. (2) ceiling. All state-level entropies are plug-in estimates from N=8 action samples per state. This estimator is downward-biased (Miller–Madow bias is roughly (m−1)/(2N ln2), i.e., several tenths of a bit for policies with entropy around 1–2 bits) and hard-capped at log2 8 = 3 bits. Figure 4 shows the base model already at the cap, and mid-training checkpoints are in the bias-prone region. The magnitude and, in borderline cases, the existence of the 'premature collapse' gap are therefore not established at the reported precision. Please re-estimate entropies with larger N or a bias-corrected estimator and report confidence intervals on the gap. The qualitative ordering is likely robust.
- [§5, Eq. (2)] The statement 'No policy with the same mean accuracy can achieve higher mean entropy on that state pool' is not justified by the derivation as written. H_max_s(p_s) is a state-dependent concave function of the per-state accuracy p_s. If the dashed curve is obtained by averaging H_max_s(p) at a common p, it is not a valid upper envelope for policies that are allowed to allocate their correct actions unevenly across states: with state-dependent (n_s,k_s), a policy with mean accuracy p̄ can achieve strictly higher average entropy than E_s[H_max_s(p̄)] by allocating its correct decisions unevenly across states. The comparison in Figure 4 can therefore overstate how far SFT goes below the frontier. Either define the ceiling as the solution of the constrained optimization over per-state accuracies, or explicitly restrict the comparison to policies with the same per-state accuracy profile, and
minor comments (5)
- [Abstract] Typo: 'would partially mitigates this effect' → 'would partially mitigate this effect.'
- [§3] 'P fis the transition kernel' appears to be a typo for 'P is the transition kernel.'
- [§4.2] The Elo expected-reward formula introduces τ without defining it; state the scaling constant used.
- [Figures 2 and 3] Legend labels ('Base SFT-Plain SFT-Aug Direct Reasoning') are visually crowded and make it hard to associate arrows with conditions; use separate subpanels or clearer glyphs.
- [§5] The description of checkpoint selection says 'three intermediate checkpoints' but lists {1,6,12,18}; clarifying which are intermediate vs final would help reproducibility.
Circularity Check
No significant circularity: the headline diversity-collapse claim is benchmarked against an externally derived analytic ceiling, not against a fitted curve or self-citation chain.
full rationale
The paper's central derivation chain is empirical comparison against an externally derived analytic ceiling. Equation (2) defines the iso-accuracy entropy ceiling from per-state numbers of optimal actions k_s, legal actions n_s, and the policy's accuracy p_s; it is a mathematical upper envelope, not a quantity fitted to the measured entropy data. Accuracy and diversity are measured independently: accuracy is the frequency of solver-verified optimal actions, and diversity is the plug-in action entropy from N=8 samples per state. The claim that plain SFT 'exceeds what is minimally required' is the gap between these measured trajectories and the ceiling, which is a benchmark rather than a fitted prediction. No parameter is fitted to the data and then reported as a prediction. There are no load-bearing self-citations by the authors; the cited game environments, baselines, and methods are external. The N=8 plug-in entropy estimator may be biased and capped at log2(8)=3 bits, and the ceiling construction could be debated on concavity or state-level aggregation grounds, but these are measurement/correctness concerns, not circularity: the comparison is not true by construction, as shown by the aug-reason condition retaining higher diversity at comparable accuracy. The stated limitation to small deterministic games and a single model family is a scope limitation, not a circular step. Therefore the analysis is self-contained with respect to circularity.
Axiom & Free-Parameter Ledger
free parameters (2)
- Evaluation samples per state N =
8
- sTTT solver search depth cutoff =
d=4
axioms (5)
- domain assumption Solver-computed game-theoretic values for oTTT/cTTT/dTTT define the true 'optimal action sets' used as SFT labels and accuracy ground truth.
- domain assumption The empirical entropy of 8 decoded samples approximates the policy's true action entropy per state.
- standard math For fixed per-state accuracy p_s, maximum entropy is attained by uniform mass over optimal actions and uniform mass over suboptimal actions (Eq. 2).
- standard math Trajectory entropy decomposes as the sum of per-step action entropies under the induced state distribution (Eq. 1).
- ad hoc to paper The curated 200-state evaluation pools (tactical-category balanced, excluding states where every action is optimal) are representative of realistic play.
read the original abstract
Supervised fine-tuning (SFT) is widely used to adapt large language models to downstream tasks, but its effect on behavioral diversity in sequential decision-making remains under-explored. We study this question in a controlled suite of deterministic board games based on tic-tac-toe variants, where optimal actions are exactly computable and diversity can be measured directly. Across state-level evaluation, arena gameplay, and training trajectories, we find that reasoning-mode generation frequently suppresses action diversity without uniformly improving action accuracy. Furthermore, standard SFT improves accuracy but often induces premature diversity collapse, which exceeds what is minimally required by the accuracy-diversity tradeoff. We then show that action augmentation, which trains on all optimal actions per state rather than a single demonstrated action, would partially mitigates this effect. Our results identify narrow-support imitation as a source of policy collapse in LLM decision-making and suggest that preserving action support during SFT is important for maintaining exploratory behavior.
Figures
Reference graph
Works this paper leans on
-
[1]
Lakshya A Agrawal, Shangyin Tan, Dilara Soylu, Noah Ziems, Rishi Khare, Krista Opsahl-Ong, Arnav Singhvi, Herumb Shandilya, Michael J Ryan, Meng Jiang, et al. Gepa: Reflective prompt evolution can outperform reinforcement learning.arXiv preprint arXiv:2507.19457,
-
[6]
D4rl: Datasets for deep data-driven reinforcement learning.arXiv preprint arXiv:2004.07219,
Justin Fu, Aviral Kumar, Ofir Nachum, George Tucker, and Sergey Levine. D4rl: Datasets for deep data-driven reinforcement learning.arXiv preprint arXiv:2004.07219,
Pith/arXiv arXiv 2004
-
[7]
Yulun Jiang, Yekun Chai, Maria Brbi´c, and Michael Moor. Marble: A hard benchmark for multimodal spatial reasoning and planning.arXiv preprint arXiv:2506.22992,
-
[8]
Robert Kirk, Ishita Mediratta, Christoforos Nalmpantis, Jelena Luketina, Eric Hambro, Edward Grefenstette, and Roberta Raileanu. Understanding the effects of rlhf on llm generalisation and diversity.arXiv preprint arXiv:2310.06452,
-
[10]
Sergey Levine, Aviral Kumar, George Tucker, and Justin Fu. Offline reinforcement learning: Tutorial, review, and perspectives on open problems.arXiv preprint arXiv:2005.01643,
Pith/arXiv arXiv 2005
-
[13]
One fish, two fish, but not the whole sea: Alignment reduces language models’ conceptual diversity
Sonia Krishna Murthy, Tomer Ullman, and Jennifer Hu. One fish, two fish, but not the whole sea: Alignment reduces language models’ conceptual diversity. InProceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers), pages 11241–11258,
2025
-
[15]
Chanwoo Park, Xiangyu Liu, Asuman Ozdaglar, and Kaiqing Zhang. Do llm agents have regret? a case study in online learning and games.arXiv preprint arXiv:2403.16843,
-
[16]
Offline learning of controllable diverse behaviors.arXiv preprint arXiv:2504.18160,
Mathieu Petitbois, Rémy Portelas, Sylvain Lamprier, and Ludovic Denoyer. Offline learning of controllable diverse behaviors.arXiv preprint arXiv:2504.18160,
-
[17]
Oguzhan Topsakal, Colby Jacob Edell, and Jackson Bailey Harper. Evaluating large language models with grid-based game competitions: An extensible LLM benchmark and leaderboard.arXiv preprint arXiv:2407.07796,
-
[18]
Chessqa: Evaluating large language models for chess understanding.arXiv preprint arXiv:2510.23948,
Qianfeng Wen, Zhenwei Tang, and Ashton Anderson. Chessqa: Evaluating large language models for chess understanding.arXiv preprint arXiv:2510.23948,
-
[19]
Zhiheng Xi, Jixuan Huang, Chenyang Liao, Baodai Huang, Honglin Guo, Jiaqi Liu, Rui Zheng, Junjie Ye, Jiazheng Zhang, Wenxiang Chen, et al. Agentgym-rl: Training llm agents for long-horizon decision making through multi-turn reinforcement learning.arXiv preprint arXiv:2509.08755,
-
[20]
Qwen3 technical report.arXiv preprint arXiv:2505.09388,
An Yang, Anfeng Li, Baosong Yang, Beichen Zhang, Binyuan Hui, Bo Zheng, Bowen Yu, Chang Gao, Chengen Huang, Chenxu Lv, et al. Qwen3 technical report.arXiv preprint arXiv:2505.09388,
-
[21]
Sheng Yue, Jiani Liu, Xingyuan Hua, Ju Ren, Sen Lin, Junshan Zhang, and Yaoxue Zhang. How to leverage diverse demonstrations in offline imitation learning.arXiv preprint arXiv:2405.17476,
-
[22]
The price of format: Diversity collapse in llms.arXiv preprint arXiv:2505.18949,
Longfei Yun, Chenyang An, Zilong Wang, Letian Peng, and Jingbo Shang. The price of format: Diversity collapse in llms.arXiv preprint arXiv:2505.18949,
-
[23]
Dylan Zhang, Yufeng Xu, Haojin Wang, Qingzhi Chen, and Hao Peng. Good sft optimizes for sft, better sft prepares for reinforcement learning.arXiv preprint arXiv:2602.01058,
-
[24]
move": <action_label>}</action> The JSON key must be
11 Appendix A Implementation Details A.1 Arena Players Table 1 summarizes the players introduced in our arena: four deterministic policies and three open- source language models. We select the deterministic policies to provide interpretable anchors that span both skill and behavioral diversity, yielding broad coverage of the Elo-versus-diversity 2D space....
2024
-
[1978]
Rvs: What is essential for offline rl via supervised learning?arXiv preprint arXiv:2112.10751,
Scott Emmons, Benjamin Eysenbach, Ilya Kostrikov, and Sergey Levine. Rvs: What is essential for offline rl via supervised learning?arXiv preprint arXiv:2112.10751,
-
[2020]
Ziniu Li, Congliang Chen, Tian Xu, Zeyu Qin, Jiancong Xiao, Zhi-Quan Luo, and Ruoyu Sun. Preserv- ing diversity in supervised fine-tuning of large language models.arXiv preprint arXiv:2408.16673,
-
[2021]
Yijie Chen, Yijin Liu, and Fandong Meng. Sed-sft: Selectively encouraging diversity in supervised fine-tuning.arXiv preprint arXiv:2602.07464,
-
[2022]
Attributing mode collapse in the fine-tuning of large language models
Laura O’Mahony, Leo Grinsztajn, Hailey Schoelkopf, and Stella Biderman. Attributing mode collapse in the fine-tuning of large language models. InICLR 2024 Workshop on Mathematical and Empirical Understanding of Foundation Models, volume 2, page 2,
2024
-
[2023]
Sai Kolasani, Maxim Saplin, Nicholas Crispino, Kyle Montgomery, Jared Quincy Davis, Matei Zaharia, Chi Wang, and Chenguang Wang. Llm chess: Benchmarking reasoning and instruction- following in llms through chess.arXiv preprint arXiv:2512.01992,
-
[2024]
Prakamya Mishra, Jiang Liu, Jialian Wu, Xiaodong Yu, Zicheng Liu, and Emad Barsoum. Ttt-bench: A benchmark for evaluating reasoning ability with simple and novel tic-tac-toe-style games.arXiv preprint arXiv:2506.10209,
-
[2025]
Yuntao Bai, Andy Jones, Kamal Ndousse, Amanda Askell, Anna Chen, Nova DasSarma, Dawn Drain, Stanislav Fort, Deep Ganguli, Tom Henighan, et al. Training a helpful and harmless assistant with reinforcement learning from human feedback.arXiv preprint arXiv:2204.05862,
-
[2026]
Lucia Cipolina-Kun, Marianna Nezhurina, and Jenia Jitsev. Game reasoning arena: A framework and benchmark for assessing reasoning capabilities of large language models via game play.arXiv preprint arXiv:2508.03368,
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.