DF<sup>2</sup>Net: Discriminative Feature Learning and Fusion Network for RGB-D Indoor Scene Classification

Authors

  • Yabei Li Institute of Automation, Chinese Academy of Sciences (CASIA)
  • Junge Zhang Institute of Automation, Chinese Academy of Sciences (CASIA)
  • Yanhua Cheng Tencent
  • Kaiqi Huang Institute of Automation, Chinese Academy of Sciences (CASIA)
  • Tieniu Tan Institute of Automation, Chinese Academy of Sciences (CASIA)

Keywords:

RGB-D, scene recognition, multi-modal

Abstract

This paper focuses on the task of RGB-D indoor scene classification. It is a very challenging task due to two folds. 1) Learning robust representation for indoor scene is difficult because of various objects and layouts. 2) Fusing the complementary cues in RGB and Depth is nontrivial since there are large semantic gaps between the two modalities. Most existing works learn representation for classification by training a deep network with softmax loss and fuse the two modalities by simply concatenating the features of them. However, these pipelines do not explicitly consider intra-class and inter-class similarity as well as inter-modal intrinsic relationships. To address these problems, this paper proposes a Discriminative Feature Learning and Fusion Network (DF2Net) with two-stage training. In the first stage, to better represent scene in each modality, a deep multi-task network is constructed to simultaneously minimize the structured loss and the softmax loss. In the second stage, we design a novel discriminative fusion network which is able to learn correlative features of multiple modalities and distinctive features of each modality. Extensive analysis and experiments on SUN RGB-D Dataset and NYU Depth Dataset V2 show the superiority of DF2Net over other state-of-the-art methods in RGB-D indoor scene classification task.

Downloads

Published

2018-04-27

How to Cite

Li, Y., Zhang, J., Cheng, Y., Huang, K., & Tan, T. (2018). DF<sup>2</sup>Net: Discriminative Feature Learning and Fusion Network for RGB-D Indoor Scene Classification. Proceedings of the AAAI Conference on Artificial Intelligence, 32(1). Retrieved from https://ojs.aaai.org/index.php/AAAI/article/view/12292