Research Article

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

Table 11

Results of comparing the performance of different classifiers on the dataset integrated with the optimal subset.

ModelsClassifiersBest epochLossAccuracyPrecisionRecallF1 scores

Pretrained DistilBERTLinear311.0350.570.430.340.31
Fine-tuned DistilBERTLinear40.2310.930.880.870.87
Fine-tuned DistilBERTCNN40.2370.930.900.870.88
Fine-tuned DistilBERTLSTM40.2340.930.860.900.88