Batch Normalization Is Blind to the First and Second Derivatives of the Loss

Published: 01 Jan 2024, Last Modified: 15 May 2025AAAI 2024EveryoneRevisionsBibTeXCC BY-SA 4.0
Abstract: We prove that when we do the Taylor series expansion of the loss function, the BN operation will block the influence of the first-order term and most influence of the second-order term of the loss. We also find that such a problem is caused by the standardization phase of the BN operation. We believe that proving the blocking of certain loss terms provides an analytic perspective for potential detects of a deep model with BN operations, although the blocking problem is not fully equivalent to significant damages in all tasks on benchmark datasets. Experiments show that the BN operation significantly affects feature representations in specific tasks.
Loading