Shen Yinhong, Zhang Chang, Yang Lin, Li Yuanyuan, Zheng Xiujuan
College of Electrical Engineering, Sichuan University, Chengdu 610065, P. R. China.
Mental Health Center of the West China Hospital, Sichuan University, Chengdu 610041, P. R. China.
Sheng Wu Yi Xue Gong Cheng Xue Za Zhi. 2023 Apr 25;40(2):335-342. doi: 10.7507/1001-5515.202204066.
When performing eye movement pattern classification for different tasks, support vector machines are greatly affected by parameters. To address this problem, we propose an algorithm based on the improved whale algorithm to optimize support vector machines to enhance the performance of eye movement data classification. According to the characteristics of eye movement data, this study first extracts 57 features related to fixation and saccade, then uses the ReliefF algorithm for feature selection. To address the problems of low convergence accuracy and easy falling into local minima of the whale algorithm, we introduce inertia weights to balance local search and global search to accelerate the convergence speed of the algorithm and also use the differential variation strategy to increase individual diversity to jump out of local optimum. In this paper, experiments are conducted on eight test functions, and the results show that the improved whale algorithm has the best convergence accuracy and convergence speed. Finally, this paper applies the optimized support vector machine model of the improved whale algorithm to the task of classifying eye movement data in autism, and the experimental results on the public dataset show that the accuracy of the eye movement data classification of this paper is greatly improved compared with that of the traditional support vector machine method. Compared with the standard whale algorithm and other optimization algorithms, the optimized model proposed in this paper has higher recognition accuracy and provides a new idea and method for eye movement pattern recognition. In the future, eye movement data can be obtained by combining it with eye trackers to assist in medical diagnosis.
在对不同任务进行眼动模式分类时,支持向量机受参数影响较大。为解决这一问题,我们提出一种基于改进鲸鱼算法的算法来优化支持向量机,以提高眼动数据分类性能。根据眼动数据的特点,本研究首先提取57个与注视和扫视相关的特征,然后使用ReliefF算法进行特征选择。为解决鲸鱼算法收敛精度低和易陷入局部极小值的问题,我们引入惯性权重来平衡局部搜索和全局搜索,以加快算法的收敛速度,还使用差分变异策略增加个体多样性以跳出局部最优。本文对八个测试函数进行了实验,结果表明改进后的鲸鱼算法具有最佳的收敛精度和收敛速度。最后,本文将改进鲸鱼算法优化后的支持向量机模型应用于自闭症眼动数据分类任务,在公开数据集上的实验结果表明,本文的眼动数据分类准确率与传统支持向量机方法相比有了很大提高。与标准鲸鱼算法和其他优化算法相比,本文提出的优化模型具有更高的识别准确率,为眼动模式识别提供了新的思路和方法。未来,可以通过与眼动仪结合获取眼动数据,辅助医学诊断。