Affiliation:
1. Islamic Azad University, Mashhad
Abstract
Abstract
The softmax loss function is a commonly used loss function in the field of classification, which aims to increase the angle between two classes in feature space. However, it has some limitations such as class overlap and treating all misclassifications equally, and issue with imbalanced classes. Recently, the I2CS (Intra concentration and inter-separability) loss function has been proposed with a different approach from the softmax loss function, which is compressing data at the center and increasing class distance through the class center, which makes it able to overcome some of the limitations such as class-imbalanced problems, outliers and discover samples of unseen classes. Nevertheless, it still suffers from class overlap problem. Therefore, we have designed a new loss function with a novel approach to not only overcome the limitations of the softmax loss function but also address the class overlap issue of I2CS, and be effective in dealing with class imbalances. Furthermore, our purpose loss function has been thoroughly tested on a variety of standard benchmark datasets such as MNIST, CIFAR, and LFW as well as on imbalanced MNIST class, showcasing enhanced performance when contrasted with the softmax loss function and other widely-used loss functions.
Publisher
Research Square Platform LLC
Reference28 articles.
1. Convolutional neural networks: an overview and application in radiology;Yamashita R;Insights into imaging,2018
2. CVPR 2020 continual learning in computer vision competition: Approaches, results, current challenges and future directions;Lomonaco V;Artif Intell,2022
3. Tan M, Pang R, Le QV (2020) Efficientdet: Scalable and efficient object detection. in Proceedings of the IEEE/CVF conference on computer vision and pattern recognition
4. Chen B, Deng W, Du J (2017) Noisy softmax: Improving the generalization ability of dcnn via postponing the early softmax saturation. in Proceedings of the IEEE conference on computer vision and pattern recognition
5. Li J et al (2018) AF-Softmax for Face Recognition. in 2018 International Conference on Network Infrastructure and Digital Content (IC-NIDC). IEEE