R2Vul - Data and Models

This Zenodo repository contains the datasets and models for our paper "R2Vul: Learning to Reason about Software Vulnerabilities with Reinforcement Learning and Structured Reasoning Distillation". Datasets raw_dataset.json - the raw dataset mined from NVD. r2vul_dataset.zip - the dataset used for fine-tuning and testing. external_java_test.zip - the external, manually annotated Java dataset (RQ2). Models and Checkpoints cls.zip - includes all models checkpoints fine-tuned using CLS. sft.zip - includes all models checkpoints fine-tuned using SFT. orpo.zip - includes all models checkpoints fine-tuned using ORPO (R2Vul). MSIVD.zip - downloaded from https://zenodo.org/records/11403208 (codellama-13b - bigvul_expl). VulLLM.zip - downloaded from https://zenodo.org/records/10677069 (codellama-13b-multi-r16-2048).本Zenodo仓库收录了我们在论文《R2Vul：结合强化学习与结构化推理蒸馏的软件漏洞推理学习》中使用的数据集与模型。 ### 数据集 raw_dataset.json：从国家漏洞数据库（National Vulnerability Database, NVD）中挖掘得到的原始数据集。 r2vul_dataset.zip：用于模型微调与测试的数据集。 external_java_test.zip：用于研究问题2（RQ2）的外部人工标注Java数据集。 ### 模型与检查点 cls.zip：包含所有基于分类（Classification, CLS）微调得到的模型检查点文件。 sft.zip：包含所有基于监督微调（Supervised Fine-Tuning, SFT）微调得到的模型检查点文件。 orpo.zip：包含所有采用ORPO微调得到的模型检查点文件（对应R2Vul方法）。 MSIVD.zip：可从链接 https://zenodo.org/records/11403208 下载，对应模型为codellama-13b - bigvul_expl。 VulLLM.zip：可从链接 https://zenodo.org/records/10677069 下载，对应模型为codellama-13b-multi-r16-2048。