لطفا منتظر بمانید ...
0% Complete
صفحه اصلی
/
سی و چهارمین کنفرانس بین المللی مهندسی برق
A Framework for Enhancing Code Security in LLMs via Hierarchical Security-Context RAG
نویسندگان :
Abbas Mirheidari
1
Amir Jalali Bidgoli
2
Ali Taei Zadeh
3
1- دانشگاه قم
2- دانشگاه قم
3- دانشگاه قم
کلمات کلیدی :
Large Language Models (LLMs)،Secure Code Generation،Retrieval-Augmented Generation (RAG)،Fine-tuning
چکیده :
While Large Language Models (LLMs) are transforming software development, they often generate insecure code, posing significant security risks. Existing methods like fine-tuning can enhance security but are computationally expensive and create static model’s incapable of adapting to the dynamic threat landscape without costly retraining. To address these limitations, we propose HSC-RAG (Hierarchical Security-Context RAG), a novel retrieval-augmented generation framework that dynamically injects security knowledge into LLMs at inference time, eliminating the need for retraining. HSC-RAG leverages a knowledge base of secure/insecure code pairs and employs a multi-stage retrieval and filtering process to construct a highly relevant, security-focused context. Our experiments on models including StarCoder and Deepseek-coder show that HSC-RAG achieves up to 91% of the security performance of a fully fine-tuned model, while preserving the model’s utility on the HumanEval benchmark - all without incurring retraining costs.
لیست مقالات
لیست مقالات بایگانی شده
طراحی و شبیه سازی مبدل کاهنده دو مرحله ای با کنترل کننده زمان روشن-خاموش تطبیقی
نوید گودرزی - حسین پاک نیت - نوید یثربی
Helmet Microwave Array Applicator for Deep-Seated Brain Tumor Hyperthermia
Mohammad Moeini Arani - Mohammad Javad Hajiahmadi - Reza Faraji-Dana
Positioning a Moving Target Using Range and Doppler-Rate Measurements with Bi-static Radar
MohammadAmin Latifi - Fereidoon Behnia
Average Secrecy Capacity Performance Analysis for SWIPT-Based SIMO Underlay Cognitive Radio
Mohammad Javad Saber1 - Seyedeh Maryam Mazloum - Seyed Mohammad Sajad Sadough
برنامه ریزی توسعه شبکه های انتقال از دیدگاه شرکت های برق منطقه ای برای حداکثر سازی درآمد حاصل از ترانزیت برق
وحید مظفری - رضا نوروزیان - امیر باقری
ارائه روشی جهت بهبود عملکرد شبکههای بیسیم حسگر ناهمگون مبتنی بر برداشت انرژی
محمد فرشته حکمت - علیرضا کشاورز حداد
VGG16-M: بهینهسازی مبتنی بر تابع فعالسازی Mish و دادههای Hounsfield Unit در طبقهبندی نوع پاتولوژیک سرطان ریه از تصاویر CT
مرضیه علی آتشی - صدیقه غفرانی
Sensorless DTC of Dual Induction Motors Using a Five-Leg Inverter for EV Applications
Alireza Saffar Bahari - Karim Abbaszadeh
طراحی بهینه پارکینگ خودروهای برقی با در نظر گرفتن عدم قطعیت منابع انرژی تجدیدپذیر
سید محمد هاشمی مصیر - میثم جعفری نوکندی - محمد بزرگپور رودباری
A New 1-D Model for Singular 2-D Systems
Kamyar Azarakhsh - Masoud Shafiee
بیشتر
ثمین همایش، سامانه مدیریت کنفرانس ها و جشنواره ها - نگارش 44.7.2