G ENE F ACE : T OWARDS H IGH -F IDELITY A UDIOD RIVEN 3D T ALKING F ACE S YNTHESIS G ENERALIZABLE T O O UT -O F -D OMAIN A UDIO
Anonymous authors
Paper under double-blind review
A BSTRACT
Generating photo-realistic video portrait with arbitrary speech audio is a crucial problem in ﬁlm-making and virtual reality.
Recently, several works explore the usage of neural radiance ﬁeld in this task to improve 3D realness and image ﬁdelity.
However, the generalizability of previous NeRF-based methods to out-of-domain audio is limited by the small scale of training data.
In this work, we propose GeneFace, a generalized and high-ﬁdelity NeRF-based talking face generation method, which can generate natural results corresponding to various out-of-domain audio.
Speciﬁcally, we learn a variaitional motion generator on a large lip-reading corpus, and introduce a domain adaptative post-net to calibrate the result.
Moreover, we learn a NeRF-based renderer conditioned on the predicted facial motion.
A head-aware torso-NeRF is proposed to eliminate the head-torso separation problem.
Extensive experiments show that our method achieves more generalized and high-ﬁdelity talking face generation compared to previous methods 1 .
I NTRODUCTION
Audio-driven face video synthesis is an important and challenging problem with several applications such as digital humans, virtual reality (VR), and online meetings.
Over the past few years, the community has exploited Generative Adversarial Networks (GAN) as the neural renderer and promoted the frontier from only predicting the lip movement Prajwal et al.
(2020)Chen et al.
(2019) to generating the whole face Zhou et al.
(2021)Lu et al.
However, GAN-based renderers suffer from several limitations such as unstable training, mode collapse, difﬁculty in modelling delicate details
Suwajanakorn et al.
(2017)Thies et al.
(2020), and ﬁxed static head pose Pham et al.
(2017)Taylor et al.
(2017)Cudeiro et al.
Recently, Neural Radiance Field (NeRF) Mildenhall et al.
(2020) has been explored in talking face generation.
Compared with GAN-based rendering techniques,
NeRF renderers could preserve more details and provide better 3D naturalness since it models a continuous 3D scene in the hidden space.
Recent NeRF-based works Guo et al.
(2021)Liu et al.
(2022)Yao et al.
(2022) manage to learn an end-to-end audio-driven talking face system with only a few-minutes-long video.
However, the current end-to-end framework is faced with two challenges.
The ﬁrst challenge is the weak generalizability due to the small scale of training data, which only consists of about thousands-many audio-image pairs.
This deﬁciency of training data makes the trained model not robust to out-ofdomain (OOD) audio in many applications (such as cross-lingual Guo et al.
(2021)Liu et al.
or singing voice).
The second challenge is the so-called ”mean face” problem .
Note that the audio to its corresponding facial motion is a one-to-many mapping, which means the same audio input may have several correct motion patterns.
Learning such a mapping with a regression-based model leads to over-smoothing and blurry results Ren et al.
(2021); speciﬁcally, for some complicated audio with several potential outputs, it tends to generate an image with a half-opened and blurry mouth, which leads to unsatisfying image quality and bad lip-synchronization.
To summarize, the current NeRF-based methods are challenged with the weak generalizability problem due to the lack of audio-to-motion training data and the ”mean face” results due to the one-to-many mapping.
In this work, we develop a talking face generation system called GeneFace to address these two challenges.
To handle the weak generalizability problem, we devise an audio-to-motion model to predict the 3D facial landmark given the input audio.
We utilize hundreds of hours of audio-motion pairs from a large-scale lip reading datasetAfouras et al.
(2018) to learn a robust mapping.
As for the ”mean face” problem, instead of using the regression-based model, we adopt a variational autoencoder (VAE) with a ﬂow-based prior as the architecture of the audio-to-motion model, which helps generate accurate and expressive facial motions.
However, due to the domain shift between the generated landmarks (in the multi-speaker domain) and the training set of NeRF (in the target person domain), we found that the NeRF-based renderer fails to generate high-ﬁdelity frames given the predicted landmarks.
Therefore, a domain adaptation process is proposed to rig the predicted landmarks into the target person’s distribution.
To summarize, our system consists of three stages:
1 Audio-to-motion .
We present a variational
motion
generator to generate accurate
and
expressive facial landmark given the input audio.
2 Motion domain adaptation .
To overcome the domain shift, we propose a semi-supervised adversarial training pipeline to train a domain adaptative post-net, which reﬁnes the predicted 3D landmark from the multi-speaker domain into the target person domain.
3 Motion-to-image.
We design a NeRF-based renderer to render high-ﬁdelity frames conditioned on the predicted 3D landmark.
The main contributions of this paper are summarized as follows:
• We present a three-stage framework that enables the NeRF-based talking face system to enjoy the large-scale lip-reading corpus and achieve high generalizability to various OOD audio.
We propose an adversarial domain adaptation pipeline to bridge the domain gap between the large corpus and the target person video.
• We are the ﬁrst work that analyzes the ”mean face” problem induced by the one-to-many audioto-motion mapping in the talking face generation task.
To handle this problem, we design a variational motion generator to generate accurate facial landmarks with rich details and expressiveness.
• Experiments show that our GeneFace outperforms other state-of-the-art GAN-based and NeRFbased baselines from the perspective of objective and subjective metrics.
R ELATED W ORK
Our approach is a 3D talking face system that utilizes a generative model to predict the 3DMMbased motion representation given the driving audio and employs a neural radiance ﬁeld to render the corresponding images of a human head.
It is related to recent approaches to audio-driven talking head generation methods and scene representation networks for the human portrait.
Audio-driven Talking Head Generation Generating talking faces in line with input audio has long attracted the attention of the computer vision community.
Earlier works focus on synthesizing the lip motions on a static facial imageJamaludin et al.
(2019)Tony Ezzat & Poggio (2002)Vougioukas et al.
(2020)Wiles et al.
Then the frontier is promoted to synthesize the full headYu et al.
(2020)Zhou et al.
(2019)Zhou et al.
However, free pose control is not feasible in these methods due to the lack of 3D modeling.
With the development of 3D face reconstruction techniquesDeng et al.
(2019), many works explore extracting 3D Morphable Model (3DMM)Paysan et al.
(2009) from the monocular video to represent the facial movementTero Karras & Lehtinen (2017)Yi et al.
(2020) in the talking face system, which is named as model-based methods.
With 3DMM, a coarse 3D face mesh M can be represented as an afﬁne model of facial expression and identity code:
[Equation]
where M is the average face shape; B id and B exp are the PCA bases of identity and expression; i and e are known as identity and expression codes.
By modeling the 3D geometry with 3DMM, the model-based works manage to manipulate the head pose and facial movement.
However, 3DMM could only deﬁne a coarse 3D mesh of the human head, and delicate details (such as hair, wrinkle, teeth, etc.) are ignored.
It raises challenges for
[Figure]
GAN-based methods to obtain realistic results.
Recent advances in neural rendering have created a prospect: instead of reﬁning the geometry of 3DMM or adding more personalized attributes as auxiliary conditions for GAN-based renderers, we could leave these delicate details to be modeled implicitly by the hidden space of the neural radiance ﬁeld.
Neural Radiance Field for Rendering Face
The recent proposed neural radiance ﬁeld (NeRF)Mildenhall et al.
(2020)Kellnhofer et al.
(2021)Pumarola et al.
(2021)Sitzmann et al.
(2019) has attracted much attention in the human portrait rendering ﬁeld since it could render high-ﬁdelity images with rich details such as hair and wrinkles.
For instance, Sitzmann et al.
(2019) presents a compositional NeRF for generating each part of the upper body.
NerFaceGafni et al.
(2021) and
Pumarola et al.
propose pose-expression-conditioned dynamic NeRFs for modeling the dynamics of a human face.
EG3DChan et al.
(2022) proposes a hybrid explicit–implicit tri-plane representation to achieve fast and geometry-aware human face rendering.
HeadNeRFHong et al.
(2022) proposes a real-time NeRF-based parametric head model.
Several works have also applied NeRF in the audio-driven talking face generation task.
Zhang et al.
(2021) devise an implicit pose code to modularize audio-visual representations.
AD-NeRF Guo et al.
ﬁrst presents an end-to-end audio-driven NeRF to generate face images conditioned on
Deepspeech Hannun et al.
(2014) audio features.
Recently, SSP-NeRF Liu et al.
(2022) proposed a semantic-aware dynamic ray sampling module to improve the sample efﬁciency and design a torso deformation module to stabilize the large-scale non-rigid torso motions.
DFA-NeRF Yao et al.
(2022) introduces two disentangled representations (eye and mouth) to provide improved conditions for NeRF.
To achieve few-shot training, DFRFShen et al.
(2022) conditions the face radiance ﬁeld on 2D reference images to learn the face prior, thus greatly reducing the required data scale (tens of seconds of video) and improve the convergence speed (about 40k iterations).
However, all of the previous NeRF-based work focuses on better image quality or reducing the training cost, while the generalizability to out-of-domain audio is relatively an oversight.
Our GeneFace could be regarded as bridging the advantages of the aforementioned two types of works.
Compared with previous 3DMM-based methods, our work could enjoy good 3D naturalness and high image quality brought by the NeRF-based renderer.
Compared with previous end-to-end
NeRF-based methods
, we improve the generalizabity to out-of-domain audio via introducing a generative audio-to-motion model trained on a large lip reading corpus.
G ENE F ACE
In this section, we introduce our proposed GeneFace.
As shown in Figure 1, GeneFace is composed of three parts: 1) a variational motion generator that transforms HuBERT features Hsu et al.
(2021) into 3D facial landmarks; 2) a post-net to reﬁne the generated motion into the target person domain; 3) a NeRF-based renderer to synthesize high-ﬁdelity frames.
We describe the designs and the training process of these three parts in detail in the following subsections.
3. V ARIATIONAL M OTION G ENERATOR
To achieve expressive and diverse 3D head motion generation, we introduce a variational autoencoder (VAE) to perform a generative and expressive audio-to-motion transform, namely the variational motion generator, as shown in Figure 2.
[Figure]
[Figure]
Audio and motion representation To better extract the semantic information, we utilize HuBERT, a state-of-the-art ASR model, to obtain audio features from the input wave and use it as the condition of the variational motion generator.
As for the motion representation, to represent detailed facial movement in Euclidean space, we select 68 key points from the reconstructed 3D head mesh and use their position as the action representations.
Speciﬁcally,
[Equation]
where LM 3 D ∈ R 68 × 3 , M and M are the 3DMM mesh and mean mesh deﬁned in Equation (1), I is the index of the key landmark in the mesh.
In this paper, we name this action representation 3D landmarks for abbreviation.
Dilated convolutional encoder and decoder Inspired by WaveNet, to better extract features from the audio sequence and construct long-term temporal relationships in the output sample, we design the encoder and decoder as fully convolutional networks where the convolutional layers have incrementally increased dilation factors that allow its receptive ﬁeld to grow exponentially with depth.
In contrast to previous works, which typically divide the input audio sequence into sliding windows to obtain a smooth result, we manage to synthesize the whole sequence of arbitrary length within a single forward.
To further improve the temporal stability of the predicted landmark sequence, a
Gaussian ﬁlter is performed to eliminate tiny ﬂuctuations in the result.
Flow-based Prior We also notice that the gaussian prior of vanilla VAE limits the performance of our 3D landmark sequence generation process from two prospectives: 1) the datapoint of each time index is independent of each other, which induces noise to the sequence generation task where there is a solid temporal correlation among frames.
2) optimizing VAE prior push the posterior distribution towards the mean, limiting diversity and hurting the generative power.
To this end, following Ren et al.
(2021), we utilize a normalizing ﬂow to provide a complex and time-related distribution as the prior distribution of the VAE.
Please refer to Appendix A.1 for more details.
Training Process Due to the introduction of prior ﬂow, the closed-form ELBO is not feasible, hence we use the Monte-Carlo ELBO loss Ren et al.
(2021) to train the VAE model.
Besides, inspired by Prajwal et al.
(2020), we independently train a sync-expert D sync that measures the possibility that the input audio and 3D landmarks are in-sync, whose training process can be found in
Appendix A.2 .
The trained sync-expert is then utilized to guide the training of VAE.
To summarize, the training loss of our variational motion generator (VG) is as follows:
[Equation]
where φ, θ, (cid:15) denote the model parameters of the encoder, decoder and the prior, respectively.
c denotes the condition features of VAE.
The ground truth and predicted 3D landmarks are represented by l and ˆ l , respectively.
3. D OMAIN A DAPTIVE P OST -N ET
As we train the variational motion generator on a large multi-speaker dataset, the model can generalize well with various audio inputs.
However, as the scale of the target person video is relatively tiny
[Figure] [Figure]
(about 4-5 minutes) compared with the multi-speaker lip reading dataset (about hundreds of hours), there exists a domain shift between the predicted 3D landmarks and the target person domain.
As a consequence, the NeRF-based renderer cannot generalize well with the predicted landmark, which results in blurry or unrealistic rendered images.
To this end, A naive solution is to ﬁne-tune the variational generator in the target person dataset.
The challenge is that we generally only have a short personalized video, and the generalizability of the model may be lost after the ﬁne-tuning.
Under such circumstances, we design a semi-supervised adversarial training pipeline to perform a domain adaptation.
To be speciﬁc, we learn a post-net to reﬁne the VAE-predicted 3D landmarks into the personalized domain.
We consider two requirements for this mapping: 1) it should preserve the temporal consistency and lip-synchronization of the input sequence; 2) it should correctly map each frame into the target person’s domain.
To fulﬁll the ﬁrst point, we utilize 1D CNN as the structure of post-net and adopt the sync-expert to supervise the lip-synchronization; for the second point, we jointly train an MLP-structured frame-level discriminator that measures the identity similarity of each landmark frame to the target person.
The detailed structure of the post-net and discriminator can be found in Appendix A.3.
Training Process
The training process of post-net is shown in Fig.3.
During training, the MLP discriminator tries to distinguish between the ground truth landmark l (cid:48) extracted from the target person’s video and the reﬁned samples G ω (ˆ l ) generated from the large-scale dataset.
We use the
LSGAN loss to update the discriminator :
[Equation] [Equation]
where ω and δ are the parameters of the post-net PN and discriminator D .
l (cid:48) is the ground truth 3DMM landmark of the target person dataset, and ˆ l is the 3D landmarks reﬁned by the post-net.
As for the training of post-net, the post-net competes with the discriminator while being guided by the pre-trained sync-expert to maintain lip synchronization.
Besides, we utilize the target person dataset to provide a weak supervised signal to help the adversarial training.
Speciﬁcally, we extract the audio c (cid:48) of the target person video for VAE to predict the landmarks ˆ l (cid:48) ∼ p θ ( l | z, c (cid:48) ) and encourage the reﬁned landmarks PN ω
(ˆ l (cid:48) ) to approximate the ground truth expression l (cid:48) .
Finally, the training loss of post-net is:
[Equation] [Equation]
3. N E RF- BASED R ENDERER
We obtain a robust and diverse audio-to-motion mapping through the variational motion generator and post-net.
Next, we design a NeRF-based renderer to render high-ﬁdelity frames conditioned on the predicted 3D landmarks.
[Figure]
3D landmark-conditioned NeRF Inspired by Guo et al.
(2021), we present a conditional NeRF to represent the dynamic talking head.
Apart from viewing direction d and 3D location x , the 3D landmarks l will act as the condition to manipulate the color and geometry of the implicitly represented head.
Speciﬁcally, the implicit function F can be formulated as follows:
[Equation] [Equation]
where c and σ denote the color and density in the radiance ﬁeld.
To improve the continuity between adjacent frames, we use the 3D landmarks from the three neighboring frames to represent the facial shape, i.e., l ∈ R 3 × 204 .
We notice that some facial landmarks only change in a small range, which numerically raises challenges for NeRF to learn the high-frequency image details.
Therefore, we normalize the input 3D landmarks point-wisely , which is necessary to achieve better visual quality.
Following the setting of volume rendering, to render each pixel, we emit a camera ray r ( t ) =
o + t · d in the radiance ﬁeld, with camera center o , viewing direction d .
The ﬁnal color C is calculated by aggregating the color c along the ray:
[Equation]
[Equation]
) where t n and t r is the near bound and far bound of ray r ; c θ and σ θ are the output of the implicit function F θ , T ( t ) is the accumulated transmittance along the ray from t n to t , which is deﬁned as:
[Equation]
Head-aware Torso
-NeRF To better model the head and torso movement, we train two NeRFs to render the head and torso parts, respectively.
As shown in Fig.
4, we ﬁrst train a head-NeRF to render the head part, then train a torso-NeRF to render the torso part with the rendering image of the head-NeRF as background.
Following Guo et al.
(2021), we assume the torso part is in canonical space and provide the head pose h to torso-NeRF as a signal to infer the torso movement.
The torso-NeRF implicitly learns to expect the location of the rendered head, then rigid the torso from canonical space to render a natural result.
However, this cooperation between head-NeRF and torso-NeRF is fragile since the torso-NeRF cannot observe the head-NeRF’s actual output.
Consequently, several recent works report that the torso-NeRF produces head-torso separation artifacts
[Table]
when the head pose is relatively large.
Based on the analysis above, we propose to provide the torso-NeRF with a perception of the rendering result of the head-NeRF.
Speciﬁcally, we use the output color C head of the head-NeRF as a pixel-wise condition of the torso-NeRF.
The torso’s implicit function F torso is expressed as:
[Equation]
where d 0 is view direction in the canonical space, Π ∈ R 3 × 4 is the head pose that composed of a rotation matrix and a transform vector.
Training Process We extract 3D landmarks from the video frames and use these landmark-image pairs to train our NeRF-based renderer.
The optimization target of head-NeRF and torso-NeRF is to reduce the photo-metric reconstruction error between rendered and ground-truth images.
Speciﬁcally, the loss function can be formulated as:
[Equation] [Equation]
where R is the set of camera rays, C g is the color of the ground image.
E XPERIMENTS 4. D ATASET PREPARATION AND PREPROCESSING
Dataset preparation.
Our method aims to synthesize high-ﬁdelity talking face images with great generalizability to out-domain audio.
To learn robust audio-to-motion mapping, a large-scale lipreading corpus is needed.
Hence we use LRS3-TEDAfouras et al.
(2018) to train our variational generator and post-net 2 .
Additionally, a certain person’s speaking video of a few minutes in length with an audio track is needed to learn a NeRF-based person portrait renderer.
To be speciﬁc, in order to compare with the state-of-the-art method, we utilize the data set of Lu et al.
(2021) and Guo et al.
(2021), which consist of 5 videos of an average length of 6,000 frames in 25 fps.
Data preprocessing.
As for the audio track, we downsample the speech wave into the sampling rate of 16000 and process it with a pretrained HuBERT model.
For the video frames of LRS3 and the target person videos, we resample them into 25 fps and use Deng et al.
(2019) to extract the head pose and 3D landmarks.
As for the target person videos, they are cropped into 512x512 and each frame is processed with the help of an automatic parsing method Lee et al.
(2020) for segmenting the head and torso part and extracting a clean background.
4. E XPERIMENTAL S ETTINGS
Comparison baselines.
We compare our GeneFace with several remarkable works: 1)
Wav2Lip
Prajwal et al.
(2020), which pretrain a sync-expert to improve the lip-synchronization performance; 2) MakeItTalk Zhou et al.
(2020), which also utilize 3D landmark as the action representation; 3)
PC-AVS Zhou et al.
(2021), which ﬁrst modularize the audio-visual representation.
LiveSpeechPortriat Lu et al.
(2021), which achieves photorealistic results at over 30fps; 5) AD-NeRF Guo et al.
(2021), which ﬁrst utilize NeRF to achieve talking head generation.
For Wav2Lip, PC-AVS, and
MakeItTalk, the LRS3-TED dataset is used to train the model, and a reference clip of the target person video is used during the inference stage; for LSP, both of LRS3-TED dataset and the target person video is used to train the model; for the NeRF-based method, AD-NeRF, only the target person video is used to train an end-to-end audio-to-image renderer.
Implementation Details.
We train the GeneFace on 1 NVIDIA RTX 3090 GPU, and the detailed training hyper-parameters of the variational generator, post-net, and NeRF-based are listed in Appendix B. For variational generator and post-net, it takes about 40k and 12k steps to converge (about 12 hours).
For the NeRF-based renderer, we train each model for 800k iterations (400k for head and 400k for the torso, respectively), which takes about 72 hours.
4. Q UANTITATIVE E VALUATION
Evaluation Metrics We employ the FID score Heusel et al.
(2017) to measure image quality.
We utilize the landmark distance (LMD)Chen et al.
and syncnet conﬁdence score Prajwal et al.
(2020) to evaluate lip synchronization.
Furthermore, to evaluate the generalizability, we additionally test all methods with a set of out-of-domain (OOD) audio, which consists of cross-lingual, crossgender, and singing voice audios.
Evaluation Results The results are shown in Table 1.
We have the following observations.
Our
GeneFace achieves good lip-synchronization with high generalizability.
Since Wav2Lip is jointly trained with SyncNet, it achieves the highest sync score that is higher than the ground truth video.
Our method performs best in LMD and achieves a better sync score than other baselines.
When tested with out-of-domain audios, while the sync-score of person-speciﬁc methods (LSP and ADNeRF) signiﬁcantly drops, GeneFace maintains good performance.
(2) Our GeneFace achieves the best visual quality.
We observe that one-shot methods (Wav2Lip, MakeItTalk, and PC-AVS) perform
[Table] [Table]
[Figure]
poorly on FID due to low image ﬁdelity.
Since we use 3D landmarks as the condition of the NeRF renderer, it address the mean face problem and leads to better lip syncronization and visual quality than AD-NeRF.
4. Q UALITATIVE E VALUATION
To compare the generated results of each method, we show the keyframes of two clips in Fig.9.
Due to space limitations, we only compare our GeneFace with AD-NeRF here and provide full results with all baselines in Appendix C.1.
We observe that although both methods manage to generate high-ﬁdelity results, GeneFace solves several problems that AD-NeRF has: 1) head-torso separation (brown arrow) due to the separate generation pipeline of head and torso part; 2) blurry mouth images due to the one-to-many audio-to-lip mapping; 3) unsynchronized lip due to the weak generalizability.
User Study We conduct user studies to test the quality of audio-driven portraits.
Speciﬁcally, we sample 10 audio clips from English, Chinese, and German for all methods to generate the videos, and then involve 20 attendees for user studies.
We adopt the Mean Opinion score (MOS) rating protocol for evaluation, which is scaled from 1 to 5.
The attendees are required to rate the videos based on three aspects: (1) lip-sync accuracy; (2) video realness; (3) image quality.
We compute the average score for each method, and the results are shown in Table 2.
We have the following observations: 1) Our GeneFace achieves comparatively high lip-sync accuracy with
Wav2LipPrajwal et al.
(2020) since both of them learn a generalized audio-to-motion mapping on a large dataset with guidance from a sync-expert. 2)
As for the video realness and image quality, the
Person-speciﬁc methods (LSP, AD-NeRF, and GeneFace) outperform one-shot methods (Wav2Lip,
MakeItTalk, and PC-AVS).
Although LSP has slightly better image quality than GeneFace, our method achieves the highest video realness and lip-sync accuracy score.
4. A BLATION S TUDY
In this section, we perform ablation study to prove the necessity of each component in GeneFace.
[Table]
[Table]
Varaiational motion generator We test two settings on the variational motion generator: (1) w/o prior ﬂow, where we replace the ﬂow-based prior with a gaussian prior.
The results are shown in
Table 3 (line 2), where the Sync score drops by a relatively large margin.
This observation suggests that the temporal enhanced latent variable contributes to the stability of the predicted landmark sequence.
(2) w/o sync-expert (line 3), where the variational motion generator is no longer supervised by a pretrained sync-expert.
We observe that it leads to a signiﬁcant degradation in Sync score.
To evaluate the contribution of the variational generator to the quality of the predicted landmark, we adopt L2 error on the predicted 3D landmarks as the metric.
We compare our vairiaitonal generator (VAE+Flow) against vanilla VAE and a simple regression model trained with MSE loss.
The results are listed in Table 5 in Appendix C.2.
It can be seen that removing the prior ﬂow or using a regression-based model leads to a performance drop.
Domain adaptative post-net
In the setting w/o post-net, we remove the domain adaptative postnet, the results are shown in Table 3 (line 4).
It can be seen that directly using the 3D landmarks predicted by the variational motion generator leads to a signiﬁcant performance drop in FID and
Sync scores.
To further investigate the efﬁcacy of post-net, we utilize T-SNE to visualize the landmarks of different domains in Figure 10.
The visualization results prove that there exists a signiﬁcant domain gap between the LRS3 dataset and the target person video, and our post-net successfully rigs the predicted landmarks from the LRS3 domain into the target person domain.
We also try to replace the post-net with directly ﬁne-tuning on the target person video (line 5), although it achieves a competitive sync score on in-domain audios, its performance in OOD audio is worse.
Head-aware torso-NeRF
In the w/o head-ware setting, we remove head image condition of the torso-NeRF.
The results are shown in Table 3 (line 6).
Due to the unawareness of the head’s location, the head-torso separation occurs occasionally, which results in a drop in the FID score.
C ONCLUSION
In this paper, we propose GeneFace for talking face generation, which aims to solve the weak generalizability and mean face problem faced by previous NeRF-based methods.
A variational motion generator is proposed to construct a generic audio-to-motion mapping based on a large corpus.
We then introduce a domain adaptative post-net with an adversarial training pipeline to rig the predicted motion representation into the target person domain.
Moreover, a head-aware torso-NeRF is present to address the head-torso separation issue.
Extensive experiments show that our method achieves more generalized and high-ﬁdelity talking face generation compared to previous methods.
Due to space limitations, we discuss the limitations and future work in Appendix D.
A D ETAILS OF M ODELS A. V ARIATIONAL M OTION G ENERATOR
Following PortaSpeech, our variational motion generator consists of an encoder, a decoder, and a ﬂow-based prior model.
The encoder, as shown in Fig.
6a, is composed of a 1D-convolution followed by ReLU activation and layer normalization, and a non-causal WaveNet.
The decoder, as shown in Fig.
6b, consists of a non-causal WaveNet and a 1D transposed convolution followed by
ReLU and layer normalization.
The prior model, as shown in Fig.
6c, is a normalizing ﬂow, which is composed of a 1D-convolution coupling layer and a channel-wise ﬂip operation.
HuBERT features are utilized as the audio condition of these three modules.
A. S YNC - EXPERT
Our sync-expert inputs a window of T l consecutive 3D landmark frames and an audio feature clip of size T a × D , where T l and T a are the lengths of the video and audio clip respectively, and D is the dimension of HuBERT features.
The sync-expert is trained to discriminate whether the input audio and landmarks are synchronized.
It consists of a landmark encoder and an audio encoder, as shown in Fig.
7, both of which are comprised of a stack of 1D-convolutions followed by batch normalization and ReLU.
We use cosine-similarity with binary cross-entropy loss to train the sync-expert.
Speciﬁcally, we compute cosine-similarity for the landmark embedding l and audio embedding a to represent the probability that the input audio-landmark pair is synchronized.
The training loss of sync-expert can be represented as:
[Equation]
A. D OMAIN A DAPTATIVE P OST - NET AND D ISCRIMINATOR
The domain adaptative post-net, as shown in Fig. 8, is composed of a stack of residual 1Dconvolution followed by ReLU and batch normalization.
The discriminator is a MLP composed of a stack of fully connected layers followed with ReLU and dropout.
B D ETAILED E XPERIMENTAL S ETTINGS
[Figure]
BN
ReLU x N ℎ𝑢𝑏𝑒𝑟𝑡 𝑐𝑙𝑖𝑝
Conv1D+BN
ReLU x N 𝑙𝑎𝑛𝑑𝑚𝑎𝑟𝑘 𝑐𝑙𝑖𝑝 cosine similarity
Pos/Neg?
Conv1D + ReLU +Batch Norm 𝑐𝑜𝑎𝑟𝑠𝑒 𝑙𝑎𝑛𝑑𝑚𝑎𝑟𝑘 𝑟𝑒𝑓𝑖𝑛𝑒𝑑 𝑙𝑎𝑛𝑑𝑚𝑎𝑟𝑘 + x N
B. M ODEL C ONFIGURATIONS
We list the hyper-parameters of GeneFace in Tab. 4.
[Table]
GeneFace
Variational Motion Generator
Encoder Layers
Decoder Layers
Encoder/Decoder Conv1D Kernel
Encoder/Decoder Conv1D Channel Size
Latent Size
Prior Flow Layers
Prior Flow Conv1D Kernel
Prior Flow Conv1D Channel Size
Sync-expert Layers
Sync-expert Channel Size
Post-net
and
Discriminator
Post-net Layers
Post-net Conv1D Kernel
Post-net Conv1D Channel Size
Discrimnator Layers
Discrimnator Linear Hidden Size
Discrimnator Dropout Rate 0.
NeRF-based Renderer
Head/Torso-NeRF Layers
Head/Torso-NeRF Hidden Size
Landmark/Head Color Encoder Layers
Landmark/Head Color Encoder Hidden Size
[Equation]
MakeItTalk
Wav2Lip /i/ /w/ /s/ /ɒ/
/ju:/ Audio
C A DDITIONAL E XPERIMENTS C. Q UALITATIVE R ESULTS WITH ALL BASELINES
To compare the generated results of each method, we show the keyframes of one in-domain audio clip in Fig.9.
We have the following observations: 1) Wav2Lip achieves competitive lip-sync performance yet generates blurry mouth results; 2) MakeItTalk and PC-AVS fail to preserve the speaker’s identity, leading to unrealistic generated results; 3) LSP generates unnatural lip movement during the transition phase of different syllables.
Please see our supplementary video for better visualization.
C. E VALUATION ON 3D L ANDMARK L2 ERROR
To evaluate the contribution of the variational generator to the quality of the predicted landmark, we adopt L2 error on the predicted 3D landmarks as the metric.
We compare our vairiaitonal generator (VAE+Flow) against vanilla VAE and a simple regression model trained with MSE loss.
The results are listed in Table 5.
It can be seen that removing the prior ﬂow or using a regression-based model leads to a performance drop.
[Table]
C. T-SNE V ISUALIZATION FOR D OMAIN A DAPTATION
To further investigate the efﬁcacy of post-net, we utilize T-SNE to visualize the landmarks of different domains in Figure 10.
The visualization results prove that there exists a signiﬁcant domain gap between the LRS3 dataset and the target person video, and our post-net successfully rigs the predicted landmarks from the LRS3 domain into the target person domain.
D L IMITATIONS AND F UTURE W ORK
There are mainly two limitations of the proposed approach.
Firstly, we found the landmark sequence generated by variational motion generator and post-net occasionally has tiny ﬂuctuations, which results in some artifacts such as shaking hairs, etc.
Currently, we utilize a heuristic post-processing method (Gaussian ﬁlter) to alleviate this problem.
In future work, we will explore better modeling the temporal information in the network architecture to further improve the stability.
Secondly, the current NeRF-based renderer is majorly based on the setting of vanilla NeRF, which results in a long training and inference time.
In future work, we will try to enhance the performance of the NeRF backend by combining recent progress in accelerated and light-weight NeRF.