English

Block removal for large language models through constrained binary optimization

Machine Learning 2026-02-03 v1 Artificial Intelligence Computation and Language Quantum Physics

Abstract

Compressing resource-intensive large language models by removing whole transformer blocks is a seemingly simple idea, but identifying which blocks to remove constitutes an exponentially difficult combinatorial problem. In this paper, we formulate block removal as a constrained binary optimization problem that can be mapped to a physical system (Ising model), whose energies are a strong proxy for downstream model performance. This formulation enables an efficient ranking of a large number of candidate block-removal configurations and yields many high-quality, non-trivial solutions beyond consecutive regions. We demonstrate that our approach outperforms state-of-the-art block-removal methods across several benchmarks, with performance gains persisting after short retraining, and reaching improvements of up to 6 points on the MMLU benchmark. Our method requires only forward and backward passes for a few active parameters, together with an (at least approximate) Ising solver, and can be readily applied to any architecture. We illustrate this generality on the recent NVIDIA-Nemotron-3-Nano-30B-A3B-FP8 model, which exhibits a highly inhomogeneous and challenging block structure.

Keywords

Cite

@article{arxiv.2602.00161,
  title  = {Block removal for large language models through constrained binary optimization},
  author = {David Jansen and Roman Rausch and David Montero and Roman Orus},
  journal= {arXiv preprint arXiv:2602.00161},
  year   = {2026}
}

Comments

7 pages, 5 figures

R2 v1 2026-07-01T09:28:31.581Z