mlsanityData to 7 Oct 2026

SEED: Semi-supervised Continual MalwarE Detection for Tackling ConcEpt Drift on a BuDget

arXiv 2605.24903 ↗First posted 24 May 2026cs.CR cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Machine learning based malware detectors become obsolete over time due to concept drift in benign and malware applications. Recent methods rely on fully labeled data and use hierarchical contrastive loss (HCL) with active learning to improve robustness against drift by exploiting semantic structure in malware representations. However, obtaining labeled data in the security domain is difficult. Under partially labeled settings, HCL suffers significant performance degradation in detecting unseen malware, especially on datasets such as BODMAS where strong semantic structure may not exist. In this paper, we propose SEED, a semantic-structure-agnostic method for malware detection under limited supervision. SEED combines a tailored binary cross-entropy objective with semi-supervised continual learning and active learning. For partially labeled seen tasks, unlabeled samples are projected into a representation space constructed from previously seen data using singular value decomposition, and paired with suitable labeled samples to encourage representation consistency. For unseen tasks with fully unlabeled data, uncertainty is quantified using cosine distance in representation space, and the most uncertain samples are selected for analyst labeling. We evaluate SEED on both Windows and Android malware datasets. Using only 20% labeled data on seen tasks, SEED achieves average AUT improvements of 40% on BODMAS and 14% on AndroZoo for unseen malware detection compared to HCL* (the semi-supervised adaptation of HCL), while remaining competitive on APIGraph. Finally, we introduce a delayed buffer update strategy to reduce label noise propagation during replay and improve learning stability.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 21 of its 54 references that are in the map. First posted on arXiv 24 May 2026.

Builds on

21 of 54 references.

References · 21

  1. 2017Attention Is All You Needcited by 59,932
  2. 2016Overcoming catastrophic forgetting in neural networkscited by 4,718
  3. 2017Gradient Episodic Memory for Continual Learningcited by 1,871
  4. 2018Efficient Lifelong Learning with A-GEMcited by 1,044
  5. 2023A Comprehensive Survey of Continual Learning: Theory, Method and Applicationcited by 860
  6. 2019Gradient based sample selection for online continual learningcited by 568
  7. 2019Online Continual Learning with Maximally Interfered Retrievalcited by 376
  8. 2019Orthogonal Gradient Descent for Continual Learningcited by 361
  9. 2021Gradient Projection Memory for Continual Learningcited by 338
  10. 2021Open-World Semi-Supervised Learningcited by 165
  11. 2018EMBER: An Open Dataset for Training Static PE Malware Machine Learning Modelscited by 164
  12. 2018TESSERACT: Eliminating Experimental Bias in Malware Classification across Space and Timecited by 108
9 more
  1. 2020Dos and Don'ts of Machine Learning in Computer Securitycited by 98
  2. 2021ORDisCo: Effective and Efficient Usage of Incremental Unlabeled Data for Semi-supervised Continual Learningcited by 68
  3. 2020Transcending Transcend: Revisiting Malware Classification in the Presence of Concept Driftcited by 53
  4. 2024Gradient Reweighting: Towards Imbalanced Class-Incremental Learningcited by 37
  5. 2023Continuous Learning for Android Malware Detectioncited by 33
  6. 2022A Theoretical Analysis of the Learning Dynamics under Class Imbalancecited by 20
  7. 2015Reviewer Integration and Performance Measurement for Malware Detectioncited by 19
  8. 2022On the Limitations of Continual Learning for Malware Classificationcited by 12
  9. 2022AnoShift: A Distribution Shift Benchmark for Unsupervised Anomaly Detectioncited by 11

Built on by

Not cited yet.