Prior methods such as LAPE binarize each feed-forward neuron's activation (active if aj(x)>0) and score a neuron by
the entropy of its per-language activation probabilities rjk=P(aj(x)>0∣k). This discards negative activations
and treats each language independently. Distribution-aware Language Neuron (DLN) selection instead works with the whole
activation distribution:
- Activation distributions. For each MLP neuron j=(ℓ,n) in a gated (GLU) model, the activation is
aj(x)=[σ(Wgateℓx)]n.
We estimate the activation density pjk(v) of every language k∈L over the full range, including
negative values, with a 400-bin histogram over [μj−3σj, μj+3σj].
- Overlap matrix. For every pair of languages, compute the overlap coefficient
Sj[k,k′]=∫min(pjk(v),pjk′(v))dv∈[0,1].
- Clustering. Split the languages into two groups so that the largest overlap between the groups is as small
as possible (equivalent to single-linkage clustering with K=2); the smaller group is the candidate language set:
Aj∗=A⊂Largmink∈A,k′∈AcmaxSj[k,k′].
- Selection. Keep the neuron only if this maximum between-group overlap is at most a threshold τ,
set at the bottom P-th percentile of overlaps over all neurons. If ∣Aj∗∣=1 the neuron is a
single-language neuron (SLN); if ∣Aj∗∣>1 it is a multi-language neuron (MLN).
To test the causal effect of the selected neurons, we mean-patch them (replace each activation with its mean over the
non-target languages) and measure the change in negative log-likelihood (ΔNLL) and downstream accuracy on
the target language versus the other languages. Experiments use Llama-3.1-8B and SmolLM3-3B-Base, seven languages
(en, zh, fr, es, vi, id, ja), and two held-out corpora (FLORES+ devtest and a held-out Wikipedia split).