MobileFaceSwap: A Lightweight Framework for Video Face Swapping

Authors

  • Zhiliang Xu Baidu Inc.
  • Zhibin Hong Baidu Inc.
  • Changxing Ding South China University of Technology
  • Zhen Zhu University of Illinois at Urbana-Champaign
  • Junyu Han Baidu Inc.
  • Jingtuo Liu Baidu Inc.
  • Errui Ding Baidu Inc.

DOI:

https://doi.org/10.1609/aaai.v36i3.20203

Keywords:

Computer Vision (CV)

Abstract

Advanced face swapping methods have achieved appealing results. However, most of these methods have many parameters and computations, which makes it challenging to apply them in real-time applications or deploy them on edge devices like mobile phones. In this work, we propose a lightweight Identity-aware Dynamic Network (IDN) for subject-agnostic face swapping by dynamically adjusting the model parameters according to the identity information. In particular, we design an efficient Identity Injection Module (IIM) by introducing two dynamic neural network techniques, including the weights prediction and weights modulation. Once the IDN is updated, it can be applied to swap faces given any target image or video. The presented IDN contains only 0.50M parameters and needs 0.33G FLOPs per frame, making it capable for real-time video face swapping on mobile phones. In addition, we introduce a knowledge distillation-based method for stable training, and a loss reweighting module is employed to obtain better synthesized results. Finally, our method achieves comparable results with the teacher models and other state-of-the-art methods.

Downloads

Published

2022-06-28

How to Cite

Xu, Z., Hong, Z., Ding, C., Zhu, Z., Han, J., Liu, J., & Ding, E. (2022). MobileFaceSwap: A Lightweight Framework for Video Face Swapping. Proceedings of the AAAI Conference on Artificial Intelligence, 36(3), 2973-2981. https://doi.org/10.1609/aaai.v36i3.20203

Issue

Section

AAAI Technical Track on Computer Vision III