REVIEW 5 major objections 5 minor 1 cited by
Test-time iteration over a query embedding—feeding each pooled state back through the same decoder-only model—improves semantic reasoning in retrieval, with gains up to 21% and no retraining.
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-04 09:55 UTC pith:ZFQOCODP
load-bearing objection RT is a plausible test-time refinement idea, but Listing 1 leaves the core state-injection mechanism unspecified, so the reported gains are not yet interpretable. the 5 major comments →
Refine Thought: A Test-Time Inference Method for Embedding Model Reasoning
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
On its own terms, RT's central claim is that semantic reasoning is a multi-step computation and a single forward pass of an embedding model lacks the computational depth to carry it out. RT turns the query encoder into a recurrent system: for T steps it re-encodes the query together with the mean-pooled hidden states of all previous steps, and uses the last pooled state, h_T, as the query embedding. Across BRIGHT subtasks the overall average rises from 22.9 to 23.1; on PJBenchmark, JD2CV jumps from 62.77 to 74.33 and CV2CV from 52.14 to 63.26; C-MTEB STS scores stay flat. The authors attribute the effect to 'activating the semantic reasoning ability learned during pretraining' in decoder-onl
What carries the argument
The central mechanism is the RT loop: after each forward pass, all token representations before the [EOS] token are mean-pooled into a state h_t; the next pass encodes the concatenation of the original query and all accumulated states; after T steps, the last pooled state becomes the query embedding. This is temporal unrolling 'implicitly in the hidden space'—the embedding-model analogue of chain-of-thought, but without generating any text. The recurrence is h_t = f(x, h_1, ..., h_{t-1}), and the key assumption is that decoder-only autoregressive models can integrate these recycled states to refine their representation.
Load-bearing premise
The load-bearing premise is that `concatenate(x_q, states)` in Listing 1 is a well-defined way to feed previous pooled embedding states back into the decoder; the paper never specifies how high-dimensional state vectors become model input tokens, and its own limitation section concedes sensitivity to step count and state aggregation.
What would settle it
Fix T=2 and compare RT against a control that appends the same number of fixed, query-independent tokens at each iteration. If accuracy stays at the same level, token addition is doing the work; if it drops to the T=1 baseline, temporal unrolling is the cause. A second check: implement state injection into a specific hidden layer and compare with a text-serialized version on the same tasks.
If this is right
- Reasoning-heavy retrieval can be improved by spending a few extra forward passes on the query alone; document embeddings and indexes remain unchanged.
- The gain saturates at T=2-3, so the latency cost is small and predictable, with diminishing returns beyond.
- Decoder-only embedding backbones become the preferred choice for tasks that require compositional or multi-constraint matching.
- Since simple similarity tasks are unaffected, the method can be applied selectively to hard queries without degrading general retrieval.
- The results give empirical support to the theoretical view that temporal unrolling trades time steps for effective computational depth.
Where Pith is reading between the lines
- The paper leaves unspecified how the high-dimensional pooled states are fed into the decoder. If they are serialized as text, the gains could come from added tokens rather than recurrence; a control with appended random or fixed tokens would separate the two.
- If the mechanism is genuine hidden-state recurrence, RT should also improve arithmetic and symbolic embedding tasks, which the paper does not test.
- The paper's own limitation discussion in Section 4.3 highlights sensitivity to step count and state aggregation; a practical extension would couple RT with an early-stopping or drift signal to control cost.
- The large gap between decoder-only and encoder-only models suggests a testable prediction: giving an encoder-only model an explicit recurrent state-injection module should recover part of the gap, but only if state feedback is implemented in the hidden space rather than as text.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes RT (Refine Thought), a test-time method for text embedding models that iteratively runs forward passes on the query, feeding intermediate pooled representations back into the model, and uses the final state as the query embedding. The method is evaluated on three benchmarks: BRIGHT (semantic reasoning retrieval), PJBenchmark (person-job matching), and C-MTEB STS (general semantic similarity). The authors report a small average gain on BRIGHT (22.9 to 23.1), large gains on two PJBenchmark Algorithm tasks (JD2CV +19%, CV2CV +21%), and stable performance on C-MTEB STS. They attribute these effects to temporal unrolling that 'activates' semantic reasoning abilities learned during pretraining in decoder-only models, and argue decoder-only architectures benefit more than encoder-only ones.
Significance. If the method is correctly specified and the reported gains are reproducible, the idea is noteworthy: it offers a training-free way to trade inference compute for improved embedding-model reasoning, with a plausible connection to looped transformers and chain-of-thought literature. The paper also addresses a real gap in embedding-model reasoning and includes a pseudocode description, a decoder-only versus encoder-only comparison, and an explicit discussion of limitations. However, the central implementation step is underspecified, the experimental evidence is too thin to support the strong causal and comparative claims, and the per-task selection of hyperparameter T raises concerns about test-set fitting. The paper's own limitations sections concede sensitivity to T and state-aggregation, and acknowledge that the architecture/task/data attribution is not established, which directly tempers the abstract's conclusions.
major comments (5)
- [§3.2, Listing 1] Step 2, 'concatenate(x_q, states)', mixes a token sequence with a list of pooled dense vectors. The paper never explains how the pooled states are converted into model input. If they are serialized into text, the method reduces to prompt augmentation and the gains may be caused by added tokens rather than temporal unrolling; if they are injected into hidden layers, the implementation is nonstandard and incompatible with the cited Qwen3-Embedding interface. This ambiguity affects every reported result. Please specify the exact tensor-level operation and add a control condition (e.g., appending random or fixed vectors of the same shape, or concatenating the same state repeatedly) to isolate the temporal-unrolling effect from simple input modification.
- [§4.2, Tables 2–3] The abstract and conclusion call the improvements 'significant', but the reported numbers do not support this. BRIGHT average improves from 22.9 to 23.1, while several subtasks decline (e.g., StackExchange 21.3→21.1, Rob 18.8→18.0, Sus 18.3→18.1). PJBenchmark gains are concentrated in two Algorithm tasks (JD2CV +19%, CV2CV +21%) while Finance JD2CV declines from 62.66 to 61.30 (-2%). No error bars, confidence intervals, significance tests, or variance estimates are provided. The strong wording is disproportionate to the effect size. Please add per-task variability measures, a multiple-comparison-aware significance test, and qualify the claim to 'numerically positive on average' unless statistical support is provided.
- [§4.1, §4.3 and Figure 2] T is selected per task after scanning values 1–10, and the paper states that 'more complex tasks require larger T' without a prespecified complexity measure. This constitutes ex post hyperparameter fitting on the test benchmarks. Because the final representation is h_T, choosing T based on benchmark results can inflate the reported gains. Please report the chosen T for every benchmark/task, show the full T sweep for all tasks (Figure 2 currently gives only an illustrative sketch), and propose an a priori rule for selecting T (e.g., based on query length or a separate complexity model) for practical use.
- [Abstract and §4.3/§5] The causal claim that RT 'activates the semantic reasoning ability learned during pretraining' is not directly tested. The paper's own limitations concede sensitivity to T and state-aggregation, drift on long inputs, lack of explicit credit assignment, and state in §5 that 'further studies are required to disentangle whether the observed improvements arise from differences in architecture, task type, or training data.' These concessions are in tension with the abstract and conclusion. Please soften the causal attribution to a phenomenological observation, or provide an experiment that isolates the pretraining-reasoning-trajectory mechanism (e.g., ablation on a decoder-only model without reasoning pretraining data).
- [§4.2, Table 4 and §5] The conclusion that 'decoder-only architectures are inherently more effective for semantic reasoning' is based on a single decoder-only model (Qwen3-Embedding-8B) and a single encoder-only model (bge-large-zh-v1.5). This is an extremely narrow basis for a general architectural claim, especially given the differing model scales and training data. Please test additional decoder-only and encoder-only models of comparable scale, or explicitly limit the conclusion to the evaluated models.
minor comments (5)
- [Listing 1 and §4.1] The pseudocode contains a typo ('previus') and the definition of T is inconsistent: h_0 is computed before the loop, but the text says 'T=1 is the single-step baseline.' Clarify whether the baseline is h_0 (no refinement) or h_1 (one refinement) and make the pseudocode's loop range match.
- [Tables 1–3] The table formatting is degraded: Table 2's header merges column labels awkwardly, and Table 3 has 'JD2CV' appearing under both Algorithmic and Finance. Please also specify the evaluation metric for each benchmark (e.g., nDCG@k, accuracy, Spearman correlation) and report dataset sizes.
- [Figures 1–2] Figure 1 is referenced but not described in detail, and Figure 2 lacks axis labels and a legend. Please ensure both figures are self-contained and that the T sensitivity curves are legible.
- [§4.2, paragraph 1] The phrase 'RT remains stable on general-purpose semantic understanding tasks' is accurate for C-MTEB STS, but 'achieves consistent gains on semantic reasoning tasks' is contradicted by the BRIGHT subtask declines and the PJBenchmark Finance decline. Rephrase to reflect the actual distribution of results.
- [§2 and §3.2] The claim that RT operates 'implicitly in the hidden space' cites only a survey [21]; please cite the original looped-transformer or latent-thought works, such as [14], more directly. Also clarify how the method differs from simply adding more tokens to the query when states are serialized.
Circularity Check
No self-citation chain or definitional circularity; the main circularity-adjacent issue is that the iteration count T is selected from the same benchmark curves used to report RT's gains, making the headline improvement partly a best-of-T construction.
specific steps
-
fitted input called prediction
[§3.2 p.3; §4.1 p.3; §4.2 p.4 and Figure 2]
"The step count T is set based on task complexity, with more complex tasks requiring larger T. ... We set the number of iterative steps to T∈{1,2,...,10}, where T=1 is the single-step baseline, and larger values correspond to deeper temporal unrolling. ... Results indicate that the most notable gains occur at T=2–3, after which performance saturates for T>3, aligning with the notion of trading time steps for effective depth."
T is the only free hyperparameter of RT. The paper scans T∈{1..10}, observes that gains peak at T=2–3, and then reports RT's improved numbers on BRIGHT/PJBenchmark. Because T=1 is the baseline inside the scanned set, choosing T per task after inspecting Figure 2 makes the reported improvement non-negative by construction on the tuning data. The statement that gains scale with task complexity is then read from the same curves used to choose T; no per-task T values or held-out T selection are reported. The benchmarks are external and real, so the claim is not completely empty, but the headline improvement is partially a best-of-T selection rather than a fixed-method prediction.
full rationale
RT is an empirical test-time procedure rather than a derivation, so there is no equation-level reduction of a prediction to its inputs. The paper has no self-citations by its own authors, no imported uniqueness theorem, and no ansatz smuggled in via citation; the cited theoretical support for temporal unrolling is external. The evaluations on BRIGHT, C-MTEB, and PJBenchmark are external benchmarks, which gives the central empirical claim real content. The one substantial circularity-adjacent flaw is hyperparameter selection: T is scanned over {1..10}, the gains are reported to peak at T=2–3, and the same curves appear to justify both the per-task T choice and the claim that more complex tasks benefit more. Since T=1 is the baseline in the scanned set, selecting the best T makes the reported improvement non-negative by construction on the tuning data. The paper does not report per-task T values or any held-out T selection, so the reported gains partly reflect fitting to the evaluation benchmarks. This is a partial fitted-input-called-prediction issue, not a full collapse: the method could have failed at all T, and the C-MTEB stability result is a genuine negative finding. The underspecified concatenate(x_q, states) in Listing 1 is a reproducibility/correctness concern rather than circularity. Overall, the central claim retains independent empirical content, but the headline improvement is not a parameter-free prediction.
Axiom & Free-Parameter Ledger
free parameters (3)
- T (number of refinement steps) =
2-3 (per task, after scanning 1-10)
- state pooling scheme =
mean pooling of all tokens before [EOS]
- state injection format =
not specified
axioms (4)
- domain assumption A single forward pass fails to capture multi-step semantic reasoning.
- domain assumption Temporal unrolling in hidden space is analogous to CoT and improves reasoning.
- ad hoc to paper Feeding previous pooled states back as input tokens gives meaningful refinement rather than noise.
- domain assumption Decoder-only pretraining imbues embedding models with latent reasoning trajectories.
Cite this review
Pith. "Pith review of Refine Thought: A Test-Time Inference Method for Embedding Model Reasoning." pith.science (2026). https://pith.science/paper/ZFQOCODP
@misc{pith2026251113726,
author = {Pith},
title = {Pith review of: Refine Thought: A Test-Time Inference Method for Embedding Model Reasoning},
year = {2026},
howpublished = {\url{https://pith.science/paper/ZFQOCODP}},
note = {Machine review of arXiv:2511.13726}
}
read the original abstract
We propose RT (Refine Thought), a method that can enhance the semantic reasoning ability of text embedding models. The method obtains the final semantic representation by running multiple forward passes of the text embedding model. Experiments show that RT achieves significant improvements on semantic reasoning tasks in BRIGHT and the person-job matching benchmark PJBenchmark, while maintaining consistent performance on general-purpose semantic understanding tasks such as C-MTEB. Our results indicate that RT is effective because it further activates the semantic reasoning ability learned during pretraining by decoder-only text embedding models (e.g., Qwen3-Embedding-8B). RT can be seen as a test-time inference method.
Figures
Forward citations
Cited by 1 Pith paper
-
A Survey of Reasoning-Intensive Retrieval: Progress and Challenges
A survey that categorizes RIR benchmarks by domain and modality, proposes a taxonomy for integrating reasoning into retrieval pipelines, and outlines key challenges.
Reference graph
Works this paper leans on
-
[1]
DeepSeek-AI, D. Guo, D. Yang, H. Zhang, J. Song, R. Zhang, R. Xu, Q. Zhu, S. Ma, P. Wang, et al. Deepseek-r1: Incentivizing reasoning capability in llms via reinforcement learning.arXiv preprint arXiv:2501.12948, 2025. URLhttps://arxiv.org/abs/2501.12948
Pith/arXiv arXiv 2025
-
[2]
Q. Dong, L. Dong, Y . Tang, T. Ye, Y . Sun, Z. Sui, and F. Wei. Reinforcement pre-training. arXiv preprint arXiv:2506.08007, 2025. URLhttps://arxiv.org/abs/2506.08007
Pith/arXiv arXiv 2025
-
[3]
W. Fan, Y . Ding, L. Ning, S. Wang, H. Li, D. Yin, T.-S. Chua, and Q. Li. A survey on rag meeting llms: Towards retrieval-augmented large language models. InProceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD). Association for Computing Machinery, 2024
2024
-
[4]
A. Graves, G. Wayne, and I. Danihelka. Neural turing machines.arXiv preprint arXiv:1410.5401, 2014. URLhttps://arxiv.org/abs/1410.5401
Pith/arXiv arXiv 2014
-
[6]
Huang and K
J. Huang and K. C.-C. Chang. Towards reasoning in large language models: A survey. In Findings of the Association for Computational Linguistics: ACL 2023. Association for Compu- tational Linguistics, 2023
2023
-
[7]
LeCun, Y
Y . LeCun, Y . Bengio, and G. Hinton. Deep learning.Nature, 521(7553):436–444, 2015. URL https://www.nature.com/articles/nature14539
2015
-
[8]
Z. Li, H. Liu, D. Zhou, and T. Ma. Chain of thought empowers transformers to solve inherently serial problems. InProceedings of the International Conference on Learning Representations (ICLR), 2024
2024
-
[9]
Lightman, V
H. Lightman, V . Kosaraju, Y . Burda, H. Edwards, B. Baker, T. Lee, J. Leike, J. Schulman, I. Sutskever, and K. Cobbe. Let’s verify step by step. InProceedings of the International Conference on Learning Representations (ICLR), 2024
2024
- [10]
-
[11]
Raffel, N
C. Raffel, N. Shazeer, A. Roberts, K. Lee, S. Narang, M. Matena, Y . Zhou, W. Li, P. J. Liu, et al. Exploring the limits of transfer learning with a unified text-to-text transformer.Journal of Machine Learning Research, 21(140):1–67, 2020. URL https://jmlr.org/papers/v21/ 20-074.html
2020
-
[12]
L. Ruis, M. Mozes, J. Bae, S. R. Kamalakara, D. Talupuru, A. Locatelli, R. Kirk, T. Rocktäschel, E. Grefenstette, M. Bartolo, et al. Procedural knowledge in pretraining drives reasoning in large language models. InProceedings of the International Conference on Learning Representations (ICLR), 2025
2025
-
[13]
Sastre and A
I. Sastre and A. Rosá. Memory tokens: Large language models can generate reversible sentence embeddings. InProceedings of the First Workshop on Large Language Model Memorization (L2M2) at ACL 2025, 2025
2025
-
[14]
Saunshi, N
N. Saunshi, N. Dikkala, Z. Li, S. Kumar, and S. J. Reddi. Reasoning with latent thoughts: On the power of looped transformers. InProceedings of the International Conference on Learning Representations (ICLR), 2025. 6
2025
-
[15]
A. Setlur, C. Nagpal, A. Fisch, X. Geng, J. Eisenstein, R. Agarwal, A. Agarwal, J. Berant, and A. Kumar. Rewarding progress: Scaling automated process verifiers for llm reasoning. arXiv preprint arXiv:2410.08146, 2024. doi: 10.48550/arXiv.2410.08146. URL https: //arxiv.org/abs/2410.08146
-
[16]
C. Snell, J. Lee, K. Xu, and A. Kumar. Scaling llm test-time compute optimally can be more effective than scaling model parameters.arXiv preprint arXiv:2408.03314, 2024. URL https://arxiv.org/abs/2408.03314
Pith/arXiv arXiv 2024
-
[17]
H. Su, H. Yen, M. Xia, W. Shi, N. Muennighoff, H.-Y . Wang, H. Liu, Q. Shi, Z. S. Siegel, M. Tang, R. Sun, J. Yoon, S. O. Arik, D. Chen, and T. Yu. Bright: A realistic and challenging benchmark for reasoning-intensive retrieval.arXiv preprint arXiv:2407.12883, 2024. URL https://arxiv.org/abs/2407.12883
Pith/arXiv arXiv 2024
-
[18]
J. Wei, X. Wang, D. Schuurmans, M. Bosma, B. Ichter, F. Xia, E. Chi, Q. Le, and D. Zhou. Chain-of-thought prompting elicits reasoning in large language models.arXiv preprint arXiv:2201.11903, 2022. URLhttps://arxiv.org/abs/2201.11903
Pith/arXiv arXiv 2022
-
[19]
S. Xiao, Z. Liu, P. Zhang, N. Muennighoff, D. Lian, and J.-Y . Nie. C-pack: Packed resources for general chinese embeddings.arXiv preprint arXiv:2309.07597, 2023. URL https:// arxiv.org/abs/2309.07597
Pith/arXiv arXiv 2023
-
[20]
Y . Zhang, M. Li, D. Long, X. Zhang, H. Lin, B. Yang, P. Xie, A. Yang, D. Liu, J. Lin, F. Huang, and J. Zhou. Qwen3 embedding: Advancing text embedding and reranking through foundation models.arXiv preprint arXiv:2506.05176, 2025. URL https://arxiv.org/abs/2506. 05176
Pith/arXiv arXiv 2025
-
[23]
URLhttps://arxiv.org/abs/2507.06203. 7
-
[2025]
URLhttps://arxiv.org/abs/2510.01265
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.