Junhyuk So
Author directory2026
Beyond Confidence: Test-Time Scaling for Multi-Turn Search Agents via Retrieval Grounding
Hyunho Kook | Junhyuk So | Tianyu Fu | Haizhong Zheng | Beidi Chen
Findings of the Association for Computational Linguistics: EMNLP 2026
Hyunho Kook | Junhyuk So | Tianyu Fu | Haizhong Zheng | Beidi Chen
Findings of the Association for Computational Linguistics: EMNLP 2026
Confidence-based voting aggregates parallel LLM rollouts by weighting each with internal signals such as token log probabilities, and has been actively studied for single-turn reasoning. However, modern LLMs increasingly act as multi-turn search agents that retrieve and condition on external documents. In this paper, we show that confidence-based voting transfers poorly to this multi-turn setting, and identify the underlying failure reason as copy inflation: when retrieved documents are appended to an agent’s context, tokens copied from those documents receive systematically inflated log probabilities. This flattens confidence scores within each question and weakens the resulting weighted vote. To address this issue, we propose Retrieval-Grounded Voting (RGV), which scores each rollout by the lexical overlap between its final answer and the documents it retrieved. By computing the signal outside the contaminated context, RGV sidesteps both token log probabilities and additional LLM calls. Across four search-agent benchmarks and five LLMs, RGV consistently outperforms confidence-based voting, with gains of up to +5.4% accuracy and +35% on minority-correct questions, where the correct answer appears in only 1-2 of 8 rollouts.
LoopCD: Loop-wise Contrastive Decoding for Improving Reasoning in Looped Language Models
Byeongho Yu | Junhyuk So | Eunhyeok Park
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Byeongho Yu | Junhyuk So | Eunhyeok Park
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Looped Language Models (LoopLMs) perform “latent reasoning” by recursively refining internal latent representations with shared weights, offering a more effective alternative to explicit verbal reasoning. Despite their effectiveness, we find that LoopLMs remain prone to loop instability: excessive iterations can lead to errors in later iterations, producing “hard” localized uncertain tokens that contribute to reasoning errors. To address this, we propose LoopCD, loop-wise contrastive decoding that enhances the reasoning performance of LoopLMs by intervening on these tokens at inference time. Specifically, we exploit the internal dynamics of LoopLMs and contrast the logits from earlier iterations with logits from the last refined iteration to form the final sampling distribution. We find that this strategy is highly efficient, introducing only negligible inference overhead and requiring no additional training, while effectively improving reasoning performance by naturally refining reasoning-critical hard tokens. Extensive experiments show that our method improves the performance of recent representative LoopLMs across various reasoning tasks.