Skip to content

Bimodal Adaptive Feature Fusion Network for Person Verification

Notifications You must be signed in to change notification settings

ZQSong1997/AVMFN-For-Person-Verification

Repository files navigation

Bimodal Adaptive Feature Fusion Network for Person Verification

Overview

本项目针对现目前仅采用人脸或声纹的单模态身份认证方法容易遭受欺骗攻击和环境噪声的不足,提出一种双模态自适应特征融合网络(BAFFN)来同时融合人员的人脸和声纹特征,以得到更鲁棒的特征表示来完成身份认证。试验在Voxceleb1数据集上进行,在EER(平均错误概率)上的最佳试验结果为0.548%,显著优于现有最先进的人脸识别或声纹识别模型。和同类型算法(人脸+声纹)相比,达到了SOTA水平。

Model architecture

本项目的人脸特征采用Facenet模型[1]提取,声纹特征采用Rawnet2模型[2]提取,特征提取过程和BAFFN的设计如下图所示。另外还采用了数据增强技术丰富了数据样本,以及利用度量学习策略,进一步降低了识别误差。 BAFFN

特别的,人脸特征和声纹特征也可采用其他更先进的单模态模型(如ArcFac和Rawnet3)进行特征提取,再利用BAFFN进行特征级训练,该方法具有普适性。

Datasets

Voxceleb1 dataset

  1. 数据集采用Voxceleb1 dev set作为训练集,test set作为测试集。 数据集官方链接
  2. Face embeddings可在'DB/Vox1_Face/emb/'获取,其中vox1_face_D_embedding.npy和vox1_face_T_embedding.npy分别是Voxcelev1训练集和测试集中关于人员人脸特征的embeddings(512维)。vox1_face_D_label.txt和vox1_face_T_label.txt为对应的label list,如:[id10001/1zcIwhmdeo4/0000375.jpg,id10001/1zcIwhmdeo4/0000475.jpg,...]。(百度云链接 提取码:38gq)
  3. Speaker embeddings可在'DB/Voxceleb1/emb/'获取,其中TTA_vox1_dev.pk和TTA_vox1_eval.pk分别是Voxcelev1训练集和测试集中关于人员语音特征的embeddings。它是一个字典格式的文件,keys为Speaker id,如'id10001/1zcIwhmdeo4/00001.wav',values为对应的embeddings(1024维)。(百度云链接 提取码:bnk9)
  4. 预训练模型可在'DNNs/Pre_trained_model/best_model.pkl'获得。

Usage

  1. 数据预处理:Run data_preprocess.py
  2. 模型训练:Run trainning_AVN.py
  3. 模型评估:Run evaluation_model.py

Reference

  1. [1] Schroff F, Kalenichenko D, Philbin J. Facenet: A unified embedding for face recognition and clustering[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. 2015: 815-823.

  2. [2] Jung J, Kim S, Shim H, et al. Improved rawnet with feature map scaling for text-independent speaker verification using raw waveforms[J]. arXiv preprint arXiv:2004.00526, 2020. @article{jung2020improved, title={Improved RawNet with Feature Map Scaling for Text-independent Speaker Verification using Raw Waveforms}, author={Jung, Jee-weon and Kim, Seung-bin and Shim, Hye-jin and Kim, Ju-ho and Yu, Ha-Jin}, journal={Proc. Interspeech}, pages={3583--3587}, year={2020} }

About

Bimodal Adaptive Feature Fusion Network for Person Verification

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages