arXiv · 2504.04520
Hessian of Perplexity for Large Language Models by PyTorch autograd (Open Source)
Abstract
Computing the full Hessian matrix -- the matrix of second-order derivatives for an entire Large Language Model (LLM) is infeasible due to its sheer size. In this technical report, we aim to provide a comprehensive guide on how to accurately compute at least a small portion of the Hessian for LLMs using PyTorch autograd library. We also demonstrate how to compute the full diagonal of the Hessian matrix using multiple samples of vector-Hessian Products (HVPs). We hope that both this guide and the accompanying GitHub code will be valuable resources for practitioners and researchers interested in better understanding the behavior and structure of the Hessian in LLMs.
Explore related subjects
Keep this discovery
Ivan Ilin. 2025-04-06. Hessian of Perplexity for Large Language Models by PyTorch autograd (Open Source). https://arxiv.org/abs/2504.04520
Cite the original work for its findings. Save a collection to share your selection of sources.