ML Security PapersDemystifying the Role of Rule-based Detection in AI Systems for Windows Malware Detection
Andrea Ponte, Luca Demetrio, Luca Oneto et al. · University of Genova · RINA Consulting +1 more
Defends ML malware detectors against adversarial PE evasion by training only on YARA-undetected samples, improving robustness and reducing attack surface
Malware detection increasingly relies on AI systems that integrate signature-based detection with machine learning. However, these components are typically developed and combined in isolation, missing opportunities to reduce data complexity and strengthen defenses against adversarial EXEmples, carefully crafted programs designed to evade detection. Hence, in this work we investigate the influence that signature-based detection exerts on model training, when they are included inside the training pipeline. Specifically, we compare models trained on a comprehensive dataset with an AI system whose machine learning component is trained solely on samples not already flagged by signatures. Our results demonstrate improved robustness to both adversarial EXEmples and temporal data drift, although this comes at the cost of a fixed lower bound on false positives, driven by suboptimal rule selection. We conclude by discussing these limitations and outlining how future research could extend AI-based malware detection to include dynamic analysis, thereby further enhancing system resilience.