Attackers Strike Back? Not Anymore -- An Ensemble of RL Defenders Awakens for APT Detection
By: Sidahmed Benabderrahmane, Talal Rahwan
Potential Business Impact:
Finds sneaky computer hackers before they steal data.
Advanced Persistent Threats (APTs) represent a growing menace to modern digital infrastructure. Unlike traditional cyberattacks, APTs are stealthy, adaptive, and long-lasting, often bypassing signature-based detection systems. This paper introduces a novel framework for APT detection that unites deep learning, reinforcement learning (RL), and active learning into a cohesive, adaptive defense system. Our system combines auto-encoders for latent behavioral encoding with a multi-agent ensemble of RL-based defenders, each trained to distinguish between benign and malicious process behaviors. We identify a critical challenge in existing detection systems: their static nature and inability to adapt to evolving attack strategies. To this end, our architecture includes multiple RL agents (Q-Learning, PPO, DQN, adversarial defenders), each analyzing latent vectors generated by an auto-encoder. When any agent is uncertain about its decision, the system triggers an active learning loop to simulate expert feedback, thus refining decision boundaries. An ensemble voting mechanism, weighted by each agent's performance, ensures robust final predictions.
Similar Papers
Ranking-Enhanced Anomaly Detection Using Active Learning-Assisted Attention Adversarial Dual AutoEncoders
Machine Learning (CS)
Finds hidden computer attacks with less work.
Adversarial Augmentation and Active Sampling for Robust Cyber Anomaly Detection
Cryptography and Security
Finds hidden computer attacks with less data.
Preliminary Investigation into Uncertainty-Aware Attack Stage Classification
Cryptography and Security
Helps computers guess hacker's next move.