DART: Input-Difficulty-AwaRe Adaptive Threshold for Early-Exit DNNs
Organizations: Brandenburg Technical University, Cottbus, Germany · Tallinn University of Technology, Tallinn, Estonia · Leibniz Institute for High Performance Microelectronics, Frankfurt Oder, Germany
Abstract
Early-exit deep neural networks enable adaptive inference by terminating computation when sufficient confidence is achieved, reducing cost for edge AI accelerators in resource-constrained settings. Existing methods, however, rely on suboptimal exit policies, ignore input difficulty, and optimize thresholds independently. This paper introduces DART (Input-Difficulty-Aware Adaptive Threshold), a framework that overcomes these limitations. DART introduces three key innovations: (1) a lightweight difficulty estimation module that quantifies input complexity with minimal computational overhead, (2) a joint exit policy optimization algorithm based on dynamic programming, and (3) an adaptive coefficient management system. Experiments on diverse DNN benchmarks (AlexNet, ResNet-18, VGG-16) demonstrate that DART achieves up to \textbf{3.3} speedup, \textbf{5.1} lower energy, and up to 42% lower average power compared to static networks, while preserving competitive accuracy. Extending DART to Vision Transformers (LeViT) yields power (5.0) and execution-time (3.6) gains but also accuracy loss (up to 17 percent), underscoring the need for transformer-specific early-exit mechanisms. We further introduce the Difficulty-Aware Efficiency Score (DAES), a novel multi-objective metric, under which DART achieves up to a 14.8 improvement over baselines, highlighting superior accuracy, efficiency, and robustness trade-offs.
Figures & tables
| Architecture | Method | Acc.(%) | Time(ms) | Energy(mJ) | Power(W) | Speedup | Power Eff. | DAES |
| MNIST Results | ||||||||
| AlexNet | Static | 98.97 | 0.09 | 5.90 | 65.56 | 1.00 | 0.562 | |
| BranchyNet | 99.13 | 0.08 | 5.27 | 65.88 | 1.12 | 0.713 | ||
| RL-Agent | 99.49 | 0.06 | 2.29 | 38.20 | 2.57 | 2.179 | ||
| DART | 99.31 | 0.03 | 1.15 | 38.33 | 5.13 | 8.684 | ||
| CIFAR-10 Results (CNNs) | ||||||||
| Model | Method | Acc. (%) | MACs (M) | Time (ms) | Speedup ( ) |
|---|---|---|---|---|---|
| LeViT-128S | Static | 95.80 | 282.1 | 11.55 | |
| DART | 81.73 | 56.4 | 4.56 | ||
| LeViT-192 | Static | 96.91 | 601.1 | 19.94 | |
| DART | 80.33 | 120.2 | 5.57 | ||
| LeViT-256 | Static | 97.28 | 1053.3 | 62.55 | |
| DART | 86.11 | 210.7 | 18.89 |
Efficient Neural Network Inference
Efficient neural network inference reduces the time, memory, and energy needed to run trained models across hardware and deployment settings.
See all 468 papers