arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2510.04573cs.LGcs.AIcs.CL

LaDiR:潜在扩散增强大语言模型进行文本推理

LaDiR: Latent Diffusion Enhances LLMs for Text Reasoning

University of California, San Diego(加州大学圣迭戈分校);Apple(苹果公司)

Haoqiang Kang, Yizhe Zhang, Nikki Lijing Kuang, Nicklas Majamaki, Navdeep Jaitly, Yi-An Ma, Lianhui Qin

AI总结:

LaDiR通过结合连续潜在表示的表达力与潜在扩散模型的迭代精炼能力,提升大语言模型在文本推理中的准确性、多样性和可解释性。

摘要:

大型语言模型(LLMs)通过链式思维(CoT)生成展现其推理能力。然而,LLM的自回归解码可能限制其全面回顾和精炼早期标记的能力,导致探索多样化解决方案的效率低下。本文提出LaDiR(潜在扩散推理器),一种新的推理框架,将连续潜在表示的表达力与潜在扩散模型的迭代精炼能力结合到现有LLM中。我们首先使用变分自编码器(VAE)构建结构化的潜在推理空间,将文本推理步骤编码为思想标记块,保留语义信息和可解释性,同时提供紧凑但具有表现力的表示。随后,我们利用潜在扩散模型,学习通过块级双向注意力掩码去噪块状潜在思想标记,实现更长的视野和适应性测试时间计算的迭代精炼。这种设计结合扩散推理期间的显式多样性指导,使生成多个多样化的推理轨迹,探索潜在空间的不同区域,而非产生重复的解决方案。我们在数学推理、代码生成和谜题规划基准上进行了评估。实验结果表明,LaDiR在准确性、多样性和可解释性上均优于现有的自回归、扩散基和潜在推理方法,揭示了通过潜在扩散进行文本推理的新范式。

英文摘要:

Large Language Models (LLMs) demonstrate their reasoning ability through chain-of-thought (CoT) generation. However, LLM's autoregressive decoding may limit the ability to revisit and refine earlier tokens in a holistic manner, which can also lead to inefficient exploration for diverse solutions. In this paper, we propose LaDiR} (Latent Diffusion Reasoner), a novel reasoning framework that unifies the expressiveness of continuous latent representation with the iterative refinement capabilities of latent diffusion models for an existing LLM. We first construct a structured latent reasoning space using a Variational Autoencoder (VAE) that encodes text reasoning steps into blocks of thought tokens, preserving semantic information and interpretability while offering compact but expressive representations. Subsequently, we utilize a latent diffusion model that learns to denoise a block of latent thought tokens with a blockwise bidirectional attention mask, enabling longer horizon and iterative refinement with adaptive test-time compute. This design, combined with explicit diversity guidance during diffusion inference, enables the generation of multiple diverse reasoning trajectories that explore distinct regions of the latent space, rather than producing repetitive solutions as often occurs in standard autoregressive sampling. We conduct evaluations on a suite of mathematical reasoning, code generation and puzzle planning benchmarks. Empirical results show that LaDiR consistently improves accuracy, diversity, and interpretability over existing autoregressive, diffusion-based, and latent reasoning methods, revealing a new paradigm for text reasoning with latent diffusion.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑