English

Hessian Surgery: Class-Targeted Post-Hoc Rebalancing via Hessian Spike Perturbation

Machine Learning 2026-05-19 v2 Computer Vision and Pattern Recognition

Abstract

The Hessian spectrum of trained deep networks exhibits a characteristic structure: a continuous bulk of near-zero eigenvalues and a small number of large outlier eigenvalues (spikes), confirming the relevance of Random Matrix Theory in deep learning. The spike count matches the number of classes minus one. While prior work has described this structure, no method has exploited it operationally to improve classification performance. We propose Hessian Surgery, a post-hoc optimization method that directly perturbs model weights along spike eigenvectors to rebalance per-class accuracy without retraining. We introduce (i) a spike-class sensitivity matrix that quantifies the directional derivative of each class's accuracy along each spike eigenvector, (ii) a constrained optimization of perturbation coefficients that targets weak classes while preserving strong ones, and (iii) an adaptive amplitude control that raises or lowers the perturbation budget based on iteration-level improvement signals. We obtain encouraging results on CIFAR-10 and ISIC-2019 on both balanced accuracy and standard deviation.

Keywords

Cite

@article{arxiv.2605.07790,
  title  = {Hessian Surgery: Class-Targeted Post-Hoc Rebalancing via Hessian Spike Perturbation},
  author = {Hugo Vigna and Samuel Bontemps},
  journal= {arXiv preprint arXiv:2605.07790},
  year   = {2026}
}

Comments

The code is available here: https://github.com/hugovigna/hessian-surgery.git