arXiv:2604.24927cs.CLcs.AI2026-04中稿 · ICML

让大模型生成更多元答案,用预测误差引导探索新思路。

Large Language Models Explore by Latent Distilling

  • 用轻量级模型在生成时预测深层表征,捕捉语义变化规律。
  • 在数学、科学、代码等任务上显著提升多样性与效率,超越现有方法。
  • 适合需要多角度推理或创意写作的场景,不牺牲连贯性。

大语言模型测试时扩展的关键在于生成多样响应,但标准随机采样仅带来表层词汇变化,限制语义探索。本文提出探索式采样(ESamp),一种在生成过程中显式鼓励语义多样性的解码方法。该方法基于神经网络对熟悉输入预测误差低、新颖输入误差高的现象,训练一个轻量级蒸馏器,在测试时从浅层表示预测深层隐藏表示,以建模语言模型的深度表征演变。生成过程中,蒸馏器随当前上下文动态适应,利用预测误差作为新颖性信号,重新加权候选词扩展,引导解码走向更少被探索的语义模式。ESamp采用异步训练-推理流水线,最坏情况开销低于5%(优化版为1.2%)。实验证明,其显著提升推理模型的Pass@k效率,在数学、科学、代码生成等基准上表现优于或相当强于现有随机与启发式基线。尤其在创意写作中打破多样性与连贯性的权衡,展现出鲁棒泛化能力。代码已开源:https://github.com/LinesHogan/tLLM。

原文摘要 · Abstract (English)

Generating diverse responses is crucial for test-time scaling of large language models (LLMs), yet standard stochastic sampling mostly yields surface-level lexical variation, limiting semantic exploration. In this paper, we propose Exploratory Sampling (ESamp), a decoding approach that explicitly encourages semantic diversity during generation. ESamp is motivated by the well-known observation that neural networks tend to make lower-error predictions on inputs similar to those encountered before, and incur higher prediction error on novel ones. Building on this property, we train a lightweight Distiller at test time to predict deep-layer hidden representations of the LLM from its shallow-layer representations to model the LLM's depth-wise representation transitions. During decoding, the Distiller continuously adapts to the mappings induced by the current generation context. ESamp uses the prediction error as a novelty signal to reweight candidate token extensions conditioned on the current prefix, thereby biasing decoding toward less-explored semantic patterns. ESamp is implemented with an asynchronous training--inference pipeline, with less than 5% worst case overhead (1.2% in the optimized release). Empirical results show that ESamp significantly boosts the Pass@k efficiency of reasoning models, showing superior or comparable performance to strong stochastic and heuristic baselines. Notably, ESamp achieves robust generalization across mathematics, science, and code generation benchmarks and breaks the trade-off between diversity and coherence in creative writing. Our code has released at: https://github.com/LinesHogan/tLLM.

大模型生成语义多样性探索采样推理增强

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。