Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

About

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.

Suresh Kumar Amalapuram, Bikraj Shresta, Siva Ram murthy Chebiyam, Bheemarjuna Reddy Tamma, Sumohana S Channappayya• 2026

Related benchmarks

TaskDatasetResultRank
Malware DetectionBODMAS seen tasks
AUT (B)94.2
16
Malware DetectionBODMAS unseen tasks
AUT (B)95.6
8
Malware DetectionAPIGraph seen tasks
AUT (B)99.9
8
Malware DetectionAPIGraph overall tasks
AUT (B)99.9
8
Malware DetectionAndroZoo seen tasks
AUT (B)99.4
8
Malware DetectionAndroZoo unseen tasks
AUT (B)0.993
8
Malware DetectionAndroZoo overall tasks
AUT (B)99.3
8
Malware DetectionAPIGraph unseen tasks
AUT (B)99.8
8
Showing 8 of 8 rows

Other info

Follow for update