Saved in:
Bibliographic Details
Main Authors: Li, Qiufu, Jia, Xi, Zhou, Jiancan, Shen, Linlin, Duan, Jinming
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2403.07289
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914710843031552
author Li, Qiufu
Jia, Xi
Zhou, Jiancan
Shen, Linlin
Duan, Jinming
author_facet Li, Qiufu
Jia, Xi
Zhou, Jiancan
Shen, Linlin
Duan, Jinming
contents This paper introduces the concept of uniform classification, which employs a unified threshold to classify all samples rather than adaptive threshold classifying each individual sample. We also propose the uniform classification accuracy as a metric to measure the model's performance in uniform classification. Furthermore, begin with a naive loss, we mathematically derive a loss function suitable for the uniform classification, which is the BCE function integrated with a unified bias. We demonstrate the unified threshold could be learned via the bias. The extensive experiments on six classification datasets and three feature extraction models show that, compared to the SoftMax loss, the models trained with the BCE loss not only exhibit higher uniform classification accuracy but also higher sample-wise classification accuracy. In addition, the learned bias from BCE loss is very close to the unified threshold used in the uniform classification. The features extracted by the models trained with BCE loss not only possess uniformity but also demonstrate better intra-class compactness and inter-class distinctiveness, yielding superior performance on open-set tasks such as face recognition.
format Preprint
id arxiv_https___arxiv_org_abs_2403_07289
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Rediscovering BCE Loss for Uniform Classification
Li, Qiufu
Jia, Xi
Zhou, Jiancan
Shen, Linlin
Duan, Jinming
Computer Vision and Pattern Recognition
This paper introduces the concept of uniform classification, which employs a unified threshold to classify all samples rather than adaptive threshold classifying each individual sample. We also propose the uniform classification accuracy as a metric to measure the model's performance in uniform classification. Furthermore, begin with a naive loss, we mathematically derive a loss function suitable for the uniform classification, which is the BCE function integrated with a unified bias. We demonstrate the unified threshold could be learned via the bias. The extensive experiments on six classification datasets and three feature extraction models show that, compared to the SoftMax loss, the models trained with the BCE loss not only exhibit higher uniform classification accuracy but also higher sample-wise classification accuracy. In addition, the learned bias from BCE loss is very close to the unified threshold used in the uniform classification. The features extracted by the models trained with BCE loss not only possess uniformity but also demonstrate better intra-class compactness and inter-class distinctiveness, yielding superior performance on open-set tasks such as face recognition.
title Rediscovering BCE Loss for Uniform Classification
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.07289