The shortest version: the paper’s robust principle survived contact with GPT-2, but its most elaborate machinery did not earn the win. That is not a failed story. It is a more useful one.
Neural network compression tries to make a model smaller by removing computation it appears not to need. The danger is that “not needed” is usually measured on a calibration sample—and a calibration sample can be very good at representing what is common while barely showing what is rare.
The intuition
The rare musician.
Imagine compressing an orchestra after listening to rehearsals. One musician plays constantly. Another enters only once, during a quiet coda—but without that note, the ending collapses.
If you rank musicians by their average sound across rehearsal, the rare player looks expendable. That judgment is locally reasonable and globally wrong. The sample has confused infrequency with redundancy.
The rare line is short. Its direction is still irreplaceable.
DR-FSP starts from this concern: a unit that looks redundant on common inputs may carry a computational direction that a rare environment relies on. Its central proposal is to judge importance across named environments, not only in the mixture average.
There is one extra fairness step. Prose, dialogue, and code can naturally produce outputs at different scales, so the paper measures each environment’s reconstruction error relative to that environment’s own original output. In plain language: what fraction of this room’s behavior did compression fail to reproduce?
Three rooms. Two ways to grade the repair.
The frequent rooms dominate the mixture, so a serious code failure can look small in the overall grade.
The mechanism
What a neuron actually contributes.
A hidden unit has two parts to its job. First, it fires: it recognizes something in the current input. Then it writes: it pushes a direction into the next part of the network.
Looking at only one half can mislead. A unit can activate loudly but write almost nothing useful. Another can fire rarely yet write a direction no surviving unit can reproduce.
The inputs that make the unit respond.
The direction it adds to the layer output.
The functional direction available after compression.
The multiplication is easier to picture than it looks. Take one vertical tape showing how strongly the neuron fired on each example. Multiply it by one horizontal arrow showing where the neuron writes. The result is a sheet: the same write direction repeated for every example, louder or quieter according to the activation tape.
One neuron leaves a sheet of evidence.
Stack one contribution sheet for each environment and the unit’s job becomes visible across prose, dialogue, and code.
Frequent activation makes this sheet easy to see in an average calibration sample.
The distinction
Choosing is not the same as repairing.
Most compression pipelines contain two decisions that are easy to blur together. Selection chooses which neurons remain. Refitting adjusts the outgoing weights of those survivors so they reproduce the old layer as closely as possible.
Refitting can repair coefficients. It cannot generally invent an activation direction that selection already deleted. Once the rare musician has left the building, turning up the remaining microphones does not recreate the missing note.
- 01Selectiondefines →
- 02Available basisconstrains →
- 03Refit ceilingsets →
- 04Worst-case riskobserved
Robust refitting helps the fragile subset, but it cannot synthesize the activation direction that average-weighted selection removed.
The first controlled experiment made that separation unusually clean. Across 20 seeds, there were six coordinate decisions per seed—120 opportunities to retain the stable direction. Source-only and prevalence-weighted selection kept it 0 out of 120 times. Worst-environment selection kept it 120 out of 120 times and matched an exact search over 729 candidate subsets.





