Agent Tuning & Optimization 相关度: 5/10

LoRA-based Parameter-Efficient LLMs for Continuous Learning in Edge-based Malware Detection

Christian Rondanini, Barbara Carminati, Elena Ferrari, Niccolò Lardo, Ashish Kundu
arXiv: 2602.11655v1 发布: 2026-02-12 更新: 2026-02-12

AI 摘要

提出一种基于LoRA的参数高效联邦学习框架,用于边缘设备上的恶意软件持续检测。

主要贡献

  • 提出基于LoRA的参数高效的边缘设备恶意软件检测持续学习架构。
  • 实现了边缘设备上的本地模型自适应和全局知识共享。
  • 实验证明该方法在应对未知攻击时具有更高的准确性和稳定性。

方法论

使用轻量级Transformer模型在边缘节点进行微调,仅聚合和重新分配LoRA模块,实现跨设备泛化。

原文摘要

The proliferation of edge devices has created an urgent need for security solutions capable of detecting malware in real time while operating under strict computational and memory constraints. Recently, Large Language Models (LLMs) have demonstrated remarkable capabilities in recognizing complex patterns, yet their deployment on edge devices remains impractical due to their resource demands. However, in edge malware detection, static or centrally retrained models degrade under evolving threats and heterogeneous traffic; locally trained models become siloed and fail to transfer across domains. To overcome these limitations, in this paper, we present a continuous learning architecture for edge-based malware detection that combines local adaptation on each device with global knowledge sharing through parameter-efficient LoRA adapters. Lightweight transformer models (DistilBERT, DistilGPT-2, TinyT5) run on edge nodes and are incrementally fine-tuned on device-specific traffic; only the resulting LoRA modules are aggregated by a lightweight coordinator and redistributed, enabling cross-device generalization without exchanging raw data. We evaluate on two public IoT security datasets, Edge-IIoTset and TON-IoT, under multi-round learning to simulate evolving threats. Compared to isolated fine-tuning, the LoRA-based exchange yields up to 20-25% accuracy gains when models encounter previously unseen attacks from another domain, while maintaining stable loss and F1 across rounds. LoRA adds less than 1% to model size (~0.6-1.8 MB), making updates practical for constrained edge hardware.

标签

LoRA 联邦学习 恶意软件检测 边缘计算

arXiv 分类

cs.CR cs.AI cs.DC