Research Article
Improved Loss Function for Image Classification
Table 2
Comparison of recall rates with three loss functions.
| | Dataset | Model | Loss function | Recall@1 | Recall@2 | Recall@4 | Recall@8 |
| | CIFAR-100 | GoogLeNet | Margin loss | 0.607 | 0.712 | 0.763 | 0.815 | | Triplet loss | 0.610 | 0.708 | 0.772 | 0.810 | | New loss | 0.632 | 0.720 | 0.779 | 0.817 | | GoogLeNet | Margin loss | 0.712 | 0.810 | 0.873 | 0.916 | | Triplet loss | 0.695 | 0.801 | 0.869 | 0.912 | | New loss | 0.722 | 0.8291 | 0.8892 | 0.9294 | | CIFAR-10 | ResNet50 | Margin loss | 0.7807 | 0.8496 | 0.8922 | 0.9312 | | Triplet loss | 0.8067 | 0.8661 | 0.9045 | 0.9319 | | New loss | 0.7743 | 0.8638 | 0.918 | 0.9504 | | Fashion-MNIST | ResNet50 | Margin loss | 0.8864 | 0.9273 | 0.9545 | 0.9704 | | Triplet loss | 0.8909 | 0.9304 | 0.959 | 0.9742 | | New loss | 0.8749 | 0.9259 | 0.961 | 0.9801 |
|
|