RetroThinker框架提升语音大语言模型推理能力
RetroThinker: Enabling Retrospective Thinking in Speech LLMs
这个框架很实用,能帮语音大模型在说话时自我修正,提升推理准确率,和普通模型比效果更好。
RetroThinker是一个多阶段后训练框架,为Moshi模型提供自我验证和向前纠正的链式思考步骤。它通过在推理过程中结合监督微调和基于长度的直接偏好优化,在GSM8K基准测试上实现了11%的绝对准确率提升,同时保持与基线相当的延迟。
RetroThinker: Enabling Retrospective Thinking in Speech LLMs
Speech large language models (SpeechLLMs) offer reduced latency and retain paralinguistic nuances that are typically lost in cascaded automatic speech recognition (ASR) and text-based LM architectures. However, they continue to lag behind text-only LLMs on complex reasoning tasks, while real-time spoken interaction imposes strict latency constraints. Although prior works employ Chain-of-Thought (CoT) and concurrent reasoning to enhance reasoning capabilities without inducing prohibitive delays, an inherent accuracy-latency trade-off persists. In this paper, we investigate whether a streaming SpeechLLM can dynamically revise its reasoning traces on the fly. We introduce RetroThinker, a multi-stage post-training framework that equips the Moshi model to self-verify and forward-correct CoT steps during inference. RetroThinker combines supervised fine-tuning (SFT) on curated retrospective thinking data with length-based direct preference optimization (DPO) to optimize retrospective during early reasoning (i.e., reasoning concurrently while the user speaks). Evaluated on the GSM8K benchmark, RetroThinker significantly improves the accuracy-latency trade-off over non-retrospective baselines, achieving an 11% absolute accuracy gain at a comparable latency.
- 小互09-10 02:42原文