Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Certification of Machine Learning Models via Directional Sharpness

About

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.

Gefei Tan, Adria Gascon, Sarah Meiklejohn, Mariana Raykova• 2026

Related benchmarks

TaskDatasetResultRank
Membership Inference AttackCIFAR-10 (test)
Attack AUC55.6
49
Generalization Gap Correlation AnalysisCIFAR-10 (test)
Runtime (s)0.21
6
Model DistinguishabilityCIFAR-10 (test)--
6
Generalization Gap PredictionCIFAR-10 and CIFAR-100 grid of 1,152 models (test)
Spearman ρ0.902
5
Model Quality AuditingCIFAR-10 (test)--
5
Zero-Knowledge Proof of Directional SharpnessGeneric dataset
Time (min)23.63
3
Zero-Knowledge Proof of TrainingGeneric dataset--
3
Model DistinguishabilityLanguage Model Dataset--
2
Showing 8 of 8 rows

Other info

Follow for update