Do You Trust Your Model? Emerging Malware Threats in the Deep Learning Ecosystem

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Hitaj, Dorjan, Pagnotta, Giulio, De Gaspari, Fabio, Ruko, Sediola, Hitaj, Briland, Mancini, Luigi V., Perez-Cruz, Fernando
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866912463385001984
author Hitaj, Dorjan
Pagnotta, Giulio
De Gaspari, Fabio
Ruko, Sediola
Hitaj, Briland
Mancini, Luigi V.
Perez-Cruz, Fernando
author_facet Hitaj, Dorjan
Pagnotta, Giulio
De Gaspari, Fabio
Ruko, Sediola
Hitaj, Briland
Mancini, Luigi V.
Perez-Cruz, Fernando
contents Training high-quality deep learning models is a challenging task due to computational and technical requirements. A growing number of individuals, institutions, and companies increasingly rely on pre-trained, third-party models made available in public repositories. These models are often used directly or integrated in product pipelines with no particular precautions, since they are effectively just data in tensor form and considered safe. In this paper, we raise awareness of a new machine learning supply chain threat targeting neural networks. We introduce MaleficNet 2.0, a novel technique to embed self-extracting, self-executing malware in neural networks. MaleficNet 2.0 uses spread-spectrum channel coding combined with error correction techniques to inject malicious payloads in the parameters of deep neural networks. MaleficNet 2.0 injection technique is stealthy, does not degrade the performance of the model, and is robust against removal techniques. We design our approach to work both in traditional and distributed learning settings such as Federated Learning, and demonstrate that it is effective even when a reduced number of bits is used for the model parameters. Finally, we implement a proof-of-concept self-extracting neural network malware using MaleficNet 2.0, demonstrating the practicality of the attack against a widely adopted machine learning framework. Our aim with this work is to raise awareness against these new, dangerous attacks both in the research community and industry, and we hope to encourage further research in mitigation techniques against such threats.
format Preprint
id arxiv_https___arxiv_org_abs_2403_03593
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Do You Trust Your Model? Emerging Malware Threats in the Deep Learning Ecosystem
Hitaj, Dorjan
Pagnotta, Giulio
De Gaspari, Fabio
Ruko, Sediola
Hitaj, Briland
Mancini, Luigi V.
Perez-Cruz, Fernando
Cryptography and Security
Artificial Intelligence
Training high-quality deep learning models is a challenging task due to computational and technical requirements. A growing number of individuals, institutions, and companies increasingly rely on pre-trained, third-party models made available in public repositories. These models are often used directly or integrated in product pipelines with no particular precautions, since they are effectively just data in tensor form and considered safe. In this paper, we raise awareness of a new machine learning supply chain threat targeting neural networks. We introduce MaleficNet 2.0, a novel technique to embed self-extracting, self-executing malware in neural networks. MaleficNet 2.0 uses spread-spectrum channel coding combined with error correction techniques to inject malicious payloads in the parameters of deep neural networks. MaleficNet 2.0 injection technique is stealthy, does not degrade the performance of the model, and is robust against removal techniques. We design our approach to work both in traditional and distributed learning settings such as Federated Learning, and demonstrate that it is effective even when a reduced number of bits is used for the model parameters. Finally, we implement a proof-of-concept self-extracting neural network malware using MaleficNet 2.0, demonstrating the practicality of the attack against a widely adopted machine learning framework. Our aim with this work is to raise awareness against these new, dangerous attacks both in the research community and industry, and we hope to encourage further research in mitigation techniques against such threats.
title Do You Trust Your Model? Emerging Malware Threats in the Deep Learning Ecosystem
topic Cryptography and Security
Artificial Intelligence
url https://arxiv.org/abs/2403.03593