atlas

Research map / Reinforcement learning / Visual and representation RL

BertRLFuzzer: A BERT and Reinforcement Learning Based Fuzzer (Student Abstract)

AAAI 2024 · Reinforcement learning › Visual and representation RL · approach: RL with tree search for discovery · 7 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Presents BertRLFuzzer, a fuzzer that uses a BERT model as the actor in a PPO reinforcement-learning loop, with a multi-armed bandit seed filter, to generate grammar-adhering attack vectors against web applications.

Problem

Finding security vulnerabilities (e.g., SQL injection and XSS) in web applications with grammar-adhering fuzzing, which is time-consuming to build by hand and must be redone when new vulnerability classes appear.

Approach

Pre-train a BERT model on attack-vector seed inputs so it learns the input grammar, then fine-tune it as the actor-critic PPO agent in an RL loop driven by victim-application rewards and penalties, with a multi-armed bandit choosing seed inputs.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use BertRLFuzzer instead of the nearest competing toolWeb application fuzzing · 9 victim websitesTFA: 54% lessHelped
Use BertRLFuzzerWeb application fuzzing · 9 victim websitesnew vulnerabilities found: 17 new vulnerabilitiesHelped
Use BertRLFuzzer instead of the nearest competing toolWeb application fuzzing · 9 victim websitesattack rate: 4.4% moreHelped
Replace a GRU-based DQN agent with a GRU-based PPO agent (ablation)BertRLFuzzer · 9 victim websitesTFA: 224 -> 197Helped
Replace a GRU-based DQN agent with an attention-based DQN agent (ablation)BertRLFuzzer · 9 victim websitesTFA: 224 -> 155Helped
Replace attention-based DQN with attention-based PPO (ablation)BertRLFuzzer · 9 victim websitesTFA: 155 -> 141Helped
Add improved reward signals (ablation)Attention based PPO · 9 victim websitesUF: 43 -> 48Helped
Add BERT pre-training (ablation)Attention based PPO with improved rewards · 9 victim websitesPP: 18% -> 5%Helped
Add MAB seed filtering (ablation)BertRLFuzzer · 9 victim websitesTFA: 118 -> 102Helped
Add MAB seed filtering (ablation)BertRLFuzzer · 9 victim websitesUF: 52 -> 52No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: over 7K injection attack vectors from public GitHub repositories

Benchmarks: benchmark of 9 victim websites (over 16K LOC)

Metrics: time to first attack unique fields vulnerabilities found parser penalties

Baselines: DeepFix DeepXSS DeepFuzz DQN fuzzer Baseline Grammar Mutator Baseline Random Mutator Baseline Random Fuzzer

Builds on: BERT PPO DeepFuzz DeepXSS DeepFix

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy