解决安卓恶意软件检测模型随时间失效的问题
TIF: Learning Temporal Invariance in Android Malware Detectors
- 按时间分组环境,用对比学习提取稳定特征
- 在十年数据上提升早期检测准确率,优于现有方法
- 适合需要长期稳定的恶意软件检测场景
基于学习的安卓恶意软件检测器会因变种和新家族导致的数据分布漂移而性能下降。本文系统分析了经验风险最小化(ERM)训练的分类器在面对此类分布变化时的挑战,指出其根本原因在于无法学习稳定的判别特征。不变性学习理论通过鼓励模型在不同环境中生成稳定表征来应对这一问题,但缺乏先验环境标签、漂移因素多样以及多家族带来的低质量表征使该任务复杂。为此,我们提出TIF——首个面向恶意软件检测的时间不变训练框架,旨在增强检测器在时间维度上的稳定性。TIF根据应用观察时间划分环境,结合专用多代理对比学习与不变梯度对齐,生成高质量且稳定的表征。TIF可无缝集成至任意学习型检测器中。在长达十年的数据集上实验表明,TIF在早期部署阶段表现尤为出色,满足实际需求并超越现有最优方法。
原文摘要 · Abstract (English)
Learning-based Android malware detectors degrade over time due to natural distribution drift caused by malware variants and new families. This paper systematically investigates the challenges classifiers trained with empirical risk minimization (ERM) face against such distribution shifts and attributes their shortcomings to their inability to learn \emph{stable} discriminative features. Invariant learning theory offers a promising solution by encouraging models to generate stable representations across environments that expose the instability of the training set. However, the lack of prior environment labels, the diversity of drift factors, and low-quality representations caused by diverse families make this task challenging. To address these issues, we propose TIF, the first temporal invariant training framework for malware detection, which aims to enhance the ability of detectors to learn stable representations across time. TIF organizes environments based on application observation dates to reveal temporal drift, integrating specialized multi-proxy contrastive learning and invariant gradient alignment to generate and align environments with high-quality, stable representations. TIF can be seamlessly integrated into any learning-based detector. Experiments on a decade-long dataset show that TIF excels, particularly in early deployment stages, addressing real-world needs and outperforming state-of-the-art methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。