Research Article

Improved Loss Function for Image Classification

Table 2

Comparison of recall rates with three loss functions.

DatasetModelLoss functionRecall@1Recall@2Recall@4Recall@8

CIFAR-100GoogLeNetMargin loss0.6070.7120.7630.815
Triplet loss0.6100.7080.7720.810
New loss0.6320.7200.7790.817
GoogLeNetMargin loss0.7120.8100.8730.916
Triplet loss0.6950.8010.8690.912
New loss0.7220.82910.88920.9294
CIFAR-10ResNet50Margin loss0.78070.84960.89220.9312
Triplet loss0.80670.86610.90450.9319
New loss0.77430.86380.9180.9504
Fashion-MNISTResNet50Margin loss0.88640.92730.95450.9704
Triplet loss0.89090.93040.9590.9742
New loss0.87490.92590.9610.9801