이 논문은 LLM 기반 소프트웨어 취약점 탐지에서 소스 코드를 어떻게 표현하느냐가 성능에 큰 영향을 미친다는 점에 착안해, 추상구문트리(AST)보다 간결하게 제어흐름·조건·실행 동작을 인코딩하는 행동트리(BT)를 대안적 중간 표현으로 제시한다. 연구진은 Java 소스코드를 AST로 파싱한 뒤 BT로 변환하는 전처리 단계를 제안하고, Juliet Java 테스트 스위트의 460개 샘플에 대해 원본 소스코드·AST·BT 세 가지 표현으로 양자화된 로컬 LLM인 Mistral Small 3.2 24B의 탐지 성능을 비교했다. 짧은 코드 샘플에서는 BT 표현이 재현율을 높였고 원본 소스코드는 정밀도가 더 높았으며, 긴 샘플에서는 BT가 원본 표현보다 전반적 성능이 우수하면서도 컨텍스트 창에 들어맞은 반면 다수의 AST는 컨텍스트 한계를 초과했다. 이는 BT가 양자화된 로컬 LLM 기반 취약점 탐지에 유용한 압축 구조 표현이 될 수 있음을 시사한다.
- •AST보다 간결한 행동트리(BT)를 LLM 기반 취약점 탐지의 대안 중간 표현으로 제안한다.
- •Java 소스코드를 AST로 파싱한 뒤 BT로 변환하는 전처리 파이프라인을 구축했다.
- •Juliet 테스트 스위트 460개 샘플에서 원본 코드·AST·BT 세 표현의 탐지 성능을 비교했다.
- •짧은 샘플에서는 BT가 재현율을, 원본 코드가 정밀도를 각각 더 높였다.
- •긴 샘플에서는 다수 AST가 컸텍스트 한계를 초과한 반면 BT는 한계 내에서 더 나은 성능을 보였다.
0단 자동
AI가 규칙대로 쓰고 그대로 게시했습니다. 사람이 따로 보지 않았습니다.
- 규칙 판
- 규칙 판 도입 이전 기사입니다.
- 남기는 것
- 규칙 판 · 모델 · 시각
- 판 기록
- 아직 없습니다.
Towards Behavior Tree-Guided Vulnerability Detection with Lightweight LLMs
- 1.LLM 기반 취약점탐지에서 소스코드 표현방식이 성능 좌우, AST(추상구문트리)는 장황해 컸텍스트창 초과 문제 지적
- 2.대안으로 행동트리(BT) 제안: AST보다 압축적으로 인코딩, Java를 AST거쳐 BT로 변환하는 전처리 구현
- 3.Juliet Java 460개 샘플, 원본코드·AST·BT를 양자화 로컬 LLM으로 비교
- 4.짧은 코드는 BT가 재현율 우위, 긴 샘플은 BT가 컸텍스트창에 들어가며 전체 성능 개선
왜 중요한가?
온디바이스·로컬 배포되는 경량 양자화 LLM으로 취약점을 탐지할 때 컨텍스트창 제약이 실질적 병목인데, 행동트리 표현이 이를 완화하면서 정확도까지 개선할 수 있음을 보여 자원제약 환경의 보안 자동화에 실용적 대안을 제시한다.
소프트웨어 취약점 탐지는 국내 기업들에게도 중요한 보안 과제이며, LLM의 활용 가능성에 대한 관심이 높습니다. 이 연구는 경량 LLM을 활용한 취약점 탐지 효율성을 높이는 방법을 제시하여, 국내 개발 환경에서 보다 빠르고 정확한 보안 검증 도구 개발에 기여할 수 있습니다. 이는 국내 소프트웨어 산업의 보안 수준을 한층 높이는 데 중요한 기술적 진보를 의미합니다.
본문 미리보기
arXiv:2609.01758v1 Announce Type: new Abstract: Large Language Models (LLMs) are increasingly used for software vulnerability detection, but their performance depends on how source code is represented in the input. Most prompting approaches use source code in its original form, while some works propose the use of structured representations. Abstract Syntax Trees (ASTs) are one of the most popular approaches, but AST verbosity increases input size relative to source code, making them hard to fit
전체 내용이 궁금하다면?
원문을 직접 읽어보세요
이 글이 만들어진 과정
- 11:24AI 초안
![[AI리더의 서가] Agentic AI 구축 개발의 모든 것](https://cdn.aitimes.com/news/photo/202610/215958_219938_244.jpg)
