arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.03024cs.SE
收藏

迈向改进的基于深度学习的漏洞检测

Toward Improved Deep Learning-based Vulnerability Detection

Adriana Sejfia, Satyaki Das, Saad Shafiq, Nenad Medvidović

AI总结:

本研究针对基于深度学习的漏洞检测器在跨基础单元漏洞(MBU)上的性能缺陷,通过实验验证了现有检测器在检测此类漏洞时准确率显著下降,并提出了一个框架以帮助正确纳入MBU漏洞。

摘要:

深度学习(DL)已成为近期多种漏洞检测技术中的共同主线。大型公开漏洞数据集的兴起推动了支撑这些技术的学习过程。虽然这些数据集有助于基于深度学习的漏洞检测器,但它们也限制了这些检测器的预测能力。这些数据集中的漏洞必须以某种方式表示,例如,漏洞所在的代码行、函数或程序切片。我们将这种表示称为基础单元。检测器学习基础单元如何可能易受攻击,然后预测其他基础单元是否易受攻击。我们假设,这种对单个基础单元的关注损害了检测器正确检测那些跨越多个基础单元(或MBU漏洞)的漏洞的能力。对于此类漏洞,当所有组成的基础单元都被检测为易受攻击时,才会发生正确的检测。验证现有技术在检测漏洞的所有部分方面的表现,对于确定它们在其他下游任务中的有效性至关重要。为了评估我们的假设,我们进行了一项研究,重点关注三个著名的基于深度学习的检测器:ReVeal、DeepWukong和LineVul。我们的研究表明,所有三个检测器的各自数据集中都包含MBU漏洞。此外,我们观察到在检测这些类型的漏洞时,准确率显著下降。我们展示了我们的研究和一个框架,该框架可用于帮助基于深度学习的检测器正确纳入MBU漏洞。

英文摘要:

Deep learning (DL) has been a common thread across several recent techniques for vulnerability detection. The rise of large, publicly available datasets of vulnerabilities has fueled the learning process underpinning these techniques. While these datasets help the DL-based vulnerability detectors, they also constrain these detectors' predictive abilities. Vulnerabilities in these datasets have to be represented in a certain way, e.g., code lines, functions, or program slices within which the vulnerabilities exist. We refer to this representation as a base unit. The detectors learn how base units can be vulnerable and then predict whether other base units are vulnerable. We have hypothesized that this focus on individual base units harms the ability of the detectors to properly detect those vulnerabilities that span multiple base units (or MBU vulnerabilities). For vulnerabilities such as these, a correct detection occurs when all comprising base units are detected as vulnerable. Verifying how existing techniques perform in detecting all parts of a vulnerability is important to establish their effectiveness for other downstream tasks. To evaluate our hypothesis, we conducted a study focusing on three prominent DL-based detectors: ReVeal, DeepWukong, and LineVul. Our study shows that all three detectors contain MBU vulnerabilities in their respective datasets. Further, we observed significant accuracy drops when detecting these types of vulnerabilities. We present our study and a framework that can be used to help DL-based detectors toward the proper inclusion of MBU vulnerabilities.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑