What if we could delete BatchNorm and LayerNorm entirely? ๐ Introducing SaluNet: a framework that replaces normalization with a simple learnable geometry (just 2 scalars per layer). It works on both CNNs and ViTs. ๐งต๐
@liuzhuang1234 Great video! To add:
Normalization rescales activations using stats. Derf learns a scaled and shifted erf. SALU learns saturation amplitude and transition width around a fixed center at zero.
Different geometric choices for normalization-free networks.
https://t.co/nwj0mLd2MS
What if we could delete BatchNorm and LayerNorm entirely? ๐ Introducing SaluNet: a framework that replaces normalization with a simple learnable geometry (just 2 scalars per layer). It works on both CNNs and ViTs. ๐งต๐
To put it simply:
Normalization moves the distribution.
Derf moves the function.
SALU deforms the function around a fixed center.
A different geometric perspective on going normalization-free. ๐
What if we could delete BatchNorm and LayerNorm entirely? ๐ Introducing SaluNet: a framework that replaces normalization with a simple learnable geometry (just 2 scalars per layer). It works on both CNNs and ViTs. ๐งต๐
Paper & Code:
๐ arXiv: https://t.co/eOiPC2fQRl
๐ป GitHub: https://t.co/B0dRtOWb9R (Pretrained checkpoints included)
Looking for collaborators on:
โข Matched baselines & seeds
โข Large-batch training
RTs appreciated! ๐ #DeepLearning#ComputerVision#ViT
What if we could delete BatchNorm and LayerNorm entirely? ๐ Introducing SaluNet: a framework that replaces normalization with a simple learnable geometry (just 2 scalars per layer). It works on both CNNs and ViTs. ๐งต๐
The proof on CIFAR-100 ๐
ResNet-18 + BatchNorm? Accuracy collapses at batch size 1. ๐ฑ
SaluNet-C-18? Retains 76.23% Top-1 at batch size 1.
It also scales to ImageNet-1K:
โข ResNet-50: 80.16%
โข ViT-B/16: 80.28%
The problem with BatchNorm/LayerNorm? They rely on batch statistics.
So performance collapses at small batch sizes (e.g., large models, high-res images, or limited GPU memory).
SALU solves this by learning an element-wise transformation. No mean/variance. No batch dependency.
Paper & Code:
๐ arXiv: https://t.co/eOiPC2fj1N
๐ป GitHub: https://t.co/B0dRtOVDkj
Looking for collaborators on:
โข Matched baselines & seeds
โข Large-batch trainingRTs appreciated! ๐ #DeepLearning#ComputerVision#ViT
The proof on CIFAR-100 ๐
ResNet-18 + BatchNorm? Accuracy collapses at batch size 1. ๐ฑ
SaluNet-C-18? Retains 76.23% Top-1 at batch size 1.
It also scales to ImageNet-1K:
โข ResNet-50: 80.16%
โข ViT-B/16: 80.28% (with only BS=128!)
The problem with BatchNorm/LayerNorm? They rely on batch statistics.
So performance collapses at small batch sizes (e.g., large models, high-res images, or limited GPU memory).
SALU solves this by learning an element-wise transformation. No mean/variance. No batch dependency.