HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Few-Shot Adversarial Learning of Realistic Neural Talking Head Models

Egor Zakharov; Aliaksandra Shysheya; Egor Burkov; Victor Lempitsky

Few-Shot Adversarial Learning of Realistic Neural Talking Head Models

Abstract

Several recent works have shown how highly realistic human head images can be obtained by training convolutional neural networks to generate them. In order to create a personalized talking head model, these works require training on a large dataset of images of a single person. However, in many practical scenarios, such personalized talking head models need to be learned from a few image views of a person, potentially even a single image. Here, we present a system with such few-shot capability. It performs lengthy meta-learning on a large dataset of videos, and after that is able to frame few- and one-shot learning of neural talking head models of previously unseen people as adversarial training problems with high capacity generators and discriminators. Crucially, the system is able to initialize the parameters of both the generator and the discriminator in a person-specific way, so that training can be based on just a few images and done quickly, despite the need to tune tens of millions of parameters. We show that such an approach is able to learn highly realistic and personalized talking head models of new people and even portrait paintings.

Code Repositories

grey-eye/talking-heads
pytorch
Mentioned in GitHub
Ierezell/PapierFewShot
pytorch
Mentioned in GitHub
ZVK/Talking-Heads
pytorch
Mentioned in GitHub
ZVK/talking_heads
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
talking-head-generation-on-voxceleb1-1-shotFew-shot Adversarial Model
FID: 43.0
talking-head-generation-on-voxceleb1-32-shotFew-shot Adversarial Model
FID: 29.5
talking-head-generation-on-voxceleb1-8-shotFew-shot Adversarial Model
FID: 38.0
talking-head-generation-on-voxceleb2-1-shotFew-shot Adversarial Model
FID: 48.5
talking-head-generation-on-voxceleb2-32-shotFew-shot Adversarial Model
FID: 30.6
talking-head-generation-on-voxceleb2-8-shotFew-shot Adversarial Model
FID: 42.2

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Few-Shot Adversarial Learning of Realistic Neural Talking Head Models | Papers | HyperAI