Lockheed Martin Assesses 5G Network Vulnerabilities with Reinforcement Learning Toolbox

“5G is a critical infrastructure that we must protect from adversarial attacks. Reinforcement Learning Toolbox allows us to quickly assess 5G vulnerabilities and identify mitigation methods.”

Key Outcomes

  • Shortened development and analysis time compared to traditional methods
  • Enabled cybersecurity engineers with little previous AI experience to implement the solution
  • Identified worst-case attack scenarios and developed mitigation methods
  • Quickly evaluated and compared multiple reinforcement learning algorithms

With approximately 35,000 employees working in 16 countries, Lockheed Martin Rotary and Mission Systems (RMS) develops a broad range of products and technologies. Recently, the RMS cybersecurity team has been working on assessing vulnerabilities in 5G infrastructure, including those related to data security, user privacy, confidentiality, integrity, and availability. These complex problems cannot be solved by traditional methods or approaches.

To assess these gaps, the team used MATLAB® and Reinforcement Learning Toolbox™ to discover attack scenarios. It began by building 5G models using EXata Cyber emulation software and defining a set of threat vectors based on security frameworks from the 3rd Generation Partnership Project (3GPP), National Security Agency (NSA), and other organizations. They then used Reinforcement Learning Toolbox to train an adversarial deep Q-network agent to optimize attack patterns and find worst-case scenarios. Based on these results, the team identified mitigation techniques to address the discovered vulnerabilities.

The team is now exploring opportunities to increase the fidelity of their simulations using 5G Toolbox™ and achieve better realism with a multi-agent reinforcement learning framework enabled by Simulink®.