Noise-Tolerant Paradigm for Training Face Recognition CNNs

Paper link: https://arxiv.org/abs/1903.10357

Presented at CVPR 2019

This is the code for the paper

Contents

1. Requirements
2. Dataset
3. How-to-use
4. Diagram
5. Performance
6. Contact
7. Citation
8. License

Requirements

1. Caffe

Dataset

Training dataset:

1. CASIA-Webface clean
2. IMDB-Face
3. MS-Celeb-1M

Testing dataset:

1. LFW
2. AgeDB
3. CFP
4. MegaFace

Both the training data and testing data are aligned by the method described in util.py

How-to-use

Firstly, you can train the network in noisy dataset as following steps:

step 1: add noise_tolerant_fr and relevant layers(at ./layers directory) to caffe project and recompile it.

step 2: download training dataset to ./data directory, and corrupt training dataset in different noise ratio which you can refer to ./code/gen_noise.py, then generate the lmdb file through caffe tool.

step 3: configure prototxt file in ./deploy directory.

step 4: run caffe command to train the network by using noisy dataset.

After training, you can evaluate the model on testing dataset by using evaluate.py.

Diagram

The figure shows three strategies in different purposes. At the beginning of the training process, we focus on all samples; then we focus on easy/clean samples; at last we focus on semi-hard clean samples.

The figure explains the fusion function of three strategies. The left part demonstrates three functions: α(δ_r), β(δ_r), and γ(δ_r). The right part shows two fusion examples. According to the ω, we can see that the easy/clean samples are emphasized in the first example(δ_r < 0.5), and the semi-hard clean samples are emphasized in the second example(δ_r > 0.5). For more detail, please click here to play demo video.

The figure shows the 2D Hist_all of CNN_common (up) and CNN_m2 (down) under 40% noise rate.

The figure shows the 3D Hist_all of CNN_common (left) and CNN_m2 (right) under 40% noise rate.

Performance

The table shows comparison of accuracies(%) on LFW, ResNet-20 models are used. CNN_clean is trained with clean data WebFace-Clean-Sub by using the traditional method. CNN_common is trained with noisy dataset WebFace-All by using the traditional method. CNN_ct is trained with noisy dataset WebFace-All by using our implemented Co-teaching(with pre-given noise rates). CNN_m1 and CNN_m2 are all trained with noisy dataset WebFace-All but through the proposed approach, and they respectively use the 1st and 2nd method to compute loss. Note: The WebFace-Clean-Sub is the clean part of the WebFace-All, the WebFace-All contains noise data with different rate as describe below.

Contact

Wei Hu

Yangyu Huang

Citation

If you find this work useful in your research, please cite

@inproceedings{Hu2019NoiseFace,
  title = {Noise-Tolerant Paradigm for Training Face Recognition CNNs},
  author = {Hu, Wei and Huang, Yangyu and Zhang, Fan and Li, Ruirui},
  booktitle = {IEEE Conference on Computer Vision and Pattern Recognition},
  month = {June},
  year = {2019},
  address = {Long Beach, CA}
}

License

The project is released under the MIT License