09 基于对抗训练和自注意力机制的恶意域名检测

刘心怡,郭树育,赵沁泽

• 1:内蒙古科技大学数智产业学院

• 2:内蒙古第一机械有限集团

• 
  

摘要(Abstract)：

针对已有的恶意域名检测方法特征工程复杂、效率低，提出一种融合对抗训练和自注意力机制的恶意域名检测方法。首先将域名序列进行字符级的表征；然后，使用双向长短期记忆网络自动的从输入中提取最佳的特征表示，并通过自注意力机制捕获域名文本中重要的信息；最后，将提取的关键信息送入分类器进行分类。另一方面，通过在输入层添加扰动来生成对抗样本扩充训练数据集，从而提高模型的鲁棒性和分类效果。实验结果表明：与现有的方法比，所提模型在各项指标上都取得了较好的效果。

关键词(KeyWords)： 恶意域名;双向长短期记忆网络;自注意力机制;对抗训练

基金项目(Foundation): 内蒙古自治区直属高校基本科研业务费项目(2024QNJS081)

作者(Author): 刘心怡,郭树育,赵沁泽

DOI: 10.16559/j.cnki.2095-2295.2024.04.009