Batch Normalization (BN) enhances neural network generalization and accelerates training by normalizing mini-batches to a uniform mean and variance. However, its performance degrades with diverse data distributions. To overcome this, we introduce Supervised Batch Normalization (SBN), which extends normalization by leveraging multiple mean and variance parameters to account for contexts identified prior to training. These contexts-defined explicitly (e.g., domains in domain adaptation) or implicitly (e.g., via clustering algorithms)-ensure effective normalization for samples with shared features. Experiments across single- and multi-task datasets demonstrate the superiority of SBN over BN and other normalization techniques. For example, integrating SBN with Vision Transformer yields a 15.13% accuracy boost on CIFAR-100, while in domain adaptation scenarios, SBN with AdaMatch achieves a 22.25% accuracy gain on MNIST and SVHN compared to BN. Our code implementation is available on our GitHub repository: https://github.com/b-faye/supervised-batch-normalization .

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Supervised Batch Normalization

  • Bilal Faye,
  • Hanane Azzag,
  • Mustapha Lebbah

摘要

Batch Normalization (BN) enhances neural network generalization and accelerates training by normalizing mini-batches to a uniform mean and variance. However, its performance degrades with diverse data distributions. To overcome this, we introduce Supervised Batch Normalization (SBN), which extends normalization by leveraging multiple mean and variance parameters to account for contexts identified prior to training. These contexts-defined explicitly (e.g., domains in domain adaptation) or implicitly (e.g., via clustering algorithms)-ensure effective normalization for samples with shared features. Experiments across single- and multi-task datasets demonstrate the superiority of SBN over BN and other normalization techniques. For example, integrating SBN with Vision Transformer yields a 15.13% accuracy boost on CIFAR-100, while in domain adaptation scenarios, SBN with AdaMatch achieves a 22.25% accuracy gain on MNIST and SVHN compared to BN. Our code implementation is available on our GitHub repository: https://github.com/b-faye/supervised-batch-normalization .