Many hands make light work: An LLM-based multi-agent system for detecting malicious PyPI packages

Read full text
Authors: Muhammad Umar Zeshan, Motunrayo Osatohanmen Ibiyo, Claudio Di Sipio, Phuong Thi Lan Nguyen, Davide Di Ruscio
Cited by: 2
References: 4

Abstract

Malicious code in open-source repositories such as PyPI poses a growing threat to software supply chains. Traditional rule-based tools often overlook the semantic patterns in source code that are crucial for identifying adversarial components. Large language models (LLMs) show promise for software analysis, yet their use in interpretable and modular security pipelines remains limited. This paper presents LAMPS , a multi-agent system that employs collaborative LLMs to detect malicious PyPI packages. The system consists of four role-specific agents for package retrieval, file extraction, classification , and verdict aggregation , coordinated through the CrewAI framework. A prototype combines a fine-tuned CodeBERT model for classification with LLaMA 3 agents for contextual reasoning. LAMPS has been evaluated on two complementary datasets: D 1 , a balanced collection of 6,000 setup.py files, and D 2 , a realistic multi-file dataset with 1,296 files and natural class imbalance. On D 1 , LAMPS achieves 97.7% accuracy, surpassing MPHunter and TD-IDF stacking models–two state-of-the-art approaches. On D 2 , it reaches 99.5% accuracy and 99.5% balanced accuracy, outperforming RAG-based approaches and fine-tuned single-agent baselines. McNemar’s test confirmed these improvements as highly significant. The results demonstrate the feasibility of distributed LLM reasoning for malicious code detection and highlight the benefits of modular multi-agent designs in software supply chain security.

Fields of Study & Subjects

Advanced Malware Detection Techniques Software Engineering Research Spam and Phishing Detection

Publication details

Journal / source
Journal of Systems and Software
Publisher
Elsevier BV
Volume
236
Pages
112792–112792
Publication date
22 January 2026
Language
EN