Learn to Reverse DNNs from AI Programs Automatically
Learn to Reverse DNNs from AI Programs Automatically
Simin Chen, Hamed Khanpour, Cong Liu, Wei Yang
Proceedings of the Thirty-First International Joint Conference on Artificial Intelligence
Main Track. Pages 666-672.
https://doi.org/10.24963/ijcai.2022/94
With the privatization deployment of DNNs on edge devices, the security of on-device DNNs has raised significant concern. To quantify the model leakage risk of on-device DNNs automatically, we propose NNReverse, the first learning-based method which can reverse DNNs from AI programs without domain knowledge. NNReverse trains a representation model to represent the semantics of binary code for DNN layers. By searching the most similar function in our database, NNReverse infers the layer type of a given function’s binary code. To represent assembly instructions semantics precisely, NNReverse proposes a more fine-grained embedding model to represent the textual and structural-semantic of assembly functions.
Keywords:
AI Ethics, Trust, Fairness: Trustworthy AI
Natural Language Processing: Embeddings
AI Ethics, Trust, Fairness: Societal Impact of AI