Optimized Deep Learning Models for Malware Detection under Concept Drift

William Maillet, Benjamin Marais
8/21/2023
cs.CRcs.AIcs.LG

No confident code implementation found for this paper.

Abstract

Despite the promising results of machine learning models in malicious files detection, they face the problem of concept drift due to their constant evolution. This leads to declining performance over time, as the data distribution of the new files differs from the training one, requiring frequent model update. In this work, we propose a model-agnostic protocol to improve a baseline neural network against drift. We show the importance of feature reduction and training with the most recent validation set possible, and propose a loss function named Drift-Resilient Binary Cross-Entropy, an improvement to the classical Binary Cross-Entropy more effective against drift. We train our model on the EMBER dataset, published in2018, and evaluate it on a dataset of recent malicious files, collected between 2020 and 2023. Our improved model shows promising results, detecting 15.2% more malware than a baseline model.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Label-efficient Training Updates for Malware Detection over Time

87% match2026

Counteracting Concept Drift by Learning with Future Malware Predictions

87% match2024

Adversarial Evasion in Non-Stationary Malware Detection: Minimizing Drift Signals through Similarity-Constrained Perturbations

87% match2026

Combating Concept Drift with Explanatory Detection and Adaptation for Android Malware Classification

87% match1 repo232024

ADAPT: A Pseudo-labeling Approach to Combat Concept Drift in Malware Detection

86% match1 repo22025

MORPH: Towards Automated Concept Drift Adaptation for Malware Detection

85% match2024

Cite this paper

@article{maillet2023optimized,
  title  = {Optimized Deep Learning Models for Malware Detection under Concept Drift},
  author = {William Maillet and Benjamin Marais},
  year   = {2023},
  eprint = {2308.10821},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2308.10821v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2308.10821.svg)](https://codeofpaper.com/paper/2308.10821)

Discussion