Search papers, labs, and topics across Lattice.
This paper introduces directional sharpness, a novel metric for certifying machine learning model generalization that addresses the limitations of existing metrics which can be misleading under training perturbations. The authors provide empirical and analytical evidence showing that directional sharpness correlates more strongly with actual generalization performance and reliably identifies poorly generalizing models. Additionally, it offers efficient computability for model auditing, even allowing for certification through zero-knowledge proofs without disclosing training data.
Directional sharpness outperforms traditional metrics in predicting model generalization, making it a game-changer for model certification.
In machine learning, model certification has been identified as an important method for gaining assurance about a model's trustworthiness and quality. A model's quality is largely determined by its ability to generalize, i.e., to perform well on data beyond what it was trained on. It is not possible to certify generalization directly, however, as it depends on unknown data and is not directly measurable. Proxies such as test accuracy can be misleading when the training process is perturbed (intentionally or accidentally), and metrics such as sharpness -- which has an empirically supported link to generalization -- are computationally expensive and can also serve as unreliable signals when training deviates from a prescribed procedure. In this work, we propose directional sharpness, a metric designed to efficiently and reliably indicate generalization despite potential training deviations. We provide empirical and analytical evidence that directional sharpness (1) correlates more strongly with generalization than existing metrics and (2) identifies models with poor generalization more reliably than existing metrics. Furthermore, directional sharpness is efficiently computable in model auditing settings, where the verifier has access to training data, and via zero-knowledge proofs that certify quality without revealing training data.