针对情绪识别中数据不均衡问题,提出注意力LSTM模型提升小类分类准确率。
Error correction in multiclass image classification of facial emotion on unbalanced samples
- 用带注意力机制的LSTM聚焦面部关键区域进行情绪识别。
- 在六类训练、一类纠错的设置下,小类准确率显著提升。
- 适合处理罕见事件检测,如反欺诈系统中的异常情绪识别。
本文研究人脸表情多分类在样本不均衡情况下的错误修正问题。基于包含不同年龄人群七种情绪状态的图像数据集,重点关注某些情绪类别远超其他类别的分布偏差问题。采用基于LSTM与注意力机制的神经网络模型,聚焦对情绪识别有信息量的面部区域。实验中,模型在六种类别子集上训练,对第七类进行误差修正。结果表明,所有类别均可实现修正,但效果各异;部分小类在测试集上的关键指标(如精确率、召回率)显著提升,显示出该方法在检测稀有事件(如反欺诈系统中的异常行为)方面的潜力。所提方法可有效应用于表情分析及类别分布偏斜场景下的稳定分类任务。
原文摘要 · Abstract (English)
This paper considers the problem of error correction in multi-class classification of face images on unbalanced samples. The study is based on the analysis of a data frame containing images labeled by seven different emotional states of people of different ages. Particular attention is paid to the problem of class imbalance, in which some emotions significantly prevail over others. To solve the classification problem, a neural network model based on LSTM with an attention mechanism focusing on key areas of the face that are informative for emotion recognition is used. As part of the experiments, the model is trained on all possible configurations of subsets of six classes with subsequent error correction for the seventh class, excluded at the training stage. The results show that correction is possible for all classes, although the degree of success varies: some classes are better restored, others are worse. In addition, on the test sample, when correcting some classes, an increase in key quality metrics for small classes was recorded, which indicates the promise of the proposed approach in solving applied problems related to the search for rare events, for example, in anti-fraud systems. Thus, the proposed method can be effectively applied in facial expression analysis systems and in tasks requiring stable classification under skewed class distribution.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。