arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1804.08087cs.CV

Anchor-based Nearest Class Mean Loss for Convolutional Neural Networks

University of Chinese Academy of Sciences(中国科学院大学);Shenzhen Institutes of Advanced Technology, CAS(中国科学院深圳先进技术研究院)

Fusheng Hao, Jun Cheng, Lei Wang, Xinchao Wang, Jianzhong Cao, Xiping Hu, Dapeng Tao

英文摘要:

Discriminative features are critical for machine learning applications. Most existing deep learning approaches, however, rely on convolutional neural networks (CNNs) for learning features, whose discriminant power is not explicitly enforced. In this paper, we propose a novel approach to train deep CNNs by imposing the intra-class compactness and the inter-class separability, so as to enhance the learned features' discriminant power. To this end, we introduce anchors, which are predefined vectors regarded as the centers for each class and fixed during training. Discriminative features are obtained by constraining the deep CNNs to map training samples to the corresponding anchors as close as possible. We propose two principles to select the anchors, and measure the proximity of two points using the Euclidean and cosine distance metric functions, which results in two novel loss functions. These loss functions require no sample pairs or triplets and can be efficiently optimized by batch stochastic gradient descent. We test the proposed method on three benchmark image classification datasets and demonstrate its promising results.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑