共 50 条
EXACT: How to train your accuracy
被引:0
|作者:
Karpukhin, Ivan
[1
]
Dereka, Stanislav
[1
,2
]
Kolesnikov, Sergey
[1
]
机构:
[1] Tinkoff, Golovinskoye Highway 5A, Moscow 125212, Russia
[2] Moscow Inst Phys & Technol MIPT, Dolgoprudnyi, Russia
关键词:
Classification;
Image recognition;
Deep learning;
Training objective;
Loss function;
Accuracy enhancement;
Label noise;
HINGE LOSS;
D O I:
10.1016/j.patrec.2024.06.033
中图分类号:
TP18 [人工智能理论];
学科分类号:
081104 ;
0812 ;
0835 ;
1405 ;
摘要:
Classification tasks are typically evaluated based on accuracy. However, due to the discontinuous nature of accuracy, it cannot be directly optimized using gradient-based methods. The conventional approach involves minimizing surrogate losses such as cross-entropy or hinge loss, which may result in suboptimal performance. In this paper, we introduce a novel optimization technique that incorporates stochasticity into the model's output and focuses on optimizing the expected accuracy, defined as the accuracy of the stochastic model. Comprehensive experimental evaluations demonstrate that our proposed optimization method significantly enhances performance across various classification tasks, including SVHN, CIFAR-10, CIFAR-100, and ImageNet.
引用
收藏
页码:23 / 30
页数:8
相关论文