本研究提出了一种基于支持向量机(Support Vector Machine)的创新方法,结合范德瓦尔登秩分数增强特征注意力机制,旨在解决垃圾邮件数据高维稀疏的挑战,提高垃圾邮件检测的分类性能。
Method Article
本研究提出了一种基于支持向量机(Support Vector Machine)的创新方法,结合范德瓦尔登秩分数增强特征注意力机制,旨在解决垃圾邮件数据高维稀疏的挑战,提高垃圾邮件检测的分类性能。
随着电子邮件使用量的扩大,垃圾邮件已成为一项严峻挑战,威胁着网络安全并降低了通信效率。传统的检测方法面临持续的局限性:传统的机器学习模型通常难以处理高维稀疏数据,而深度学习则需要大量的计算资源。
本研究引入了 Van der Waerden 等级分数特征注意力增强支持向量机 (VWR-Attn-SVM) 来解决这些问题。该方法应用 Van der Waerden 秩变换来规范化文本特征,提高针对异常值的鲁棒性并保留序数关系。增强的注意力机制通过具有正则化的非线性处理进一步优化特征选择,突出显示与垃圾邮件检测最相关的特征。
在UCI Spambase和印尼垃圾邮件数据集上的实验表明,VWR-Attn-SVM在准确率、精密度、召回率、F1分数和AUC方面优于传统分类器。通过将高性能与降低的计算成本相结合,该方法为垃圾邮件分类提供了一种高效且可解释的解决方案,并有可能扩展到其他基于文本的平台,例如消息传递和社交媒体。
在互联网和数字技术快速发展的当代数字时代,尽管即时通讯和社交媒体平台不断出现和创新,但电子邮件仍然是电子交易和企业通信领域不可或缺的基石1。它超越时空界限的能力赋予其独特的优势,可以随时在全球范围内进行无缝通信。然而,这种广泛的采用引发了一个紧迫且有害的问题——垃圾邮件的猖獗传播。恶意行为者利用电子邮件系统作为分发大量未经请求的商业广告、恶意软件和非法内容的工具。根据研究,从 2012 年到 2023 年,全球垃圾邮件在电子邮件总流量中的比例猛增了 7700%2,3。垃圾邮件的泛滥不仅严重扰乱了用户的正常电子邮件作,还带来了多方面的威胁。它通过可能暴露敏感信息来破坏个人隐私,通过数据泄露和恶意软件感染的风险危及企业安全,甚至通过促进欺诈活动来破坏经济秩序 4,5。有效的垃圾邮件分类可将与网络钓鱼相关的经济损失减少 40-60%6,凸显了高效、准确的过滤方法的实用价值。因此,开发高效、准确的垃圾邮件检测模型已成为确保网络安全和提高效率的重要研究领域。
关于垃圾邮件检测的现有研究的大量内容都集中在机器学习和深度学习方法上。在传统机器学习领域,已经探索和应用了多种技术。基于规则的方法,例如决策树7,已被用于根据从数据特征派生的预定义规则做出分类决策。将多个弱学习器聚合为强学习器的提升方法8、9、10 和处理数据不确定性和不精确性的粗糙集合理论11 也显示出潜力。此外,包括逻辑回归、K最近邻(KNN)12,13、朴素贝叶斯14,15,16和SVM 17,18,19在内的统计方法已被广泛采用。这些方法通常依赖于 TF-IDF 等传统特征提取方法。虽然 TF-IDF 可以有效地量化文档中单词的重要性,但它很难捕捉电子邮件文本中固有的复杂语义关系和上下文细微差别。此外,当面对电子邮件特征空间中典型的高维和稀疏数据时,这些方法经常会遇到计算瓶颈。它们有限的鲁棒性可能导致在训练过程中陷入局部最优解,从而严重限制模型的分类精度和泛化能力。
深度学习凭借其卓越的自动特征提取能力,已成为垃圾邮件检测的强大替代方案。卷积神经网络(CNN)20,21,22,循环神经网络(RNN)23和长短期记忆网络(LSTM)24,25等算法,以及最近基于Transformer的模型,如Word2vec和BERT26,27,在提高分类性能方面取得了重大进展。CNN 擅长从数据中提取局部特征,RNN 和 LSTM 可以很好地处理顺序数据,捕获文本中的时间依赖关系,基于 Transformer 的模型擅长挖掘复杂的语义关系和上下文信息。最近有效的NLP方法,例如基于TinyML的文本分类器28,为垃圾邮件分类提供了强大的基线。TinyML 模型针对内存有限的边缘设备进行了优化。我们在结果部分将我们的方法与这些方法进行了比较,强调了准确性、计算效率和部署灵活性之间的权衡。然而,这些深度学习模型有其自身的一系列局限性。它们通常需要大量的训练参数,导致计算资源需求高并延长训练时间。与传统 SVM 相比,BERT 等深度学习模型需要多 3-5 倍的内存和 10 倍长的训练时间29,这使得它们不太适合资源受限的环境。这使得它们在资源受限的环境中部署不太实用,例如移动设备或低端服务器。此外,它们复杂的架构通常使它们的可解释性降低,这在理解模型决策过程至关重要的应用程序中可能是一个重大缺点。
在此背景下,本研究的总体目标是开发一种创新方法,克服现有方法的局限性,有效应对垃圾邮件数据高维和稀疏性带来的挑战。所提出的 Van der Waerden Rank Score Feature Attention-Enhanced SVM (VWR-Attn-SVM) 代表了旨在增强垃圾邮件检测性能的技术的新颖集成(图 1)。VWR-Attn-SVM 背后的基本原理在于其独特的设计,结合了多个组件的优势。

图 1:使用 VWR-Attn-SVM 进行垃圾邮件分类的总体研究流程图。 该流程图展示了基于Van der Waerden等级得分和特征注意力增强SVM的垃圾邮件分类工作流程,涵盖了数据准备(加载、拆分、预处理)、实验准备、TF-IDF特征标签统计相关性验证、基于注意力增强SVM的垃圾邮件检测和多分类器比较。 请点击此处查看此图的大图。
核心增强功能注意力机制处理具有特定维度的单个电子邮件样本。通过应用 Van der Waerden 秩变换,它将因异常词频而扭曲的电子邮件文本特征归一化为类似标准正态分布的形式。这种转变显着增强了模型的稳健性,使其能够更好地处理电子邮件数据的可变性。Van der Waerden 秩分数优于对数缩放和分位数变换,原因有三:(1) 对垃圾邮件特征异常值(例如极端词频)具有鲁棒性,这与放大低频噪声的对数缩放不同;(2) 保留特征序号关系(对于垃圾邮件指标层次结构(如“免费”与“获胜”)至关重要,而分位数转换则扁平化分布;(3)归一化为[0,1],简化注意力机制整合,确保权重一致(图2)。

图 2:实验流程图。 (A-C) 垃圾邮件分类工作流程,涵盖数据处理、特征选择、模型训练、评估以及有/没有 Van der Waerden 等级分数转换的比较。 请点击此处查看此图的大图。
在结构上,该机制具有用于非线性特征变换的两层全连接网络(图 2)。第一层配备了 LeakyReLU 激活函数,在引入非线性的同时减小了输入维度,并结合了 Dropout 层以减轻过度拟合。第二层使用 Sigmoid 函数输出注意力权重,可以精确量化每个特征的重要性。模型中集成了 L1/L2 正则化策略以优化特征选择,其中 L1 正则化促进稀疏性,有效筛选出相关性较低的特征,而 L2 正则化通过约束权重的大小来防止过度拟合。在训练阶段,采用多任务学习框架,结合特征重建损失和分类损失来优化模型参数。这使得 VWR-Attn-SVM 能够精确地适应电子邮件文本的高维、稀疏 TF-IDF 特征,这是电子邮件内容复杂性的特征。
我们的方法针对数千到一万个基于文本的垃圾邮件数据集(例如,Spambase、印度尼西亚垃圾邮件数据集(补充文件 1))进行了优化,并且需要标准计算资源(Intel Core i7 处理器、16 GB RAM)进行训练;推理可以在标准笔记本电脑(Intel Core i5,8 GB RAM)上运行,延迟为亚秒级。主要限制包括对非文本垃圾邮件(例如图像嵌入垃圾邮件)的性能有限以及对结构化文本功能的依赖。与现有的替代技术相比,VWR-Attn-SVM具有几个显着的优势。与传统的机器学习方法不同,它不仅仅依赖于基本的特征提取,而是通过增强的注意力机制主动学习根据重要性对特征进行加权,从而更好地捕获与垃圾邮件分类更相关的特征。与深度学习模型相比,它在性能和计算效率之间实现了良好的平衡。它需要更少的计算资源和更短的训练时间,使其更适合广泛的应用,尤其是那些资源有限的应用。这种创新方法不仅适用于电子邮件系统中垃圾邮件检测的特定任务,而且还有可能扩展到其他基于文本的通信渠道,例如即时通讯应用程序、社交媒体平台和短信服务,这些渠道存在类似的有害和恶意内容传播问题。总体而言,VWR-Attn-SVM 代表了垃圾邮件检测领域的重大进步,提供了更实用、更高效和多功能的解决方案来解决数字通信领域长期存在的垃圾邮件问题。
Access restricted. Please log in or start a trial to view this content.
1. 实验准备(补充文件 2 和补充文件 3)
表 1:数据集统计和特征定义摘要。 此表列出了垃圾邮件分类的变量,包括词频 (word_freq_WORD)、字符频数 (char_freq_CHAR)、大写运行长度指标和目标类变量,并描述了每种变量类型和含义。 请按此下载此表。
2. 验证 TF-IDF 特征与标签之间统计关联的实验(补充文件 2 和补充文件 3)
3. 用于垃圾邮件检测的注意力增强型 SVM 分类(补充文件 2 和补充文件 3)


R、k×d 和 k=64(隐藏神经元)。
Rd×k 和 R
k 是每个特征的注意力权重。选择 Sigmoid 而不是 SoftMax,以保持多个特征重要性的独立性。
表示元素乘法。


4. 多个分类器的比较(补充文件 2 和补充文件 3)
5. 不同分类器在训练/测试时间和内存方面的多指标性能对比图(补充文件 4)
6. CNN、RNN、LSTM或变压器的实验结果(补充文件5)
7. 补充代码说明
Access restricted. Please log in or start a trial to view this content.
首先,根据既定的实验方案, 图1 概述了本研究的总体流程图。 图2,依次描绘了实验2的作流程图。此外, 表 1 主要显示垃圾邮件数据集中的单词和字符频率,spam.csv。
在模型性能评估方面,采用了准确率、精确度、召回率、F1 分数和受试者工作特征曲线下面积 (AUC) 五个关键指标。 表 2 定义了真阳性 (TP)、假阳性 (FP)、真阴性 (TN) 和假阴性 (FN) 的概念。F1 分数是一种精确度和召回率的谐波手段,用于平衡分类性能的这两个关键方面。受试者工作特征 (ROC) 曲线,假阳性率 (FPR) 绘制在 x 轴上,真阳性率 (TPR) 绘制在 y 轴上,提供了跨各种决策阈值的分类性能的全面可视化。因此,量化 ROC 曲线下面积的 AUC 成为评估二元分类器有效性的关键指标。
表2:分类绩效评价指标
Access restricted. Please log in or start a trial to view this content.
本研究基于Spambase数据集验证了VWR-Attn-SVM的有效性,为解决垃圾邮件数据的高维和稀疏性质提供了见解。实验表明,垃圾邮件数据中只有少数特征与标签有很强的相关性;传统模型对所有特征一视同仁,导致性能较差,而该模型的注意力机制可以动态地对关键特征进行加权。在集成范德瓦尔登(VWR)秩变换后,该模型实现了更快的损失收敛、更强的泛化、平衡的特征权重,并捕获了更多的交互信息。它在测试集上表现出出色的分类指标,优于传统方法,同时节省资源。其创新在于解决了传统机器学习和深度学习的固有问题,为文本分类提供了新的范式,并具有良好的可解释性,能够适应资源受限的场景。
实验作的关键步骤
本研究实验作中的几个关键步骤对垃圾邮件分类的结果产生了显著影响。在数据准备方面,从UCI机器学习存储库中选择了垃圾邮件数据集,该数据集具有4601个实例、57个连续特征和二进制类标签,奠定了坚实的基础。验证TF-IDF特征与标签之间的统计关联的过程至关重要。采用卡方检验进行特征筛选,并通过热图可视化...
Access restricted. Please log in or start a trial to view this content.
作者没有需要披露的利益冲突。
感谢福建省数学联盟(资助号:2023SXLMMS10)和福建省自然科学基金(2023J05083、2022J011396、2023J011434)对这项工作的资助。
Access restricted. Please log in or start a trial to view this content.
| Name | Company | Catalog Number | Comments |
|---|---|---|---|
| 补充档案2:code_new.py;补充文件3:code_indonesian.py. | |||
| 数字 | NumPy 开发者 | Python 数值计算库 | |
| 熊猫 | 熊猫开发团队 | 数据作与分析库 | |
| Matplotlib | Matplotlib 开发者 | 用于创建静态、动画和交互可视化的库 | |
| 海生 | 迈克尔·瓦斯科姆等人。 | 基于 matplotlib 的统计数据可视化库 | |
| Scikit-Learn | scikit-learn 开发团队 | 机器学习库,包含多种分类、回归和聚类算法 | |
| 张量流 | 谷歌 | 开源机器学习框架,包括用于构建神经网络的Keras API | |
| IMBLEARN | imbalanced-learn 开发者 | 处理不平衡数据集的库,包括用于过采样的SMOTE | |
| 警告 | Python 标准库 | 用于发出警告消息的模块 | |
| <强>补充档案4:code_compute_time.py | |||
| 数字 | NumPy 开发者 | Python 数值计算库 | |
| 熊猫 | 熊猫开发团队 | 数据作与分析库 | |
| Matplotlib | Matplotlib 开发者 | 用于创建图表和图形的可视化库 | |
| 海生 | 迈克尔·瓦斯科姆等人。 | 基于 matplotlib 的统计数据可视化库 | |
| Scikit-Learn | scikit-learn 开发团队 | 带有分类、回归和预处理工具的机器学习库 | |
| 张量流 | 谷歌 | 开源机器学习框架,采用Keras API用于神经网络 | |
| IMBLEARN | imbalanced-learn 开发团队 | 处理不平衡数据集的库(包括SMOTE) | |
| 警告 | Python 标准库 | 用于发出警告消息的模块 | |
| 时间 | Python 标准库 | 时间相关函数模块 | |
| 普苏蒂尔 | 贾姆保罗·罗多拉 | 用于检索系统信息和监控资源使用情况的库 | |
| 操作系统 | Python 标准库 | 用于与作系统交互的模块 | |
| 补充文件5:DNN.py. | |||
| 熊猫 | 熊猫开发团队 | 数据作与分析库 | |
| 数字 | NumPy 开发者 | Python 数值计算库 | |
| 时间 | Python 标准库 | 时间相关函数模块 | |
| 普苏蒂尔 | 贾姆保罗·罗多拉 | 系统信息检索和资源监控库 | |
| Matplotlib | Matplotlib 开发者 | 用于创建图表和图形的可视化库 | |
| Scikit-Learn | scikit-learn 开发团队 | 具备数据预处理、模型选择和度量工具的机器学习库 | |
| IMBLEARN | imbalanced-learn 开发团队 | 处理不平衡数据集的库(包括SMOTE) | |
| 张量流 | 谷歌 | 基于Keras API构建神经网络的开源机器学习框架 |
Access restricted. Please log in or start a trial to view this content.
Request permission to reuse the text or figures of this JoVE article
Request Permission