本文研究将通用漏洞披露(CVE)记录自动映射到通用弱点枚举(CWE)类别的文本分类方法。CVE-to-CWE映射是漏洞分析中的关键步骤,但目前主要依赖人工。作者将该任务建模为文本分类问题,比较了两种建模策略:多类分类(每个CVE预测一个CWE)和多标签分类(允许预测多个CWE)。实验采用了三种Transformer编码器(BERT Base、SecureBERT和CySecBERT),并在三个嵌套的标签空间(83、47和25个CWE类别)上进行评估。结果表明,多类训练在所有设置下均取得更高的宏F1分数,但随着标签空间缩小,与多标签的差距从21个百分点缩小到2个百分点。在多标签设置中,通过后处理阈值优化可在25类设置下缩小差距。混淆分析显示,主要的错误分类模式遵循CWE层次结构,且三个编码器的错误模式高度相似(Pearson相关系数r > 0.92),表明错误结构更多由分类法设计而非编码器选择驱动。层次宽松评估(允许家族内混淆)将宏F1从约81%提升至约90%,表明严格指标低估了分支级分类器质量。总体而言,CySecBERT在多标签设置中表现最佳,具有统计显著性。该研究为自动化CVE-to-CWE映射提供了方法论指导,并揭示了分类法结构对分类性能的影响。
💡 推荐理由: CVE-to-CWE映射是漏洞管理的基础,自动化可显著降低人工成本。本文揭示了分类法结构对映射错误的主导影响,对开发更鲁棒的自动映射工具具有指导意义。
🎯 建议动作: 研究跟进