arXiv:2412.19325cs.LGcs.AI2024-12被引 4

用新方法让大模型跑得快还更准,成本却像小模型

Performance Control in Early Exiting to Deploy Large Models at the Same Cost of Smaller Ones

  • 基于验证集平均准确率而非单样本置信度做退出决策
  • 相同计算量下,大模型性能超越小模型,且可精确控制准确率
  • 适合需要灵活平衡速度与精度的部署场景

早期退出(Early Exiting, EE)是一种通过根据数据难度动态分配计算资源来加速推理的技术。该方法对简单样本在浅层即完成预测,对复杂样本则保留更多计算。本文提出新视角:使用EE的大型模型可在相似计算成本下实现优于小型模型的性能。现有方法依赖各退出点的置信度判断,但存在过自信影响计算-性能权衡的可控性问题。为此,我们提出性能控制早期退出(PCEE),其决策不依赖单个样本置信度,而是基于持有验证集上具有相似置信度样本的平均准确率。实验表明,PCEE提供更优的性能控制能力,且能提升模型规模以获得性能增益的同时降低计算成本。

原文摘要 · Abstract (English)

Early Exiting (EE) is a promising technique for speeding up inference by adaptively allocating compute resources to data points based on their difficulty. The approach enables predictions to exit at earlier layers for simpler samples while reserving more computation for challenging ones. In this study, we first present a novel perspective on the EE approach, showing that larger models deployed with EE can achieve higher performance than smaller models while maintaining similar computational costs. As existing EE approaches rely on confidence estimation at each exit point, we further study the impact of overconfidence on the controllability of the compute-performance trade-off. We introduce Performance Control Early Exiting (PCEE), a method that enables accuracy thresholding by basing decisions not on a data point's confidence but on the average accuracy of samples with similar confidence levels from a held-out validation set. In our experiments, we show that PCEE offers a simple yet computationally efficient approach that provides better control over performance than standard confidence-based approaches, and allows us to scale up model sizes to yield performance gain while reducing the computational cost.

早期退出大模型性能控制推理加速

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