递归推理缩放让大模型推理更高效,性能提升超55种方法。
Recursive Inference Scaling: A Winning Path to Scalable Inference in Language and Multimodal Systems
- 通过递归深度机制实现推理时长可扩展,优于55种现有策略。
- 在固定算力下,语言建模性能显著提升,0-样本图像识别准确率增2%。
- 轻量适配器+随机丢弃,训练时无损,推理时即生效,适合预训练集成。
受语言分形几何启发,本文提出递归推理缩放(RINS),一种适用于语言与多模态系统的可插拔推理扩展方法。RINS是一种特定的递归深度形式,在计算资源匹配条件下,显著优于包括‘重复全部’(RAO)和潜在循环思维在内的55种其他变体。在固定模型规模与训练算力预算下,RINS大幅提升了语言建模性能,并推广至多模态系统:例如,在SigLIP-B/16上实现0-样本ImageNet准确率提升2%。通过推导数据缩放定律,RINS同时改善了性能上限与缩放指数。更重要的是,仅需<1%参数的轻量适配器与随机丢弃,RINS具备“无后悔”特性——即使推理时不启用递归深度,训练阶段仍能提升语言建模表现,表明其在训练、参数与推理全匹配环境下具有成为大模型预训练组件的巨大潜力。
原文摘要 · Abstract (English)
Inspired by recent findings on the fractal geometry of language, we introduce Recursive INference Scaling (RINS) as a complementary, plug-in recipe for scaling inference time in language and multimodal systems. RINS is a particular form of recursive depth that significantly outperforms +55 other variants, including the recent "repeat-all-over" (RAO) strategy in Mobile LLM (Liu et al., 2024) and latent recurrent thinking (Geiping et al., 2025). Unlike prior works, we carry out our comparisons on a compute-matched regime, and demonstrate that for a fixed model size and training compute budget, RINS substantially improves language modeling performance. It also generalizes beyond pure language tasks, delivering gains in multimodal systems, including a +2% improvement in 0-shot ImageNet accuracy for SigLIP-B/16. Additionally, by deriving data scaling laws, we show that RINS improves both the asymptotic performance limits and the scaling exponents. More importantly, with light-weight (linear) adapters (comprising <1% of model parameters) and stochastic dropout, RINS offers a no-regret strategy, meaning that RINS-enabled pretraining improves performance in language modeling even when recursive depth is not applied at inference time. This corresponds to improving performance on a training compute-, parameter-, and inference-matched regime, suggesting its potential as a viable component of LLM pretraining!
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。