Skip to content

Distilling Knowledge from Large Language Models into Lightweight Reinforcement Learning Agents for Autonomous Cyber Operations

Konur Tholl F. Rivest Mariam El Mezouar Adrian Taylor Ranwa Al Mallah
Jul 2026 · arXiv.org · Vol abs/2607.28826 · 0 citations · 28 references
Computer Science

TL;DR

The use of a Large Language Model (LLM) to improve autonomous defensive decision-making within an ACO environment is investigated and an online policy distillation framework is proposed that transfers the LLM's defensive policy into a lightweight RL agent containing only 64,910 parameters, reducing model size by several orders of magnitude while maintaining effective defensive capabilities.

Abstract

Autonomous Cyber Operations (ACO) are increasingly important for defending enterprise networks as cyber threats continue to evolve in sophistication. ACO applications commonly employ Reinforcement Learning (RL) agents to learn defensive behaviors through interaction with environments. However, RL agents typically require extensive exploration during training, often resulting in unstable behavior and poor initial decision-making before converging toward effective defense strategies. In this work, we investigate the use of a Large Language Model (LLM) to improve autonomous defensive decision-making within an ACO environment. Through prompt engineering rather than fine-tuning, we demonstrate that an 8-billion parameter LLM pretrained on cybersecurity data can outperform a baseline RL agent in a modified CybORG CAGE Challenge 2 environment. We then propose an online policy distillation framework that transfers the LLM's defensive policy into a lightweight RL agent containing only 64,910 parameters, reducing model size by several orders of magnitude while maintaining effective defensive capabilities. This provides a pathway toward operationalizing frontier cybersecurity models within lightweight, deployable agents. To evaluate transferability, we construct CybORG scenarios ranging from 4 to 12 hosts and assess the approach across varying network configurations. We also evaluate teacher-guided RL stabilization strategies and observe that none consistently surpass the optimized teacher policy, suggesting policy-alignment limitations between reward-driven RL optimization and teacher-guided defense strategies. Our results demonstrate the potential of cybersecurity-focused LLMs as sources of expertise for autonomous cyber defense, while policy distillation provides a practical path toward operationalizing frontier cybersecurity models within efficient, scalable agents.

View source

Similar papers

#machine learning Preprint Aug 2026

Trident : How to Break Deep Reinforcement Learning Cyber Defenses (Agentic)

Empirical evaluations reveal a fundamental brittleness in existing defenses: with a single trainable 7B planner, Trident reduces blue agent defensive performance by an average of 522% compared to static red agent baselines while autonomously discovering emergent behaviors such as decoy avoidance and adaptive state prio...

Ryozo Masukawa, Ian Bryant, Armita Kazeminajafabadi et al. · 1 citation
2026

Cyber Task Automation With Knowledge-Infused Reinforcement Learning and LLM-Guided Policies

As cyber threats continue to evolve, there is a need for Autonomous Cyber Defense (ACD) strategies capable of fast and context-aware responses. Reinforcement learning (RL) has shown promise in automating cyber defense by exploring and learning effective countermeasures. However, RL often struggles with sparse reward si...

Md. Shamim Towhid, Shahrear Iqbal, E. C. Pinto et al. · 0 citations
Preprint Aug 2026

Proving the Utility of Large Language Models in Cybersecurity Simulations: A Comprehensive Examination

YAML is employed as a structured representation format for simulating complex network configurations, thereby enabling Large Language Model-driven pipelines to support and improve reinforcement learning (RL) agent training, and underscore the transformative potential of integrating LLMs into cybersecurity research.

S. Kampakis, F. Rovai, Marcos Charalambides et al. · 1 citation
Conference Jul 2026

Can LLM Agents Replace Reinforcement Learning Agents in Cyber Defence Automation: A Case Study Using the DARPA CAGE-2 Challenge

As cyber attacks grow more sophisticated, defenders need autonomous systems that are fast, adaptable, and explainable. Over the last decade, various strategies have been proposed for automated cyber defence (as opposed to static rule-based or signature-based), including those based on reinforcement learning (RL). Resea...

Arijit Diganto, S. Lohrasbi, Euclides Carlos Pinto et al. · 0 citations
Open access Aug 2026

L-ARLPT: An LLM-Augmented Reinforcement Learning Framework for Autonomous Penetration Testing

A Large Language Model-enhanced Autonomous Reinforcement Learning Penetration Testing framework that leverages the domain knowledge embedded in a Large Language Model to perform tactical planning, thereby pruning the original action space into a compact set of candidate actions.

Rufeng Zhan, Junyi Zhu, Yinghui Xu et al. · 0 citations
Jul 2026

Agentic AI for Offensive Security: LLM-guided Autonomous Red Teaming in a Limited Cyber-range Environment

Results show that, within a controlled cyber-range environment, LLM-guided agents can approximate deterministic performance when appropriate constraints are applied, suggesting that agent-based approaches to offensive security can support semi-autonomous red teaming workflows, provided that decision-making is governed...

Atif Chowhan, Yasir Hamid · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.