Large Language Models for In-File Vulnerability Localization Can Be "Lost in the End"

TL;DR

This study evaluates how well chat-based large language models detect file-based vulnerabilities, revealing a significant drop in performance for vulnerabilities near the end of large files and proposing input size adjustments to improve detection accuracy.

Contribution

The paper provides the first comprehensive analysis of LLM effectiveness in in-file vulnerability detection, highlighting the 'lost-in-the-end' effect and strategies for optimizing input size.

Findings

LLMs underperform in detecting vulnerabilities near file ends

Adjusting input size improves detection recall by over 37%

Vulnerability location and input size significantly influence LLM performance

Abstract

Recent advancements in artificial intelligence have enabled processing of larger inputs, leading everyday software developers to increasingly rely on chat-based large language models (LLMs) like GPT-3.5 and GPT-4 to detect vulnerabilities across entire files, not just within functions. This new development practice requires researchers to urgently investigate whether commonly used LLMs can effectively analyze large file-sized inputs, in order to provide timely insights for software developers and engineers about the pros and cons of this emerging technological trend. Hence, the goal of this paper is to evaluate the effectiveness of several state-of-the-art chat-based LLMs, including the GPT models, in detecting in-file vulnerabilities. We conducted a costly investigation into how the performance of LLMs varies based on vulnerability type, input size, and vulnerability location within the file. To give enough statistical power to our study, we could only focus on the three most common (as well as dangerous) vulnerabilities: XSS, SQL injection, and path traversal. Our findings indicate that the effectiveness of LLMs in detecting these vulnerabilities is strongly influenced by both the location of the vulnerability and the overall size of the input. Specifically, regardless of the vulnerability type, LLMs tend to significantly (p < .05) underperform when detecting vulnerabilities located toward the end of larger files, a pattern we call the 'lost-in-the-end' effect. Finally, to further support software developers and practitioners, we also explored the optimal input size for these LLMs and presented a simple strategy for identifying it, which can be applied to other models and vulnerability types. Eventually, we show how adjusting the input size can lead to significant improvements in LLM-based vulnerability detection, with an average recall increase of over 37% across all models.