Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Audrey Cherilyn, Houman Safaai• 2026

Related benchmarks

TaskDatasetResultRank
Language ModelingWikiText-2 (test)
PPL19.5
2333
Language ModelingWikiText-2
Perplexity (PPL)10.71
2320
Commonsense ReasoningWinoGrande
Accuracy57
1442
Commonsense ReasoningHellaSwag
HellaSwag Accuracy37
711
Question AnsweringARC Easy--
597
Question AnsweringOpenBookQA
Accuracy34
305
Reading ComprehensionBoolQ
Accuracy (BoolQ)66
228
Multi-task Language UnderstandingMMLU
MMLU Score26
86
Question AnsweringARC Challenge
Accuracy26
50
Language UnderstandingMMLU
MMLU Score30
40
Showing 10 of 15 rows

Other info

Follow for update