WB-KNN for emotion recognition from physiological signals

Abstract

K-nearest neighbor (KNN) has yielded excellent performance in physiological signals based on emotion recognition. But there are still some issues: the majority vote only by the nearest neighbors is too simple to deal with complex (like skewed) class distribution; features with the same contribution to the similarity will degrade the classification accuracy; samples in boundaries between classes are easily misclassified when k is larger. Therefore, we propose an improved KNN algorithm called WB-KNN, which takes into account the weight (both features and classification) and boundaries between classes. Firstly, a novel weighting method based on the distance and farthest neighbors named WDF is proposed to weight the classification, which improves the voting accuracy by making the nearer neighbors contribute more to the classification and using the farthest neighbors to reduce the weight of non-target class. Secondly, feature weight is introduced into the distance formula, so that the significant features contribute more to the similarity than noisy or irrelevant features. Thirdly, a voting classifier is adopted in order to overcome the weakness of KNN in boundaries between classes by combining different classifiers. Results of WB-KNN algorithm are encouraging compared with the traditional KNN and other classification algorithms on the physiological dataset with a skewed class distribution. Classification accuracy for 29 participants achieves 94.219 2% for the recognition of four emotions.

References

[1] CoverT, HartP. IEEE Transactions on Information Theory, 1967, 13: 21

[2] PedregosaF, VaroquauxG, GramfortA. Journal of Machine Learning Research, 2011, 12: 2825

[3] KreibigSD. Biological Psychology, 2010, 84: 394

[4] KimKH, BangSW, KimSR. Medical & Biological Engineering & Computing, 2004, 42: 419

[5] Li L and Chen J.H, International Conference on Intelligent Information Hiding and Multimedia, 355 (2006).

[6] Chang C.Y, Tsai J.S, Wang C.J and Chung P.C, IEEE Conference on Computational Intelligence in Bioinformatics &Computational Biology, 278 (2009).

[7] GouJP, QiuWM, YiZ, XuY, MaoQR, ZhanYZ. ACM Transactions on Intelligent Systems and Technology, 2019, 10: 1

[8] KoelstraS, MuhlC, SoleymaniM, LeeJS, YazdaniA, EbrahimiT, PunT, NijholtA, PatrasI. IEEE Transactions on Active Computing, 2012, 3: 18