Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Reasoning with LLMs for Zero-Shot Vulnerability Detection

Created by
  • Haebom
Category
Empty

저자

Arastoo Zibaeirad, Marco Vieira

개요

본 논문은 점점 더 복잡해지는 소프트웨어 시스템에서 소프트웨어 취약점 탐지(SVD) 자동화의 어려움을 해결하기 위해, C/C++로 개발된 대규모 오픈소스 시스템 소프트웨어 프로젝트에서 수집한 데이터셋과 종합적인 평가 프레임워크인 VulnSage를 제시합니다. 기존 데이터셋과 달리, 휴리스틱 노이즈 사전 필터링과 LLM 기반 추론을 결합하여 대표적이고 노이즈가 최소화된 취약점 스펙트럼을 확보합니다. VulnSage는 함수, 파일, 함수 간 상호작용 등 다양한 수준에서 분석을 지원하며, Baseline, Chain-of-Thought, Think, Think & Verify 네 가지 제로샷 프롬프트 전략을 사용합니다. 평가 결과, 구조화된 추론 프롬프트가 LLM 성능을 크게 향상시키며, 특히 Think & Verify는 모호한 응답을 20.3%에서 9.1%로 줄이고 정확도를 높였음을 보여줍니다. 또한 코드 특화 모델이 범용 모델보다 성능이 우수하며, 취약점 유형에 따라 성능 차이가 크게 나타나 모든 보안 상황에서 단일 접근 방식이 우수한 것은 아님을 보여줍니다. 데이터셋과 코드는 GitHub에서 제공됩니다.

시사점, 한계점

시사점:
LLM 기반 소프트웨어 취약점 탐지 자동화의 성능 향상을 위한 새로운 평가 프레임워크와 데이터셋 제공.
구조화된 추론 프롬프트(Think & Verify)가 LLM의 정확성과 모호성 감소에 효과적임을 실증.
코드 특화 모델이 범용 모델보다 소프트웨어 취약점 탐지에 더 효과적임을 확인.
다양한 취약점 유형에 대한 LLM 성능의 차이를 분석하여 향후 연구 방향 제시.
한계점:
평가에 사용된 데이터셋이 C/C++ 기반 오픈소스 프로젝트에 한정됨. 다른 언어나 소프트웨어 유형으로의 일반화 가능성에 대한 추가 연구 필요.
LLM 성능이 취약점 유형에 따라 크게 달라지는 점은 모든 유형의 취약점 탐지에 대한 단일 솔루션의 어려움을 시사. 더욱 정교한 모델 및 프롬프트 엔지니어링 전략 개발 필요.
휴리스틱 노이즈 사전 필터링의 정확성 및 신뢰도에 대한 추가적인 검증 필요.
👍