English

AdaStop: Cost-Aware Early Stopping for DNN Test Selection

Machine Learning 2026-07-06 v1 Artificial Intelligence

Abstract

Existing methods for testing deep neural networks (DNNs) primarily prioritize test inputs likely to reveal model faults under a fixed labeling budget. In practice, choosing that budget is difficult: too little testing misses failures, while too much incurs unnecessary labeling costs. This work studies the stopping problem in DNN testing. We formulate testing as a cost--benefit decision process in which labeling an input incurs cost cc and discovering a fault yields value vv. Based on this formulation, we introduce \textit{AdaStop}, a framework that estimates the marginal fault discovery rate during testing and stops labeling when the estimated rate falls below the threshold τ=c/v\tau = c/v. Experiments across multiple datasets, architectures, and selection strategies show that 6565--84%84\% of faults can be discovered using only 99--31%31\% of the labeling budget.

Cite

@article{arxiv.2607.05461,
  title  = {AdaStop: Cost-Aware Early Stopping for DNN Test Selection},
  author = {Bonan Shen and Wei-Jung Huang and Xin Liu and Jiazhou Gao and Tao Ning},
  journal= {arXiv preprint arXiv:2607.05461},
  year   = {2026}
}