Confidence-gated training for efficient early-exit neural networks

Saad Mokssit, Ouassim Karrakchou, Alejandro Mousist +1

Early-exit neural networks reduce inference cost by enabling confident predictions at intermediate layers. However, joint training often leads to gradient interference, with deeper…