SecCodePRM: A Process Reward Model for Code Security

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yu, Weichen, Mangal, Ravi, Luo, Yinyi, Hu, Kai, He, Jingxuan, Pasareanu, Corina S., Fredrikson, Matt
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915790788231168
author Yu, Weichen
Mangal, Ravi
Luo, Yinyi
Hu, Kai
He, Jingxuan
Pasareanu, Corina S.
Fredrikson, Matt
author_facet Yu, Weichen
Mangal, Ravi
Luo, Yinyi
Hu, Kai
He, Jingxuan
Pasareanu, Corina S.
Fredrikson, Matt
contents Large Language Models are rapidly becoming core components of modern software development workflows, yet ensuring code security remains challenging. Existing vulnerability detection pipelines either rely on static analyzers or use LLM/GNN-based detectors trained with coarse program-level supervision. Both families often require complete context, provide sparse end-of-completion feedback, and can degrade as code length grows, making them ill-suited for real-time, prefix-level assessment during interactive coding and streaming generation. We propose SecCodePRM, a security-oriented process reward model that assigns a context-aware, step-level security score along a code trajectory. To train the model, we derive step-level supervision labels from static analyzers and expert annotations, allowing the model to attend more precisely to fine-grained regions associated with inter-procedural vulnerabilities. SecCodePRM has three applications: full-code vulnerability detection (VD), partial-code VD, and secure code generation (CG). For VD, SecCodePRM uses risk-sensitive aggregation that emphasizes high-risk steps; for CG, SecCodePRM supports inference-time scaling by ranking candidate continuations and favoring higher cumulative reward. This design yields dense, real-time feedback that scales to long-horizon generation. Empirically, SecCodePRM outperforms prior approaches in all three settings, while preserving code functional correctness, suggesting improved security without a safety-utility tradeoff.
format Preprint
id arxiv_https___arxiv_org_abs_2602_10418
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle SecCodePRM: A Process Reward Model for Code Security
Yu, Weichen
Mangal, Ravi
Luo, Yinyi
Hu, Kai
He, Jingxuan
Pasareanu, Corina S.
Fredrikson, Matt
Cryptography and Security
Software Engineering
Large Language Models are rapidly becoming core components of modern software development workflows, yet ensuring code security remains challenging. Existing vulnerability detection pipelines either rely on static analyzers or use LLM/GNN-based detectors trained with coarse program-level supervision. Both families often require complete context, provide sparse end-of-completion feedback, and can degrade as code length grows, making them ill-suited for real-time, prefix-level assessment during interactive coding and streaming generation. We propose SecCodePRM, a security-oriented process reward model that assigns a context-aware, step-level security score along a code trajectory. To train the model, we derive step-level supervision labels from static analyzers and expert annotations, allowing the model to attend more precisely to fine-grained regions associated with inter-procedural vulnerabilities. SecCodePRM has three applications: full-code vulnerability detection (VD), partial-code VD, and secure code generation (CG). For VD, SecCodePRM uses risk-sensitive aggregation that emphasizes high-risk steps; for CG, SecCodePRM supports inference-time scaling by ranking candidate continuations and favoring higher cumulative reward. This design yields dense, real-time feedback that scales to long-horizon generation. Empirically, SecCodePRM outperforms prior approaches in all three settings, while preserving code functional correctness, suggesting improved security without a safety-utility tradeoff.
title SecCodePRM: A Process Reward Model for Code Security
topic Cryptography and Security
Software Engineering
url https://arxiv.org/abs/2602.10418