Frontier AI training focused on maximizing pretraining parameter scale regardless of operational cost; new scaling laws prove that architectural innovations like mixture-of-experts and grouped query attention deliver superior reasoning at fractions of inference latency.

The modern AI revolution was ignited by Chinchilla and Kaplan scaling laws, which dictated that larger parameter models trained on more compute tokens inevitably yield superior downstream performance.
However, massive dense transformer models incur catastrophic inference costs in real-world deployment, requiring multi-billion-dollar data centers and high electrical power to serve user queries.
This foundational computer science study formulates revised scaling laws that explicitly incorporate inference compute budgets. The authors demonstrate that sparse mixture-of-experts (MoE) routing and optimized key-value cache architectures achieve equivalent accuracy to massive dense models while slashing inference memory and latency by up to seventy percent.
These inference-aware scaling laws provide the architectural blueprint for next-generation generative AI, shifting the industry frontier from unsustainable brute-force scaling to cost-effective, high-throughput model architectures.
Scaling Laws Meet Model Architecture: Toward Inference-Efficient LLMs
Scaling the number of parameters and the size of training data has proven to be an effective strategy for improving large language model (LLM) performance. Yet, as these models grow increasingly powerful and widely deployed, the cost of inference has become a pressing concern. Despite its importance, the trade-off between model accuracy and inference efficiency remains underexplored. In this work, we examine how key architectural factors, hidden size, the allocation of parameters between MLP and attention (mlp-to-attention ratio), and grouped-query attention (GQA), influence both inference cost and accuracy. We introduce a conditional scaling law that augments the Chinchilla framework with architectural information, along with a search framework for identifying architectures that are simultaneously inference-efficient and accurate. To validate our approach, we train more than 200 models spanning 80M to 3B parameters and 8B to 100B training tokens, and fit the proposed conditional scaling law. Our results show that the conditional scaling law reliably predicts optimal architectural choices and that the resulting models outperform existing open-source baselines. Under the same training budget, optimized architectures achieve up to 2.1% higher accuracy and 42% greater inference throughput compared to LLaMA-3.2.
Ask this paper your own questions, or keep browsing the verified research catalogue.