Classification of attention levels using a Random Forest algorithm optimized with Particle Swarm Optimization

Attention is one of the most important cognitive functions since it allows us to discriminate irrelevant stimuli when performing an activity. The presence of an attention deficit significantly affects a person’s performance. This is one of the reasons why it is of utmost importance to determine the state of attention mechanisms.A tool that allows determining the level of attention could be of great help in the diagnosis of syndromes or disorders, as well as in the rehabilitation and treatment of people suffering from attention deficits. In this work, a methodology is proposed based on a Random Forest algorithm optimized with PSO (Particle Swarm Optimization) for the classification of attention levels. These attention levels are divided into three main categories: High Attention, Normal Attention, and Low Attention. The proposed approach demonstrated reaching an accuracy of up to 96%. Finally, the approach from this contribution was compared with the state of the art, demonstrating that this work is a feasible methodology for this application.