mlsanityData to 14 Oct 2026

You Only Train Once: A Flexible Training Framework for Code Vulnerability Detection Driven by Vul-Vector

arXiv 2506.10988 ↗First posted 12 Mar 2025cs.SE cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

With the pervasive integration of computer applications across industries, the presence of vulnerabilities within code bases poses significant risks. The diversity of software ecosystems coupled with the intricate nature of modern software engineering has led to a shift from manual code vulnerability identification towards the adoption of automated tools. Among these, deep learning-based approaches have risen to prominence due to their superior accuracy; however, these methodologies encounter several obstacles. Primarily, they necessitate extensive labeled datasets and prolonged training periods, and given the rapid emergence of new vulnerabilities, the frequent retraining of models becomes a resource-intensive endeavor, thereby limiting their applicability in cutting-edge scenarios. To mitigate these challenges, this paper introduces the underlineYOTO--underlineYou underlineOnly underlineTrain underlineOnce framework. This innovative approach facilitates the integration of multiple types of vulnerability detection models via parameter fusion, eliminating the need for joint training. Consequently, YOTO enables swift adaptation to newly discovered vulnerabilities, significantly reducing both the time and computational resources required for model updates.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 20 of its 31 references that are in the map. First posted on arXiv 12 Mar 2025.

Builds on

20 of 31 references.

References · 20

  1. 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
  2. 2013Efficient Estimation of Word Representations in Vector Spacecited by 4,375
  3. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
  4. 2020CodeBERT: A Pre-Trained Model for Programming and Natural Languagescited by 1,310
  5. 2022Editing Models with Task Arithmeticcited by 1,120
  6. 2021CodeT5: Identifier-aware Unified Pre-trained Encoder-Decoder Models for Code Understanding and Generationcited by 906
  7. 2021Robust fine-tuning of zero-shot modelscited by 797
  8. 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 637
  9. 2021Merging Models with Fisher-Weighted Averagingcited by 621
  10. 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 471
  11. 2019Devign: Effective Vulnerability Identification by Learning Comprehensive Program Semantics via Graph Neural Networkscited by 272
  12. 2020Deep Learning based Vulnerability Detection: Are We There Yet?cited by 144
8 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 102
  3. 2018SySeVR: A Framework for Using Deep Learning to Detect Software Vulnerabilitiescited by 92
  4. 2023Deep Model Fusion: A Surveycited by 91
  5. 2018Automated software vulnerability detection with machine learningcited by 34
  6. 2024Beyond Task Vectors: Selective Task Arithmetic Based on Importance Metricscited by 11
  7. 2024PEPL: Precision-Enhanced Pseudo-Labeling for Fine-Grained Image Classification in Semi-Supervised Learningcited by 2
  8. 2024Improving the Classification Effect of Clinical Images of Diseases for Multi-Source Privacy Protectioncited by 1

Built on by

Not cited yet.