一句话:提出 MalWhiteout,用 Confident Learning、集成学习与应用间关系降低 Android 恶意软件数据集标签噪声。
机器学习驱动的 Android 恶意软件检测依赖可靠的数据集,但现有数据集多由 VirusTotal 等杀毒服务标注,容易产生误标,进而影响检测技术的评估准确性。大规模去除标签噪声十分困难。本文提出 MalWhiteout,将先进的噪声估计方法 Confident Learning 引入 Android 恶意软件检测领域,并针对其可能引入的假阳性,结合集成学习和应用间关系,以增强噪声检测的鲁棒性。作者在一个精心构建的大规模可靠基准数据集上评估 MalWhiteout。实验表明,即使数据集噪声率高达 30%,MalWhiteout 检测标签噪声的准确率仍超过 94%;在不同设置下,其效果比现有最先进方法提升 8% 至 218%,效率快 70 至 249 倍。通过减少标签噪声,现有恶意软件检测方法的性能也能得到提升。
Machine learning based Android malware detection has attracted a great deal of research work in recent years. A reliable malware dataset is critical to evaluate the effectiveness of malware detection approaches. Unfortunately, existing malware datasets used in our community are mainly labelled by leveraging existing anti-virus services (i.e., VirusTotal), which are prone to mislabelling. This, however, would lead to the inaccurate evaluation of the malware detection techniques. Removing label noises from Android malware datasets can be quite challenging, especially at a large data scale. To address this problem, we propose an effective approach called MalWhiteout to reduce label errors in Android malware datasets. Specifically, we creatively introduce Confident Learning (CL), an advanced noise estimation approach, to the domain of Android malware detection. To combat false positives introduced by CL, we incorporate the idea of ensemble learning and inter-app relation to achieve a more robust capability in noise detection. We evaluate MalWhiteout on a curated large-scale and reliable benchmark dataset. Experimental results show that MalWhiteout is capable of detecting label noises with over 94% accuracy even at a high noise ratio (i.e., 30%) of the dataset. MalWhiteout outperforms the state-of-the-art approach in terms of both effectiveness (8% to 218% improvement) and efficiency (70 to 249 times faster) across different settings. By reducing label noises, we show that the performance of existing malware detection approaches can be improved.