Refinando o mapeamento de CVE para CWE com atenção aprimorada em modelos baseados em BERT

Neste estudo, apresentamos o CySecBERT-ARD, uma abordagem avançada para classificar vulnerabilidades de software que mapeia Vulnerabilidades e Exposições Comuns (CVE) para Enumerações de Fraquezas Comuns (CWE). Nossa abordagem é usar um modelo baseado em transformer pré-treinado, o CySecBERT, adaptado para contextos de cibersegurança; o modelo é aprimorado com atenção aditiva e codificação de posição relativa, que permitem uma compreensão mais profunda das descrições de vulnerabilidades de CVE ao capturar as relações contextuais. Nossa abordagem alcança uma impressionante precisão de 91,34% e uma pontuação F1 de 91,32% durante a fase de avaliação e teste em comparação com os modelos base. Os resultados demonstram o potencial do CySecBERT-ARD em melhorar a eficiência e a eficácia da classificação de vulnerabilidades.