Scaling up ZK for DNN Inference
As ML models have increased in capabilities and accuracy, so has the complexity of their deployments. Increasingly, ML model consumers are turning to service providers to serve the ML models in the ML-as-a-service (MLaaS) paradigm. As MLaaS proliferates, a critical requirement emerges: how can model consumers verify that the correct predictions were served, in the face of malicious, lazy, or buggy service providers? In this work, we present the first practical ImageNet-scale method to verify ML model inference non-interactively, i.e., after the inference has been done. To do so, we leverage recent developments in ZK-SNARKs (zero-knowledge succinct non-interactive argument of knowledge), a form of zero-knowledge proofs. ZK-SNARKs allows us to verify ML model execution non-interactively and with only standard cryptographic hardness assumptions. In particular, we provide the first ZK-SNARK proof of valid inference for a full resolution ImageNet model, achieving 79\% top-5 accuracy. We further use these ZK-SNARKs to design protocols to verify ML model execution in a variety of scenarios, including for verifying MLaaS predictions, verifying MLaaS model accuracy, and using ML models for trustless retrieval. Together, our results show that ZK-SNARKs have the promise to make verified ML model inference practical.
As ML models have increased in capabilities and accuracy, so has the complexity of their deployments. Increasingly, ML model consumers are turning to service providers to serve the ML models in the ML-as-a-service (MLaaS) paradigm. As MLaaS proliferates, a critical requirement emerges: how can model consumers verify that the correct predictions were served, in the face of malicious, lazy, or buggy service providers? In this work, we present the first practical ImageNet-scale method to verify ML model inference non-interactively, i.e., after the inference has been done. To do so, we leverage re
Explore this link on the map →related reading
- Machine learning and zero-knowledge proofs - a16z cryptoa16zcrypto.com
- ZKML Research Initiatives - HackMDhackmd.io
- Where Zero-Knowledge Machine Learning (zkML) Fits into the Bigger Picture of AIinsidejuice.substack.com
- An introduction to zero-knowledge machine learning (ZKML)worldcoin.org
- ZK Machine Learning - HackMDhackmd.io
- AI Model & API Providers Analysis | Artificial Analysisartificialanalysis.ai
- Sumcheck good. Lookups good. JOLT good. Particularly for zero-knowledge machine learning.blog.icme.io
- OPML: Optimistic Machine Learning on Blockchain - Layer 2 - Ethereum Researchethresear.ch
- Aleo | Zero-knowledge by designaleo.org
- [2211.04775] ZK-IMG: Attested Images via Zero-Knowledge Proofs to Fight Disinformationarxiv.org
- 17 misconceptions about SNARKs - a16z cryptoa16zcrypto.com
- Aleo | Zero-knowledge by designaleo.org