Evaluating and Improving the Robustness of Security Attack Detectors Generated by LLMs
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2024
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866916953732415488 |
|---|---|
| author | Pasini, Samuele Kim, Jinhan Aiello, Tommaso Lozoya, Rocio Cabrera Sabetta, Antonino Tonella, Paolo |
| author_facet | Pasini, Samuele Kim, Jinhan Aiello, Tommaso Lozoya, Rocio Cabrera Sabetta, Antonino Tonella, Paolo |
| contents | Large Language Models (LLMs) are increasingly used in software development to generate functions, such as attack detectors, that implement security requirements. A key challenge is ensuring the LLMs have enough knowledge to address specific security requirements, such as information about existing attacks. For this, we propose an approach integrating Retrieval Augmented Generation (RAG) and Self-Ranking into the LLM pipeline. RAG enhances the robustness of the output by incorporating external knowledge sources, while the Self-Ranking technique, inspired by the concept of Self-Consistency, generates multiple reasoning paths and creates ranks to select the most robust detector. Our extensive empirical study targets code generated by LLMs to detect two prevalent injection attacks in web security: Cross-Site Scripting (XSS) and SQL injection (SQLi). Results show a significant improvement in detection performance while employing RAG and Self-Ranking, with an increase of up to 71%pt (on average 37%pt) and up to 43%pt (on average 6%pt) in the F2-Score for XSS and SQLi detection, respectively. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2411_18216 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Evaluating and Improving the Robustness of Security Attack Detectors Generated by LLMs Pasini, Samuele Kim, Jinhan Aiello, Tommaso Lozoya, Rocio Cabrera Sabetta, Antonino Tonella, Paolo Software Engineering Cryptography and Security Machine Learning Large Language Models (LLMs) are increasingly used in software development to generate functions, such as attack detectors, that implement security requirements. A key challenge is ensuring the LLMs have enough knowledge to address specific security requirements, such as information about existing attacks. For this, we propose an approach integrating Retrieval Augmented Generation (RAG) and Self-Ranking into the LLM pipeline. RAG enhances the robustness of the output by incorporating external knowledge sources, while the Self-Ranking technique, inspired by the concept of Self-Consistency, generates multiple reasoning paths and creates ranks to select the most robust detector. Our extensive empirical study targets code generated by LLMs to detect two prevalent injection attacks in web security: Cross-Site Scripting (XSS) and SQL injection (SQLi). Results show a significant improvement in detection performance while employing RAG and Self-Ranking, with an increase of up to 71%pt (on average 37%pt) and up to 43%pt (on average 6%pt) in the F2-Score for XSS and SQLi detection, respectively. |
| title | Evaluating and Improving the Robustness of Security Attack Detectors Generated by LLMs |
| topic | Software Engineering Cryptography and Security Machine Learning |
| url | https://arxiv.org/abs/2411.18216 |