Paper Reading AI Learner

Beyond Test-Time Compute Strategies: Advocating Energy-per-Token in LLM Inference

2026-03-04 15:20:55
Patrick Wilhelm, Thorsten Wittkopp, Odej Kao

Abstract

Large Language Models (LLMs) demonstrate exceptional performance across diverse tasks but come with substantial energy and computational costs, particularly in request-heavy scenarios. In many real-world applications, the full scale and capabilities of LLMs are often unnecessary, as Small Language Models (SLMs) can provide accurate responses for simpler text generation tasks. When enhanced with advanced reasoning strategies, such as Chain-of-Thought (CoT) prompting or Majority Voting, SLMs can approach the performance of larger models while reducing overall computational requirements. However, these strategies can also introduce additional energy costs, creating an energy-accuracy trade-off. Our analysis examines these trade-offs in test-time compute strategies for smaller models compared to larger ones, using the MMLU benchmark. Additionally, we explore the input-output token dynamics of transformer architectures, which result in nonlinear hardware energy operation curves for LLMs. To bridge AI research with its physical impact, we propose \textit{energy efficiency metrics}, including Energy-per-Token, as complements to traditional accuracy benchmarks. Beyond model selection, we propose controlled reasoning in CoT token generation, using operating curves to regulate reasoning depth dynamically. This vision integrates a energy-aware routing mechanism, ensuring that model selection and inference strategies balance accuracy for sustainable AI deployment.

Abstract (translated)

大型语言模型(LLMs)在各类任务中展现出卓越性能,但其高昂的能耗与计算成本在请求密集场景中尤为突出。在实际应用中,LLM的完整规模与能力往往并不必要,因为小型语言模型(SLMs)已能针对简单文本生成任务提供准确响应。当SLMs结合思维链(CoT)提示或多数投票等先进推理策略时,其性能可逼近大型模型,同时降低总体计算需求。然而,这些策略也可能带来额外能耗,形成能耗-精度权衡。本研究通过MMLU基准测试,对比分析了小型与大型模型在测试时计算策略中的此类权衡。此外,我们探讨了Transformer架构的输入输出令牌动态特性,该特性导致LLMs呈现非线性硬件能耗运行曲线。为连接AI研究与物理影响,我们提出将包括每令牌能耗在内的能效指标作为传统精度基准的补充。除模型选择外,我们还提出在CoT令牌生成中实施可控推理,利用运行曲线动态调节推理深度。这一愿景整合了能感知能耗的路由机制,确保模型选择与推理策略在精度与可持续性部署间取得平衡。

URL

https://arxiv.org/abs/2603.20224

PDF

https://arxiv.org/pdf/2603.20224.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot