atlas

Research map / Architectures and efficiency / RNNs and normalization

DEEP NEURAL NETWORK INITIALIZATION WITH SPARSITY INDUCING ACTIVATIONS

ICLR 2024 · Architectures and efficiency › RNNs and normalization · approach: Theory of depth and width · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Shows that shifted-ReLU and soft-threshold sparsifying activations are untrainable at the edge of chaos in deep nets due to an unstable variance-map fixed point, and proposes magnitude-clipped variants (CReLU, CST) that restore stability and train with up to 85% activation sparsity.

Problem

Sparsifying activations (shifted ReLU and soft thresholding) that induce high fractional sparsity at initialization fail to train in very deep DNNs and CNNs, because at the edge of chaos their Gaussian process variance map has an unstable fixed point.

Approach

Clip the activation magnitude at level m (CReLU_{tau,m}, CST_{tau,m}); this makes V'(q*)<1 while keeping chi_1=1 at the EoC, so the variance fixed point is locally stable, enabling trainable deep sparse networks at up to 85% sparsity.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add magnitude clipping to shifted ReLU100-layer DNN · MNISTtest accuracy: 0.10 -> 0.92Helped
add magnitude clipping to SoftThreshold100-layer DNN · MNISTtest accuracy: 0.10 -> 0.92Helped
use shifted ReLU at s=0.60 instead of standard ReLU100-layer DNN · MNISTtest accuracy: 0.94 -> 0.76Hurt
increase CReLU clipping magnitude from m=0.81 to m=1.17 (ablation)100-layer DNN · MNIST, s=0.85test accuracy: 0.78 -> 0.94Helped
increase CReLU clipping magnitude from m=1.17 to m=1.74 (ablation)100-layer DNN · MNIST, s=0.85test accuracy: 0.94 -> 0.28Hurt
increase CST clipping magnitude from m=0.67 to m=1.00 (ablation)100-layer DNN · MNIST, s=0.85test accuracy: 0.76 -> 0.93Helped
increase CST clipping magnitude from m=1.00 to m=1.53 (ablation)100-layer DNN · MNIST, s=0.85test accuracy: 0.93 -> 0.14Hurt
increase CST clipping magnitude from m=0.72 to m=1.06 (ablation)100-layer DNN · MNIST, s=0.80test accuracy: 0.92 -> 0.95Helped
increase CST clipping magnitude from m=1.06 to m=1.61 (ablation)100-layer DNN · MNIST, s=0.80test accuracy: 0.95 -> 0.11Hurt
use clipped activations at s=0.70 rather than unclipped thresholded activations50-layer CNN · CIFAR10test accuracy: 0.1 -> 0.70Helped
use clipped activations at s=0.8550-layer CNN · CIFAR10test accuracy: 0.63 -> 0.66Mixed

Surprising findings

Limitations

Experimental setup

Datasets: MNIST CIFAR10

Benchmarks: MNIST CIFAR10

Metrics: test accuracy test sparsity V'(q*) V''(q*)

Baselines: ReLU ReLU_tau ST_tau CReLU_{tau,m} CST_{tau,m}

Builds on: edge of chaos initialization dynamical isometry FATReLU

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy