无需标注数据,用概率约束控制推理成本,让大模型按难易自动分级处理。
C3PO: Optimized Large Language Model Cascades with Probabilistic Cost Constraints for Reasoning
- 基于无监督输出构建级联,用置信区间控制计算开销
- 在GSM8K等4个基准上同时提升准确率与成本效率
- 适合需要低成本部署复杂推理的大模型应用
大语言模型在复杂推理任务中表现优异,但高推理成本阻碍实际部署。级联推理通过小模型处理简单问题、仅将难题交由强模型解决,是有效方案。然而现有方法依赖有标签数据训练,缺乏理论保障,且对测试时计算成本控制有限。本文提出C3PO(Cost Controlled Cascaded Prediction Optimization),一种在概率成本约束下优化级联的自监督框架。通过最小化相对于最强模型的后悔值,C3PO仅使用无标签模型输出构建级联,避免标注需求;利用合取预测技术,确保推理成本超过用户设定预算的概率可控。理论证明其具有成本控制和泛化误差保证,且小校准集即可有效优化。实证显示,C3PO在GSM8K、MATH-500、BigBench-Hard和AIME等多个推理基准上均达到当前最优性能,优于主流级联基线,在准确率与成本效率间实现更好平衡。结果表明,基于原理设计的无标签级联优化可推动大模型的可扩展部署。
原文摘要 · Abstract (English)
Large language models (LLMs) have achieved impressive results on complex reasoning tasks, but their high inference cost remains a major barrier to real-world deployment. A promising solution is to use cascaded inference, where small, cheap models handle easy queries, and only the hardest examples are escalated to more powerful models. However, existing cascade methods typically rely on supervised training with labeled data, offer no theoretical generalization guarantees, and provide limited control over test-time computational cost. We introduce C3PO (Cost Controlled Cascaded Prediction Optimization), a self-supervised framework for optimizing LLM cascades under probabilistic cost constraints. By focusing on minimizing regret with respect to the most powerful model (MPM), C3PO avoids the need for labeled data by constructing a cascade using only unlabeled model outputs. It leverages conformal prediction to bound the probability that inference cost exceeds a user-specified budget. We provide theoretical guarantees on both cost control and generalization error, and show that our optimization procedure is effective even with small calibration sets. Empirically, C3PO achieves state-of-the-art performance across a diverse set of reasoning benchmarks including GSM8K, MATH-500, BigBench-Hard and AIME, outperforming strong LLM cascading baselines in both accuracy and cost-efficiency. Our results demonstrate that principled, label-free cascade optimization can enable scalable LLM deployment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。