The Normalized Maximum Likelihood for Regular Non-Smooth Models: Measure-Theoretic Foundations and Geometric Sampling
Not on the map. Why?
Abstract (arXiv)
The Normalized Maximum Likelihood (NML) codelength, or stochastic complexity, represents a principled criterion for universal coding. While recent coarea-based formulations provided a calculation method for smooth models, this framework collapses for the non-smooth estimators ubiquitous in modern machine learning (e.g., Lasso, Sparse SVMs). In this work, we provide a rigorous framework for computing the NML for regular path-differentiable Lipschitz (PDL) estimators. By applying classical geometric measure theory and bridging the coarea formula with conservative Jacobians, we prove that the stochastic complexity for non-smooth models is well-posed and theoretically consistent with the outputs of modern Automatic Differentiation. To compute this quantity exactly, we introduce the Propose-and-Project Metropolis-Hastings (PDL-PPMH) sampler, a geometric MCMC algorithm capable of traversing the non-differentiable level sets of the maximum likelihood estimator. We theoretically justify its components, including a stochastic tangent space proposal and a provably convergent non-smooth projection solver. We demonstrate the method's robustness by sampling from a high-dimensional Lasso posterior (P=2000), while simultaneously quantifying the computational scaling that governs the trade-off between exactness and mixing time. Crucially, we empirically demonstrate that our exact NML criterion provides a highly data-efficient alternative to cross-validation, achieving statistically indistinguishable predictive optima without requiring data splitting. Altogether, our work paves the way for the theoretical analysis of the NML codelength for regular non-smooth models.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 15 of its 43 references that are in the map. First posted on arXiv 23 May 2026.
Builds on
15 of 43 references.
References · 15
- 2019Conservative set valued fields, automatic differentiation, stochastic gradient method and deep learningcited by 71
- 2020Deep Learning is Singular, and That's Goodcited by 49
- 2020A mathematical model for automatic differentiation in machine learningcited by 44
- 2019Minimum Description Length Revisitedcited by 36
- 2025Symmetry in Neural Network Parameter Spacescited by 31
- 2018Gradient Sampling Methods for Nonsmooth Optimizationcited by 23
- 2021A survey of Monte Carlo methods for noisy and costly densities with application to reinforcement learning and ABCcited by 8
- 2022On the complexity of nonsmooth automatic differentiationcited by 4
- 2023Learning Capacity: A Measure of the Effective Dimensionality of a Modelcited by 4
- 2024Foundation of Calculating Normalized Maximum Likelihood for Continuous Probability Modelscited by 3
- 2026On randomized step sizes in Metropolis-Hastings algorithmscited by 2
- 2023A projected semismooth Newton method for a class of nonconvex composite programs with strong prox-regularitycited by 1
3 more
Built on by
Citing papers per year: 2026: 1.
1 paper.