Zero Time Waste in Pre-trained Early Exit Neural Networks
Dec 1, 2023·
,
,,,,,,·
0 min read
Bartosz Wójcik
Marcin Przewięźlikowski

Filip Szatkowski
Maciej Wołczyk
Klaudia Bałazy
Igor Podolak
Jacek Tabor
Marek Śmieja
Tomasz Trzciński
Abstract
Early exit neural networks enable adaptive inference by allowing samples to exit at different network depths based on prediction confidence. However, existing approaches often waste computational resources during training and inference. We propose a novel training methodology that eliminates this waste, achieving superior efficiency-accuracy trade-offs in pre-trained models while maintaining the flexibility of adaptive computation.
Publication
In Neural Networks