Kinetics: Rethinking Test-Time Scaling Laws

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Sadhukhan, Ranajoy, Chen, Zhuoming, Zheng, Haizhong, Zhou, Yang, Strubell, Emma, Chen, Beidi
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918065318395904
author Sadhukhan, Ranajoy
Chen, Zhuoming
Zheng, Haizhong
Zhou, Yang
Strubell, Emma
Chen, Beidi
author_facet Sadhukhan, Ranajoy
Chen, Zhuoming
Zheng, Haizhong
Zhou, Yang
Strubell, Emma
Chen, Beidi
contents We rethink test-time scaling laws from a practical efficiency perspective, revealing that the effectiveness of smaller models is significantly overestimated. Prior work, grounded in compute-optimality, overlooks critical memory access bottlenecks introduced by inference-time strategies (e.g., Best-of-$N$, long CoTs). Our holistic analysis, spanning models from 0.6B to 32B parameters, reveals a new Kinetics Scaling Law that better guides resource allocation by incorporating both computation and memory access costs. Kinetics Scaling Law suggests that test-time compute is more effective when used on models above a threshold than smaller ones. A key reason is that in TTS, attention, rather than parameter count, emerges as the dominant cost factor. Motivated by this, we propose a new scaling paradigm centered on sparse attention, which lowers per-token cost and enables longer generations and more parallel samples within the same resource budget. Empirically, we show that sparse attention models consistently outperform dense counterparts, achieving over 60 points gains in low-cost regimes and over 5 points gains in high-cost regimes for problem-solving accuracy on AIME, encompassing evaluations on state-of-the-art MoEs. These results suggest that sparse attention is essential and increasingly important with more computing invested, for realizing the full potential of test-time scaling where, unlike training, accuracy has yet to saturate as a function of computation, and continues to improve through increased generation. The code is available at https://github.com/Infini-AI-Lab/Kinetics.
format Preprint
id arxiv_https___arxiv_org_abs_2506_05333
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Kinetics: Rethinking Test-Time Scaling Laws
Sadhukhan, Ranajoy
Chen, Zhuoming
Zheng, Haizhong
Zhou, Yang
Strubell, Emma
Chen, Beidi
Machine Learning
Computation and Language
We rethink test-time scaling laws from a practical efficiency perspective, revealing that the effectiveness of smaller models is significantly overestimated. Prior work, grounded in compute-optimality, overlooks critical memory access bottlenecks introduced by inference-time strategies (e.g., Best-of-$N$, long CoTs). Our holistic analysis, spanning models from 0.6B to 32B parameters, reveals a new Kinetics Scaling Law that better guides resource allocation by incorporating both computation and memory access costs. Kinetics Scaling Law suggests that test-time compute is more effective when used on models above a threshold than smaller ones. A key reason is that in TTS, attention, rather than parameter count, emerges as the dominant cost factor. Motivated by this, we propose a new scaling paradigm centered on sparse attention, which lowers per-token cost and enables longer generations and more parallel samples within the same resource budget. Empirically, we show that sparse attention models consistently outperform dense counterparts, achieving over 60 points gains in low-cost regimes and over 5 points gains in high-cost regimes for problem-solving accuracy on AIME, encompassing evaluations on state-of-the-art MoEs. These results suggest that sparse attention is essential and increasingly important with more computing invested, for realizing the full potential of test-time scaling where, unlike training, accuracy has yet to saturate as a function of computation, and continues to improve through increased generation. The code is available at https://github.com/Infini-AI-Lab/Kinetics.
title Kinetics: Rethinking Test-Time Scaling Laws
topic Machine Learning
Computation and Language
url https://arxiv.org/abs/2506.05333