Kernel Normalized Convolutional Networks

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Nasirigerdeh, Reza, Torkzadehmahani, Reihaneh, Rueckert, Daniel, Kaissis, Georgios
Format: Preprint
Veröffentlicht: 2022
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909127470481408
author Nasirigerdeh, Reza
Torkzadehmahani, Reihaneh
Rueckert, Daniel
Kaissis, Georgios
author_facet Nasirigerdeh, Reza
Torkzadehmahani, Reihaneh
Rueckert, Daniel
Kaissis, Georgios
contents Existing convolutional neural network architectures frequently rely upon batch normalization (BatchNorm) to effectively train the model. BatchNorm, however, performs poorly with small batch sizes, and is inapplicable to differential privacy. To address these limitations, we propose the kernel normalization (KernelNorm) and kernel normalized convolutional layers, and incorporate them into kernel normalized convolutional networks (KNConvNets) as the main building blocks. We implement KNConvNets corresponding to the state-of-the-art ResNets while forgoing the BatchNorm layers. Through extensive experiments, we illustrate that KNConvNets achieve higher or competitive performance compared to the BatchNorm counterparts in image classification and semantic segmentation. They also significantly outperform their batch-independent competitors including those based on layer and group normalization in non-private and differentially private training. Given that, KernelNorm combines the batch-independence property of layer and group normalization with the performance advantage of BatchNorm.
format Preprint
id arxiv_https___arxiv_org_abs_2205_10089
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Kernel Normalized Convolutional Networks
Nasirigerdeh, Reza
Torkzadehmahani, Reihaneh
Rueckert, Daniel
Kaissis, Georgios
Machine Learning
Computer Vision and Pattern Recognition
Existing convolutional neural network architectures frequently rely upon batch normalization (BatchNorm) to effectively train the model. BatchNorm, however, performs poorly with small batch sizes, and is inapplicable to differential privacy. To address these limitations, we propose the kernel normalization (KernelNorm) and kernel normalized convolutional layers, and incorporate them into kernel normalized convolutional networks (KNConvNets) as the main building blocks. We implement KNConvNets corresponding to the state-of-the-art ResNets while forgoing the BatchNorm layers. Through extensive experiments, we illustrate that KNConvNets achieve higher or competitive performance compared to the BatchNorm counterparts in image classification and semantic segmentation. They also significantly outperform their batch-independent competitors including those based on layer and group normalization in non-private and differentially private training. Given that, KernelNorm combines the batch-independence property of layer and group normalization with the performance advantage of BatchNorm.
title Kernel Normalized Convolutional Networks
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2205.10089