Proceedings

Ensemble Soft-Margin Softmax Loss for Image Classification

Download BibTeX Cite

Proceedings of the Twenty-Seventh International Joint Conference on Artificial Intelligence. Main track. Pages 992-998.

https://doi.org/10.24963/ijcai.2018/138

Abstract

Softmax loss is arguably one of the most popular losses to train CNN models for image classification. However, recent works have exposed its limitation on feature discriminability. This paper casts a new viewpoint on the weakness of softmax loss. On the one hand, the CNN features learned using the softmax loss are often inadequately discriminative. We hence introduce a soft-margin softmax function to explicitly encourage the discrmination between different classes. On the other hand, the learned classifier of softmax loss is weak. We propose to assemble multiple these weak classifiers to a strong one, inspired by the recognition that the diversity among weak classifiers is critical to a good ensemble. To achieve the diversity, we adopt the Hilbert-Schmidt Independence Criterion (HSIC). Considering these two aspects in one framework, we design a novel loss, named as Ensemble Soft-Margin Softmax (EM-Softmax). Extensive experiments on benchmark datasets are conducted to show the superiority of our design over the baseline softmax loss and several state-of-the-art alternatives.

  1. [Bell and Bala., 2015] S. Bell and K. Bala. Learning vi- sual similarity for product design with convolutional neu- ral networks. In TOG, 2015.
  2. [Chrabaszcz et al., 2017] P. Chrabaszcz, L. Ilya, and H. Frank. A downsampled variant of imagenet as an alternative to the cifar datasets. arXiv preprint arXiv:1707.08819, 2017.
  3. [Fredricks and Nelsen., 2007] G. Fredricks and R. Nelsen. On the relationship between spearman's rho and kendall's tau for pairs of continuous random variables. Journal of statistical planning and inference., 2007.
  4. [Goodfellow et al., 2013] I.
  5. Goodfellow, D. Warde-Farley, and M. Mirza. Maxout networks. In ICML, 2013.
  6. [Gretton et al., 2005] A. Gretton, O. Bousquet, and A. Smola. Measuring statistical dependence with hilbert-schmidt norms. In ICAL, 2005.
  7. [Guo et al., 2017] X. Guo, X. Wang, and H. Ling. Exclusiv- ity regularized machine. In IJCAI, 2017.
  8. [He et al., 2016] K. He, X. Zhang, and S. Ren. Deep residual learning for image recognition. In CVPR, 2016.
  9. [Hinton et al., 2012] G. Hinton, N. Srivastava, and A. Krizhevsky. Improving neural networks by pre- venting co-adaptation of feature detectors. arXiv preprint arXiv:1207.0580., 2012.
  10. [Jia et al., 2014] Y. Jia, E. Shelhamer, J. Donahue, S. Karayev, J. Long, R. Girshick, S. Guadarrama, and T. Darrell. Caffe: Convolutional architecture for fast feature embedding. arXiv preprint arXiv:1408.5093, 2014.
  11. [Krizhevsky and Hinton, 2009] A. Krizhevsky and G. Hin- ton. Learning multiple layers of features from tiny images. Technical report, 2009.
  12. [Krizhevsky et al., 2012] A. Krizhevsky, I. Sutskever, and G. Hinton. Imagenet classification with deep convolu- tional neural networks. In NIPS, 2012.
  13. [LeCun et al., 1998a] Y. LeCun, L. Bottou, and Y. Bengio. Gradient-based learning applied to document recognition. In Proceedings of the IEEE, 1998.
  14. [LeCun et al., 1998b] Y. LeCun, C. Cortes, and C. Burges. The mnist database of handwritten digits. 1998.
  15. [Lee et al., 2015] Y. Lee, S. Xie, and P. Gallagher. Deeply- supervised nets. In AISTATS, 2015.
  16. [Li et al., 2012] N. Li, Y. Yu, and Z. Zhou. Diversity regu- larized ensemble pruning. In ECML, 2012.
  17. [Li et al., 2016] Y. Li, K. He, and J. Sun. R-fcn: Object de- tection via region-based fully convolutional networks. In NIPS, 2016.
  18. [Liang et al., 2017] X. Liang, X. Wang, Z. Lei, S. Liao, and Stan. Li. Soft-margin softmax for deep classification. In ICONIP, 2017.
  19. [Lin et al., 2013] M. Lin, Q. Chen, and Y. Yan. Network in network. In ICLR, 2013.
  20. [Liu et al., 2016] W. Liu, Y. Wen, and Z. Yu. Large-margin softmax loss for convolutional neural networks. In ICML, 2016.
  21. [Liu et al., 2017a] W. Liu, Y. Wen, Z. Yu, M. Li, and L. Song. Sphereface: Deep hypersphere embedding for face recognition. In CVPR, 2017.
  22. [Liu et al., 2017b] Y. Liu, H. Li, and X. Wang. Learn- ing deep features via congenerous cosine loss for person recognition. arXiv preprint arXiv:1702.06890., 2017.
  23. [Martins and Astudillo., 2016] A. Martins and R. Astudillo. From softmax to sparsemax: A sparse model of attention and multi-label classification. In ICML, 2016.
  24. [Schroff et al., 2015] F. Schroff, D. Kalenichenko, and J. Philbin. Facenet: A unified embedding for face recog- nition and clustering. In CVPR, 2015.
  25. [Simonyan and Andrew, 2014] K. Simonyan and Z. Andrew. Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556, 2014.
  26. [Sun et al., 2014] Y. Sun, Y. Chen, and X. Wang. Deep learn- ing face representation by joint identification-verification. In NIPS, 2014.
  27. [Tang, 2013] Y. Tang. Deep learning using linear support vector machines. arXiv preprint arXiv:1306.0239, 2013.
  28. [Wang et al., 2015] X. Wang, X. Guo, and S. Li. Adap- tively unified semi-supervised dictionary learning with ac- tive points. In CVPR, 2015.
  29. [Wang et al., 2017a] F. Wang, X. Xiang, J. Chen, and A. Yuille. Normface: l 2 hypersphere embedding for face verification.. In ACM MM, 2017.
  30. [Wang et al., 2017b] X. Wang, X. Guo, Z. Lei, C. Zhang, and S. Li. Exclusivity-consistency regularized multi-view sub- space clustering. In CVPR, 2017.
  31. [Wang et al., 2018] X. Wang, Z. Lei, S. Liao, X. Guo, Y. Yang, and S. Li. Dependence-aware feature coding for person re-identification. Signal Processing Letters., 2018.
  32. [Wen et al., 2016] Y. Wen, K. Zhang, and Z. Li. A discrimi- native feature learning approach for deep face recognition. In ECCV, 2016.
  33. [Zhang et al., 2016] X. Zhang, Z. Fang, and Y. Wen. Range loss for deep face recognition with long-tail. arXiv preprint arXiv:1611.08976., 2016.
  34. [Zhou et al., 2016] B. Zhou, A. Khosla, and A. Lapedriza. Learning deep features for discriminative localization. In CVPR, 2016.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards