AICRLGMar 5, 2024

Precise Extraction of Deep Learning Models via Side-Channel Attacks on Edge/Endpoint Devices

arXiv:2403.02870v13 citationsh-index: 30
Originality Incremental advance
AI Analysis

This addresses security vulnerabilities for companies deploying proprietary AI models on edge devices, though it appears incremental as it builds on existing side-channel attack research.

The paper tackles the problem of model extraction attacks on proprietary deep learning models deployed on edge/endpoint devices by analyzing how side-channel attacks can expose model information, showing that such attacks can achieve up to 5.8 times better performance when leveraging this information compared to having no prior knowledge.

With growing popularity, deep learning (DL) models are becoming larger-scale, and only the companies with vast training datasets and immense computing power can manage their business serving such large models. Most of those DL models are proprietary to the companies who thus strive to keep their private models safe from the model extraction attack (MEA), whose aim is to steal the model by training surrogate models. Nowadays, companies are inclined to offload the models from central servers to edge/endpoint devices. As revealed in the latest studies, adversaries exploit this opportunity as new attack vectors to launch side-channel attack (SCA) on the device running victim model and obtain various pieces of the model information, such as the model architecture (MA) and image dimension (ID). Our work provides a comprehensive understanding of such a relationship for the first time and would benefit future MEA studies in both offensive and defensive sides in that they may learn which pieces of information exposed by SCA are more important than the others. Our analysis additionally reveals that by grasping the victim model information from SCA, MEA can get highly effective and successful even without any prior knowledge of the model. Finally, to evince the practicality of our analysis results, we empirically apply SCA, and subsequently, carry out MEA under realistic threat assumptions. The results show up to 5.8 times better performance than when the adversary has no model information about the victim model.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes