mlsanityData to 7 Oct 2026

VulStyle: A Multi-Modal Pre-Training for Code Stylometry-Augmented Vulnerability Detection

arXiv 2604.26313 ↗First posted 29 Apr 2026cs.CR cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

We present VulStyle, a multi-modal software vulnerability detection model that jointly encodes function-level source code, non-terminal Abstract Syntax Tree (AST) structure, and code stylometry (CStyle) features. Prior work in code representation primarily leverages token-level models or full AST trees, often missing stylistic cues indicative of risky programming practices, or incurring high structural overhead. Our approach selects only non-terminal AST nodes, reducing input complexity while preserving semantic hierarchy, and integrates syntactic and lexical CStyle features as auxiliary vulnerability signals. VulStyle is pre-trained using masked language modeling on 4.9M functions across seven programming languages, and fine-tuned across five benchmark datasets: Devign, BigVul, DiverseVul, REVEAL, and VulDeePecker. VulStyle achieves state-of-the-art performance on BigVul and VulDeePecker, improving F1 by 4-48% over strong transformer baselines, and attains competitive or best-average performance across all benchmarks. We contribute an ablation study isolating the effect of CStyle and AST structure, error case analysis, and a threat model situating the detection task in attacker-realistic scenarios.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 21 of its 37 references that are in the map. First posted on arXiv 29 Apr 2026.

Builds on

21 of 37 references.

References · 21

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
  4. 2015Neural Machine Translation of Rare Words with Subword Unitscited by 2,898
  5. 2020CodeBERT: A Pre-Trained Model for Programming and Natural Languagescited by 1,286
  6. 2021CodeXGLUE: A Machine Learning Benchmark Dataset for Code Understanding and Generationcited by 732
  7. 2019CodeSearchNet Challenge: Evaluating the State of Semantic Code Searchcited by 638
  8. 2020GraphCodeBERT: Pre-training Code Representations with Data Flowcited by 600
  9. 2022UniXcoder: Unified Cross-Modal Pre-training for Code Representationcited by 355
  10. 2019Devign: Effective Vulnerability Identification by Learning Comprehensive Program Semantics via Graph Neural Networkscited by 266
  11. 2020IntelliCode Compose: Code Generation Using Transformercited by 225
  12. 2020Deep Learning based Vulnerability Detection: Are We There Yet?cited by 142
9 more
  1. 2018Automated Vulnerability Detection in Source Code Using Deep Representation Learningcited by 114
  2. 2023DiverseVul: A New Vulnerable Source Code Dataset for Deep Learning Based Vulnerability Detectioncited by 98
  3. 2018SySeVR: A Framework for Using Deep Learning to Detect Software Vulnerabilitiescited by 91
  4. 2022VulBERTa: Simplified Source Code Pre-Training for Vulnerability Detectioncited by 51
  5. 2022LineVD: Statement-level Vulnerability Detection using Graph Neural Networkscited by 35
  6. 2020Exploring Software Naturalness through Neural Language Modelscited by 32
  7. 2024Top Score on the Wrong Exam: On Benchmarking in Machine Learning for Vulnerability Detectioncited by 27
  8. 2017Git Blame Who?: Stylistic Authorship Attribution of Small, Incomplete Source Code Fragmentscited by 7
  9. 2023PACE: A Program Analysis Framework for Continuous Performance Predictioncited by 3

Built on by

Not cited yet.