Search papers, labs, and topics across Lattice.
This paper introduces CREDIT, a novel certified ownership verification method for deep neural networks (DNNs) against model extraction attacks (MEAs). CREDIT leverages mutual information to quantify the similarity between a suspect model and the claimed original, establishing a practical verification threshold. The authors provide theoretical guarantees for ownership verification based on this threshold, demonstrating state-of-the-art performance across various datasets and tasks.
Finally, a defense against model extraction comes with rigorous theoretical guarantees, using mutual information to certify DNN ownership.
Machine Learning as a Service (MLaaS) has emerged as a widely adopted paradigm for providing access to deep neural network (DNN) models, enabling users to conveniently leverage these models through standardized APIs. However, such services are highly vulnerable to Model Extraction Attacks (MEAs), where an adversary repeatedly queries a target model to collect input-output pairs and uses them to train a surrogate model that closely replicates its functionality. While numerous defense strategies have been proposed, verifying the ownership of a suspicious model with strict theoretical guarantees remains a challenging task. To address this gap, we introduce CREDIT, a certified ownership verification against MEAs. Specifically, we employ mutual information to quantify the similarity between DNN models, propose a practical verification threshold, and provide rigorous theoretical guarantees for ownership verification based on this threshold. We extensively evaluate our approach on several mainstream datasets across different domains and tasks, achieving state-of-the-art performance. Our implementation is publicly available at: https://github.com/LabRAI/CREDIT.
