Focal Loss for End-to-end Short Utterances Chinese Dialect Identification.

Qiuxian Zhang,Jiangyan Yi,Jianhua Tao,Mingliang Gu,Yong Ma
DOI: https://doi.org/10.1109/apsipaasc47483.2019.9023354
2019-01-01
Abstract:Short utterances dialect identification is a challenging task because of the substantial similarity between dialects. The previous cross-entropy loss function does not consider the category and probability of prediction error, which result in insensitivity to easily misclassified and unbalanced samples. To solve this problem, we propose to use an improved cross-entropy loss function, namely focal loss, introducing category weights and tunable focusing parameter to improve the classification accuracy. Experiments are carried out on AI Dialect Contest database. The results demonstrate that our proposed end-to-end model trained with focal loss achieves better performance than the model trained with cross-entropy loss function.
What problem does this paper attempt to address?