Develop a theory of middle-layer vulnerability and principled sparsity allocation
Develop a formal theoretical account of how internal representations evolve with depth, explain why middle layers exhibit heightened sensitivity to pruning, and use that account to derive principled rather than heuristic layer-wise sparsity allocations.
References
In Theorem~5, the authors provide an estimate for the accuracy of the resulting network approximation in terms of the fidelity of the retrained network. We hypothesize that a similar result holds in the deep-network case, which we leave for future research.
The normalization preserves the function only for a positively homogeneous activation, which OPT provides through ReLU; most recent LLMs use GELU or gated activations, for which the algorithm can still be run but neither the equivalence of Section~\ref{sec:theory} nor the invariance of Section~\ref{sec:exp-robust} applies, and whether the robustness carries over is untested.
Finally, while the middle-layer sensitivity finding is empirically robust, our framework does not yet explain it theoretically. A formal account of how the structure of internal representations evolves with depth, and how this motivates principled rather than heuristic sparsity allocation, remains an open problem.
One possible account is that removing a contiguous sub-block creates one residual-stream discontinuity whereas dispersed removal creates two. This explanation is untested; causal activation analysis and a larger factorial candidate set are needed.
A finer $k$-grid (e.g.\ $k \in {0,1,2,3,4,5,6,8}$) would reduce the granularity-induced noise in the fraction metric and is left as future work.