Open AccessProspects for the Application of Reinforcement Learning to Network Traffic Classification Tasks
Author(s) -
G. D. Asyaev
Publication year - 2021
Publication title -
journal of physics. conference series
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.21
H-Index - 85
eISSN - 1742-6596
pISSN - 1742-6588
DOI - 10.1088/1742-6596/2096/1/012175
Subject(s) - reinforcement learning , computer science , machine learning , artificial intelligence , task (project management) , hyperparameter , artificial neural network , learning classifier system , set (abstract data type) , reinforcement , engineering , systems engineering , structural engineering , programming language
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.