mlsanityData to 7 Oct 2026

Enhancing Continual Learning for Software Vulnerability Prediction: Addressing Catastrophic Forgetting via Hybrid-Confidence-Aware Selective Replay for Temporal LLM Fine-Tuning

arXiv 2602.23834 ↗First posted 27 Feb 2026cs.CR cs.AI cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Recent work applies Large Language Models (LLMs) to source-code vulnerability detection, but most evaluations still rely on random train-test splits that ignore time and overestimate real-world performance. In practice, detectors are deployed on evolving code bases and must recognise future vulnerabilities under temporal distribution shift. This paper investigates continual fine-tuning of a decoder-style language model (microsoft/phi-2 with LoRA) on a CVE-linked dataset spanning 2018-2024, organised into bi-monthly windows. We evaluate eight continual learning strategies, including window-only and cumulative training, replay-based baselines and regularisation-based variants. We propose Hybrid Class-Aware Selective Replay (Hybrid-CASR), a confidence-aware replay method for binary vulnerability classification that prioritises uncertain samples while maintaining a balanced ratio of VULNERABLE and FIXED functions in the replay buffer. On bi-monthly forward evaluation Hybrid-CASR achieves a Macro-F1 of 0.667, improving on the window-only baseline (0.651) by 0.016 with statistically significant gains (p = 0.026) and stronger backward retention (IBR@1 of 0.741). Hybrid-CASR also reduces training time per window by about 17 percent compared to the baseline, whereas cumulative training delivers only a minor F1 increase (0.661) at a 15.9-fold computational cost. Overall, the results show that selective replay with class balancing offers a practical accuracy-efficiency trade-off for LLM-based temporal vulnerability detection under continuous temporal drift.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 18 of its 32 references that are in the map. First posted on arXiv 27 Feb 2026.

Builds on

18 of 32 references.

References · 18

  1. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  2. 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,355
  3. 2016iCaRL: Incremental Classifier and Representation Learningcited by 2,313
  4. 2017Gradient Episodic Memory for Continual Learningcited by 1,871
  5. 2020CodeBERT: A Pre-Trained Model for Programming and Natural Languagescited by 1,286
  6. 2018Efficient Lifelong Learning with A-GEMcited by 1,044
  7. 2020AdapterFusion: Non-Destructive Task Composition for Transfer Learningcited by 780
  8. 2021CodeXGLUE: A Machine Learning Benchmark Dataset for Code Understanding and Generationcited by 732
  9. 2019Three scenarios for continual learningcited by 610
  10. 2019Online Continual Learning with Maximally Interfered Retrievalcited by 376
  11. 2019Devign: Effective Vulnerability Identification by Learning Comprehensive Program Semantics via Graph Neural Networkscited by 266
  12. 2024Continual Learning for Large Language Models: A Surveycited by 144
6 more
  1. 2018The Adverse Effects of Code Duplication in Machine Learning Models of Codecited by 123
  2. 2021CVEfixes: Automated Collection of Vulnerabilities and Their Fixes from Open-Source Softwarecited by 89
  3. 2019A Manually-Curated Dataset of Fixes to Vulnerabilities of Open-Source Softwarecited by 28
  4. 2023ConPET: Continual Parameter-Efficient Tuning for Large Language Modelscited by 16
  5. 2024On Inter-dataset Code Duplication and Data Leakage in Large Language Modelscited by 10
  6. 2024On the Convergence of Continual Learning with Adaptive Methodscited by 7

Built on by

Not cited yet.