This work explores Multi-Task Learning (MTL) for face recognition. First, we propose a multi-task Convolutional Neural Network (CNN) for face recognition where identity classification is the main task and Pose, Illumination, and Expression (PIE) estimations are the side tasks. Second, we develop a dynamic-weighting scheme to automatically assign the loss weights to each side task, which solves the crucial problem of balancing between different tasks in MTL. Third, we propose a pose-directed multi-task CNN by grouping different poses to learn pose-specific identity features, simultaneously across all poses in a joint framework. Last but not least, we propose an energy-based weight analysis method to explore how CNN- based MTL works. We observe that the side tasks serve as regularizations to disentangle the PIE variations from the learnt identity features. Extensive experiments on the entire Multi-PIE dataset demonstrate the effectiveness of the proposed approach. To the best of our knowledge, this is the first work using all data in Multi-PIE for face recognition. Our approach is also applicable to in-the-wild datasets for pose-invariant face recognition and achieves comparable or better performance than state of the art on LFW, CFP, and IJB-A datasets.

Overview MultiTask CNN

Figure 1: Multi-Task Learning (MTL) for face recognition. We observe that the side tasks serve as regularization terms to learn task specific features for each task. The final weights in the FC layer acts like feature selection to select only task-specific features for each task, resulting in disentangled (PIE-invariant) features for face recognition.

Multi-Task Convolutional Neural Network for Pose-Invariant Face Recognition Source Code

You can download the Multi-Task CNN Source Code from here.

Publications

  • Multi-Task Convolutional Neural Network for Pose-Invariant Face Recognition
    Xi Yin, Xiaoming Liu
    IEEE Transactions on Image Processing, Vol. 27, No. 2, pp.964-975, , Aug. 2017
    Bibtex | PDF | Code
  • @article{ multi-task-convolutional-neural-network-for-pose-invariant-face-recognition,
      author = { Xi Yin and Xiaoming Liu },
      title = { Multi-Task Convolutional Neural Network for Pose-Invariant Face Recognition },
      journal = { IEEE Transactions on Image Processing },
      volume = { 27 },
      number = { 2 },
      month = { August },
      year = { 2017 },
      pages = { 964--975 },
    }