Early Time Classification with Accumulated Accuracy Gap Control
Liran Ringel, Regev Cohen, Daniel Freedman, Michael Elad, Yaniv Romano
Abstract
Early time classification algorithms aim to label a stream of features without processing the full input stream, while maintaining accuracy comparable to that achieved by applying the classifier to the entire input. In this paper, we introduce a statistical framework that can be applied to any sequential classifier, formulating a calibrated stopping rule. This data-driven rule attains finite-sample, distribution-free control of the accuracy gap between full and early-time classification. We start by presenting a novel method that builds on the Learn-then-Test calibration framework to control this gap marginally, on average over i.i.d. instances. As this algorithm tends to yield an excessively high accuracy gap for early halt times, our main contribution is the proposal of a framework that controls a stronger notion of error, where the accuracy gap is controlled conditionally on the accumulated halt times. Numerical experiments demonstrate the effectiveness, applicability, and usefulness of our method. We show that our proposed early stopping mechanism reduces up to 94% of timesteps used for classification while achieving rigorous accuracy gap control.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers3
- Fast yet Safe: Early-Exiting with Risk ControlMetod Jazbec, Alexander Timans, Tin Hadzi Veljkovic, Kaspar Sakmann et al.NeurIPS 2024 · 35 citations
- Conformal Thinking: Risk Control for Reasoning on a Compute BudgetXi Wang, Anushri Suresh, Alvin Zhang, Rishi More et al.ICML 2026 · 10 citations
- Temporal Difference Calibration in Sequential Tasks: Application to Vision-Language-Action ModelsShelly Francis-Meretzki, Mirco Mutti, Yaniv Romano, Aviv TamarICML 2026 · 2 citations
Builds on8
- Efficient Memory Management for Large Language Model Serving with PagedAttentionWoosuk Kwon, Zhuohan Li, Siyuan Zhuang, Ying Sheng et al.SOSP 2023 · 1,016 citations
- Adaptive Conformal Inference Under Distribution ShiftIsaac Gibbs, Emmanuel J. CandèsNeurIPS 2021 · 665 citations
- Classification with Valid and Adaptive CoverageYaniv Romano, Matteo Sesia, Emmanuel J. CandèsNeurIPS 2020 · 586 citations
- Confident Adaptive Language ModelingTal Schuster, Adam Fisch, Jai Gupta, Mostafa Dehghani et al.NeurIPS 2022 · 394 citations
- Conformal Risk ControlAnastasios Nikolas Angelopoulos, Stephen Bates, Adam Fisch, Lihua Lei et al.ICLR 2024 · 242 citations
Related papers
- Recurrent Halting Chain for Early Multi-label ClassificationThomas Hartvigsen, Cansu Sen, Xiangnan Kong, Elke A. RundensteinerKDD 2020 · 18 citations
- Conformal Prediction for Early Stopping in Mixed Integer OptimizationStefan Clarke, Bartolomeo StellatoICML 2026
- Conformal Inference is (almost) Free for Neural Networks Trained with Early StoppingZiyi Liang, Yanfei Zhou, Matteo SesiaICML 2023 · 19 citations
- A Skip-Connected Evolving Recurrent Neural Network for Data Stream Classification under Label Latency ScenarioMonidipa Das, Mahardhika Pratama, Jie Zhang, Yew-Soon OngAAAI 2020 · 14 citations
- Rethinking Calibration for Early-Exit Neural NetworksPiotr Kubaty, Filip Szatkowski, Grzegorz Choczyński, Eric Nalisnick et al.ICML 2026
