Prospects for the Application of Reinforcement Learning to Network Traffic Classification Tasks

Abstract

The basic principles and methods of reinforcement learning are reviewed. The problems and approaches for applying a model based on reinforcement learning in the framework of attack prevention are described. The model is built and the hyperparameters of machine learning for the task of classifying network traffic are selected, and its performance on the test data set is evaluated by such quality metrics as accuracy and completeness. The dataset used to implement an agent for selecting the optimal defense strategy for a particular attack has been finalized. Developed an algorithm for using a reinforcement learning neural network for the traffic classification task. A table of rules and rewards for the problem is generated. An agent has been developed and trained to interact with the system. We describe the application of reinforcement learning to the traffic classification task.