Using Large Language Models to Classify Some Vulnerabilities in Program Code

The paper studies the effectiveness of using large language models to detect common types of vulnerabilities in Python program code. In particular, using the technique of low-rank adaptation of (LoRA) models, fine-tuning of the CodeBERT-python model is performed. To train the models, we use the author's dataset, which consists of marked-up program code in Python. The trained models are used to detect and classify potential vulnerabilities. To evaluate the effectiveness of models, the number of false positives, false negatives, true positives and true negatives is determined. Also, accuracy, recall and F1-measures are calculated on a test data set for various configurations of model training macro parameters

Authors

Publication Details

Journal
PROGRAMMNAYA INGENERIA
Published
2024-09-09
DOI
https://doi.org/10.17587/prin.15.465-475
Primary Topic
Software Reliability and Analysis Research
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Using Large Language Models to Classify Some Vulnerabilities in Program Code

D. A. Kapustin, T. I. Shulika, V. V. Shvyrov, R.N. Sentyay
PROGRAMMNAYA INGENERIA
Software Reliability and Analysis Research
article

Using Large Language Models to Classify Some Vulnerabilities in Program Code

D. A. Kapustin, T. I. Shulika, V. V. Shvyrov, R.N. Sentyay
article en

Abstract

The paper studies the effectiveness of using large language models to detect common types of vulnerabilities in Python program code. In particular, using the technique of low-rank adaptation of (LoRA) models, fine-tuning of the CodeBERT-python model is performed. To train the models, we use the author's dataset, which consists of marked-up program code in Python. The trained models are used to detect and classify potential vulnerabilities. To evaluate the effectiveness of models, the number of false positives, false negatives, true positives and true negatives is determined. Also, accuracy, recall and F1-measures are calculated on a test data set for various configurations of model training macro parameters

PROGRAMMNAYA INGENERIAVol. 15(9)
Openalex Percentile: Top 85%
Software Reliability and Analysis Research
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.