arXiv:2608.09351cs.LGcs.AI2026-08

输入多样性比输出多样性更高效提升大模型推理准确率

Test-Time Augmentation for LLMs: When Input Diversity Beats Output Diversity at Matched Compute

论文配图:Test-Time Augmentation for LLMs: When Input Diversity Beats Output Diversity at Matched Compute
图 1 · 摘自论文原文
  • 通过改写输入、扰动词汇等方法增强输入多样性
  • 相同算力下准确率提升1.8倍,5/6任务优于自一致性方法
  • 适合预算有限的中等规模模型部署使用

测试时扩展(TTA)通过在输入侧引入多样性(如语义重述、词汇扰动、视觉变换),在六大数据集上与链式思维提示和自一致性方法进行匹配算力对比。结果显示,语义重述策略在所有任务中均显著提升准确率,且成本效益高出约1.8倍,在五项任务上超越自一致性。分析表明,该方法对中等规模模型最有效,当更强模型不可用或成本过高时尤为适用。结果表明,当前中等规模大模型中,输入多样性比仅改变推理路径更高效地将计算资源转化为准确率。代码已开源。

原文摘要 · Abstract (English)

Test-time scaling improves LLM accuracy but multiplies inference cost, making the accuracy gained per unit of compute the metric that matters in deployment. Self-consistency is one of the established approaches, which spends this budget entirely on the output side by sampling repeated reasoning paths. We study Test-Time Augmentation (TTA), which extends self-consistency by also perturbing the input, aggregating predictions across transformed versions of the input, and ask whether input-side diversity converts compute into accuracy more efficiently than output-side diversity. We perform a systematic, matched-compute comparison: we evaluate three simple input-side strategies (semantic rephrasing, lexical perturbations, and visual transformations) across six datasets covering general and multilingual knowledge, mathematical reasoning, multi-modal question answering, and sentiment classification, against chain-of-thought prompting and self-consistency. Semantic rephrasing delivers consistent and statistically significant accuracy gains while Pareto-dominating self-consistency on cost-effectiveness, delivering roughly 1.8X more accuracy per dollar and outperforming it on five of six tasks. We further analyze the number of augmentations, multi-modal strategies, and base model scaling, finding that TTA is most cost-effective for mid-tier models where a stronger model is unavailable or too expensive. Our findings indicate that for current mid-tier LLMs, varying the input converts inference compute into accuracy more efficiently than varying the reasoning path alone. The TTA implementation is available at https://github.com/aws-samples/sample-genai-reflection-for-bedrock.

大模型推理测试时增强成本效益

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