AIToday

Researchers discover that normalization-free transformers achieve better gradient stability through subcritical signal propagation, offering insights into deep neural network initialization.

arXiv cs.LGApr 15, 20261 min read
Researchers discover that normalization-free transformers achieve better gradient stability through subcritical signal propagation, offering insights into deep neural network initialization.

Summaries like this, in your inbox every morning.

Sign up free →

3 Key Points

  1. Study analyzes signal propagation in transformers using averaged partial Jacobian norm (APJN) to measure gradient amplification across layers

  2. Theory extends to bidirectional attention and permutation-symmetric token configurations by deriving recurrence relations for activation statistics

  3. Pre-LayerNorm architectures show power-law APJN growth, while tanh-like nonlinearities exhibit stretched-exponential growth indicating subcritical behavior

  4. Findings apply to Dynamic Tanh (DyT) and Dynamic erf (Derf) transformers, explaining why these normalization-free designs achieve better training stability

  5. Predictions match empirical measurements in deep vision transformers, bridging theoretical understanding and practical deep network behavior

Discussion

No comments yet. Be the first to share your thoughts!

Log in to join the discussion

Related Articles

Stay ahead with AI news

Get curated AI news from 200+ sources delivered daily to your inbox. Free to use.

Get Started Free

Free · takes 30 seconds · unsubscribe anytime

1 minute a day. The AI essentials.

200+ sources · Email / LINE / Slack

Get it free →