Supernodes and Halos: Loss-Critical Hubs in LLM Feed-Forward Layers
About
We study the organization of channel-level importance in transformer feed-forward networks (FFNs). Using a Fisher-style loss proxy (LP) based on activation-gradient second moments, we show that loss sensitivity is concentrated in a small set of channels within each layer. In Llama-3.1-8B, the top 1% of channels per layer accounts for a median of 58.7% of LP mass, with a range of 33.0% to 86.1%. We call these loss-critical channels supernodes. Although FFN layers also contain strong activation outliers, LP-defined supernodes overlap only weakly with activation-defined outliers and are not explained by activation power or weight norms alone. Around this core, we find a weaker but consistent halo structure: some non-supernode channels share the supernodes' write support and show stronger redundancy with the protected core. We use one-shot structured FFN pruning as a diagnostic test of this organization. At 50% FFN sparsity, baselines that prune many supernodes degrade sharply, whereas our SCAR variants explicitly protect the supernode core; the strongest variant, SCAR-Prot, reaches perplexity 54.8 compared with 989.2 for Wanda-channel. The LP-concentration pattern appears across Mistral-7B, Llama-2-7B, and Qwen2-7B, remains visible in targeted Llama-3.1-70B experiments, and increases during OLMo-2-7B pretraining. These results suggest that LLM FFNs develop a small learned core of loss-critical channels, and that preserving this core is important for reliable structured pruning.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Language Modeling | WikiText-2 (test) | PPL19.5 | 2333 | |
| Language Modeling | WikiText-2 | Perplexity (PPL)10.71 | 2320 | |
| Commonsense Reasoning | WinoGrande | Accuracy57 | 1442 | |
| Commonsense Reasoning | HellaSwag | HellaSwag Accuracy37 | 711 | |
| Question Answering | ARC Easy | -- | 597 | |
| Question Answering | OpenBookQA | Accuracy34 | 305 | |
| Reading Comprehension | BoolQ | Accuracy (BoolQ)66 | 228 | |
| Multi-task Language Understanding | MMLU | MMLU Score26 | 86 | |
| Question Answering | ARC Challenge | Accuracy26 | 50 | |
| Language Understanding | MMLU | MMLU Score30 | 40 |