Research Article

VulDistilBERT: A CPS Vulnerability Severity Prediction Method Based on Distillation Model

Table 7

Results of evaluating the raw dataset with the fine-tuned model.

ModelsBest epochLossAccuracyPrecisionRecallF1 scores

Fine-tuned BERT20.5800.770.750.620.66
Fine-tuned XLNET40.5700.780.740.650.68
Fine-tuned ROBERTA40.5660.780.760.650.68
Fine-tuned DistilBERT40.5710.780.760.680.70