Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
The paper introduces SEED, a semi-supervised continual learning approach for malware detection designed to combat concept drift with limited labeled data. SEED uses singular value decomposition to project unlabeled samples into a representation space built from past data, pairing them with labeled samples to enforce representation consistency. Experiments on Windows and Android malware datasets demonstrate that SEED significantly outperforms a semi-supervised adaptation of hierarchical contrastive loss (HCL*) in detecting unseen malware, particularly on datasets lacking strong semantic structure, while using only 20% labeled data on seen tasks.
Even with just 20% labeled data, SEED boosts unseen malware detection by 40% compared to existing semi-supervised methods, proving semantic-structure-agnostic continual learning can dramatically improve security on a budget.
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.