Machine Learning Techniques for Python Source Code Vulnerability Detection

Talaya Farasat, Joachim Posegga
4/15/2024
cs.SEcs.AI

No official code found. Best implementation: Tf-arch/Python-Source-Code-Vulnerability-Detection (10 stars, Unlicense, last commit 2024-03-29).

Abstract

Software vulnerabilities are a fundamental reason for the prevalence of cyber attacks and their identification is a crucial yet challenging problem in cyber security. In this paper, we apply and compare different machine learning algorithms for source code vulnerability detection specifically for Python programming language. Our experimental evaluation demonstrates that our Bidirectional Long Short-Term Memory (BiLSTM) model achieves a remarkable performance (average Accuracy = 98.6%, average F-Score = 94.7%, average Precision = 96.2%, average Recall = 93.3%, average ROC = 99.3%), thereby, establishing a new benchmark for vulnerability detection in Python source code.

View on arXivView PDF

Code Implementations(1)

Tf-arch/Python-Source-Code-Vulnerability-DetectionCommunity100%
Unlicense

Comparing Machine Learning Algorithms for Python Source Code Vulnerability Detection

103Python, Jupyter NotebookFeb 25, 20242 years agoUnlicense

Similar Papers

Harnessing the Power of LLMs in Source Code Vulnerability Detection

90% match2024

VulDetectBench: Evaluating the Deep Capability of Vulnerability Detection with Large Language Models

89% match1 repo382024

Harnessing Large Language Models for Software Vulnerability Detection: A Comprehensive Benchmarking Study

87% match2024

LLbezpeky: Leveraging Large Language Models for Vulnerability Detection

87% match2024

Explaining the Contributing Factors for Vulnerability Detection in Machine Learning

87% match2024

A Survey of Source Code Representations for Machine Learning-Based Cybersecurity Tasks

86% match2024

Cite this paper

@article{farasat2024machine,
  title  = {Machine Learning Techniques for Python Source Code Vulnerability Detection},
  author = {Talaya Farasat and Joachim Posegga},
  year   = {2024},
  eprint = {2404.09537},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2404.09537v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2404.09537.svg)](https://codeofpaper.com/paper/2404.09537)

Discussion