Robust Machine Learning for Malware Detection over Time
Published in In the proceedings of Proceedings of the Italian Conference on Cybersecurity (ITASEC 2022), Rome, Italy, June 20-23, 2022, 2022
Recommended citation: Daniele Angioni, Luca Demetrio, Maura Pintor, Battista Biggio, "Robust Machine Learning for Malware Detection over Time." In the proceedings of Proceedings of the Italian Conference on Cybersecurity (ITASEC 2022), Rome, Italy, June 20-23, 2022, 2022. http://ceur-ws.org/Vol-3260/paper12.pdf
Abstract:
The presence and persistence of Android malware is an on-going threat that plagues this information era, and machine learning technologies are now extensively used to deploy more effective detectors that can block the majority of these malicious programs. However, these algorithms have not been developed to pursue the natural evolution of malware, and their performances significantly degrade over time because of such concept-drift. Currently, state-of-the-art techniques only focus on detecting the presence of such drift, or they address it by relying on frequent updates of models. Hence, there is a lack of knowledge regarding the cause of the concept drift, and ad-hoc solutions that can counter the passing of time are still under-investigated. In this work, we commence to address these issues as we propose (i) a drift-analysis framework to identify which characteristics of data are causing the drift, and (ii) SVM-CB, a time-aware classifier that leverages the drift-analysis information to slow down the performance drop. We highlight the efficacy of our contribution by comparing its degradation over time with a state-of-the-art classifier, and we show that SVM-CB better withstands the distribution changes that naturally characterize the malware domain. We conclude by discussing the limitations of our approach and how our contribution can be taken as a first step towards more time-resistant classifiers that not only tackle, but also understand the concept drift that affects data.
BibTeX:
@inproceedings{Angioni22Robust,
author = {Angioni, Daniele and Demetrio, Luca and Pintor, Maura and Biggio, Battista},
editor = {Demetrescu, Camil and Mei, Alessandro},
title = {Robust Machine Learning for Malware Detection over Time},
booktitle = {Proceedings of the Italian Conference on Cybersecurity {(ITASEC} 2022), Rome, Italy, June 20-23, 2022},
series = {CEUR Workshop Proceedings},
volume = {3260},
pages = {169–180},
publisher = {CEUR-WS.org},
year = {2022},
url = {http://ceur-ws.org/Vol-3260/paper12.pdf},
timestamp = {Sat, 05 Nov 2022 00:03:03 +0100},
biburl = {https://dblp.org/rec/conf/itasec/AngioniDPB22.bib},
bibsource = {dblp computer science bibliography, https://dblp.org}
}