DetectBERT: Towards Full App-Level Representation Learning to Detect Android Malware

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Sun, Tiezhu, Daoudi, Nadia, Kim, Kisub, Allix, Kevin, Bissyandé, Tegawendé F., Klein, Jacques
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914928792698880
author Sun, Tiezhu
Daoudi, Nadia
Kim, Kisub
Allix, Kevin
Bissyandé, Tegawendé F.
Klein, Jacques
author_facet Sun, Tiezhu
Daoudi, Nadia
Kim, Kisub
Allix, Kevin
Bissyandé, Tegawendé F.
Klein, Jacques
contents Recent advancements in ML and DL have significantly improved Android malware detection, yet many methodologies still rely on basic static analysis, bytecode, or function call graphs that often fail to capture complex malicious behaviors. DexBERT, a pre-trained BERT-like model tailored for Android representation learning, enriches class-level representations by analyzing Smali code extracted from APKs. However, its functionality is constrained by its inability to process multiple Smali classes simultaneously. This paper introduces DetectBERT, which integrates correlated Multiple Instance Learning (c-MIL) with DexBERT to handle the high dimensionality and variability of Android malware, enabling effective app-level detection. By treating class-level features as instances within MIL bags, DetectBERT aggregates these into a comprehensive app-level representation. Our evaluation demonstrates that DetectBERT not only surpasses existing state-of-the-art detection methods but also adapts to evolving malware threats. Moreover, the versatility of the DetectBERT framework holds promising potential for broader applications in app-level analysis and other software engineering tasks, offering new avenues for research and development.
format Preprint
id arxiv_https___arxiv_org_abs_2408_16353
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle DetectBERT: Towards Full App-Level Representation Learning to Detect Android Malware
Sun, Tiezhu
Daoudi, Nadia
Kim, Kisub
Allix, Kevin
Bissyandé, Tegawendé F.
Klein, Jacques
Software Engineering
Artificial Intelligence
Cryptography and Security
Recent advancements in ML and DL have significantly improved Android malware detection, yet many methodologies still rely on basic static analysis, bytecode, or function call graphs that often fail to capture complex malicious behaviors. DexBERT, a pre-trained BERT-like model tailored for Android representation learning, enriches class-level representations by analyzing Smali code extracted from APKs. However, its functionality is constrained by its inability to process multiple Smali classes simultaneously. This paper introduces DetectBERT, which integrates correlated Multiple Instance Learning (c-MIL) with DexBERT to handle the high dimensionality and variability of Android malware, enabling effective app-level detection. By treating class-level features as instances within MIL bags, DetectBERT aggregates these into a comprehensive app-level representation. Our evaluation demonstrates that DetectBERT not only surpasses existing state-of-the-art detection methods but also adapts to evolving malware threats. Moreover, the versatility of the DetectBERT framework holds promising potential for broader applications in app-level analysis and other software engineering tasks, offering new avenues for research and development.
title DetectBERT: Towards Full App-Level Representation Learning to Detect Android Malware
topic Software Engineering
Artificial Intelligence
Cryptography and Security
url https://arxiv.org/abs/2408.16353