用投票机制实现更准更快的神经网络早退推理
SQUAD: Scalable Quorum Adaptive Decisions via ensemble of early exit neural networks
- 通过多模型中间结果投票决定是否提前退出,避免单模型置信度不可靠
- 相比顶尖动态方法准确率提升5.95%,推理延迟降低70.60%
- 适合需要低延迟高精度推理的应用,如实时AI系统
早退神经网络通过在置信度足够时提前输出中间结果来降低推理延迟。然而,传统方法通常依赖单模型置信度阈值,易受校准问题影响。为此,我们提出SQUAD(可扩展的共识自适应决策),首个将早退机制与分布式集成学习结合的推理方案,在提升不确定性估计的同时减少推理时间。SQUAD采用基于共识的停止策略,按计算复杂度顺序逐步收集各早退学习器的中间预测,直到达成统计显著的共识即停止计算。为最大化投票效果,我们还提出QUEST(共识搜索技术),一种神经架构搜索方法,用于选择在每一层都具备互补性的早退学习器。该共识驱动方法实现了统计上更稳健的早退,相较于最先进的动态方案,测试准确率最高提升5.95%,计算成本相当;相比静态集成,推理延迟最高降低70.60%,同时保持良好准确率。
原文摘要 · Abstract (English)
Early-exit neural networks have become popular for reducing inference latency by allowing intermediate predictions when sufficient confidence is achieved. However, standard approaches typically rely on single-model confidence thresholds, which are frequently unreliable due to inherent calibration issues. To address this, we introduce SQUAD (Scalable Quorum Adaptive Decisions), the first inference scheme that integrates early-exit mechanisms with distributed ensemble learning, improving uncertainty estimation while reducing the inference time. Unlike traditional methods that depend on individual confidence scores, SQUAD employs a quorum-based stopping criterion on early-exit learners by collecting intermediate predictions incrementally in order of computational complexity until a consensus is reached and halting the computation at that exit if the consensus is statistically significant. To maximize the efficacy of this voting mechanism, we also introduce QUEST (Quorum Search Technique), a Neural Architecture Search method to select early-exit learners with optimized hierarchical diversity, ensuring learners are complementary at every intermediate layer. This consensus-driven approach yields statistically robust early exits, improving the test accuracy up to 5.95% compared to state-of-the-art dynamic solutions with a comparable computational cost and reducing the inference latency up to 70.60% compared to static ensembles while maintaining a good accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。