mlsanityData to 14 Oct 2026

Attention Saturation and Gradient Suppression at Inflection Layers: Diagnosing and Mitigating Bottlenecks in Transformer Adaptation

arXiv 2511.00797 ↗First posted 2 Nov 2025cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Pre-trained Transformers often exhibit over-confidence in source patterns and difficulty in forming new target-domain patterns during fine-tuning. We formalize the mechanism of output saturation leading to gradient suppression through standard cross-entropy and softmax analysis, showing that gradient suppression at inflection layers confines adaptation to high-level recombination of existing features while preventing low-level reconstruction. We introduce a set of layer-wise diagnostic metrics -- attention entropy (saturation proxy), activation gradient norm, parameter gradient norm, and Delta-CKA under a shared PCA basis -- to identify inflection layers characterized by both low attention entropy and steep gradient decay. Building on these findings, we propose a diagnose-first, inject-light fine-tuning strategy: selectively inserting LoRA adapters at inflection layers to restore suppressed backward signals with minimal parameter overhead. Experiments on BERT-base transfer from SST-2 to Rotten Tomatoes under under-trained and over-trained source regimes reveal that over-trained initialization benefits from inflection-layer LoRA injection, while under-trained initialization suffers performance degradation. When base features are strong, unblocking inflection layers facilitates high-level compositional adaptation; when base features are weak, full-pathway unblocking is required for low-level reconstruction, as supported by joint analysis of layer-wise activation gradients and Delta-CKA dynamics.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 8 of its 9 references that are in the map. First posted on arXiv 2 Nov 2025.

Builds on

8 of 9 references.

References · 8

  1. 2019Parameter-Efficient Transfer Learning for NLPcited by 4,236
  2. 2019Similarity of Neural Network Representations Revisitedcited by 1,755
  3. 2019Are Sixteen Heads Really Better than One?cited by 816
  4. 2020AdapterFusion: Non-Destructive Task Composition for Transfer Learningcited by 788
  5. 2019Revealing the Dark Secrets of BERTcited by 301
  6. 2020Gradient Starvation: A Learning Proclivity in Neural Networkscited by 267
  7. 2020On the Stability of Fine-tuning BERT: Misconceptions, Explanations, and Strong Baselinescited by 255
  8. 2020What Happens To BERT Embeddings During Fine-tuning?cited by 113

Built on by

Not cited yet.