PHÁT HIỆN LỖ HỔNG PHẦN MỀM DỰA TRÊN PHƯƠNG PHÁP XỬ LÝ NGÔN NGỮ TỰ NHIÊN

Tóm tắt

Đảm bảo chất lượng phần mềm là ưu tiên cao nhất trong toàn bộ quá trình phát triển phần mềm. Bài báo đề xuất một phương pháp kết hợp dựa trên 3 kỹ thuật xử lý chính: Word2vec kết hợp LSTM; Word2vec kết hợp Bidirectional LSTM; Phát hiện lỗ hổng phần mềm sử dụng BERT. Trong đó Word2vec kết hợp LSTM, BiLSTM thực hiện việc chuyển mã đã được xử lý thành các vectơ nhúng. Các vectơ nhúng được đưa vào mạng LSTM và BiLSTM với số lượng các nút ẩn khác nhau để tiến hành học và phân loại lỗ hổng. Mô hình BERT biễu diễn từ theo cơ chế hai chiều sử dụng kỹ thuật Transformer, các attention mechanism dùng để hiểu ngữ cảnh từ cả hai phía của các từ trong câu. Kết quả phân loại đưa qua hàm softmax để trả về một vectơ chứa dự đoán xác suất từng loại lỗ hổng. Để minh họa sự vượt trội của phương pháp, tiến hành thực nghiệm trên bộ dữ liệu SARD. Kết quả thực nghiệm cho độ chính xác lên đến 82,63% tương ứng tỷ lệ bỏ sót chỉ còn 17,37% cho thấy hiệu quả tốt.