DE$^3$-BERT: Distance-Enhanced Early Exiting for BERT based on Prototypical Networks

Early exiting has demonstrated its effectiveness in accelerating the inference of pre-trained language models like BERT by dynamically adjusting the number of layers executed. However, most existing early exiting methods only consider local information from an individual test sample to determine their exiting indicators, failing to leverage the global information offered by sample population. This leads to suboptimal estimation of prediction correctness, resulting in erroneous exiting decisions. To bridge the gap, we explore the necessity of effectively combining both local …

adjusting bert cs.cl cs.lg global inference information language language models networks sample test