Retrieval-Augmented Gaussian Avatars: Improving Expression Generalization
Organizations: The Hebrew University of Jerusalem, Israel · OriginAI, Israel · Bar-Ilan University, Israel · NVIDIA ·
Abstract
Template-free animatable head avatars can achieve high visual fidelity by learning expression-dependent facial deformation directly from a subject's capture, avoiding parametric face templates and hand-designed blendshape spaces. However, since learned deformation is supervised only by the expressions observed for a single identity, these models suffer from limited expression coverage and often struggle when driven by motions that deviate from the training distribution. We introduce RAF (Retrieval-Augmented Faces), a simple training-time augmentation designed for template-free head avatars that learn deformation from data. RAF constructs a large unlabeled expression bank and, during training, replaces a subset of the subject's expression features with nearest-neighbor expressions retrieved from this bank while still reconstructing the subject's original frames. This exposes the deformation field to a broader range of expression conditions, encouraging stronger identity-expression decoupling and improving robustness to expression distribution shift without requiring paired cross-identity data, additional annotations, or architectural changes. We further analyze how retrieval augmentation increases expression diversity and validate retrieval quality with a user study showing that retrieved neighbors are perceptually closer in expression and pose. Experiments on the NeRSemble benchmark demonstrate that RAF consistently improves expression fidelity over the baseline, in both self-driving and cross-driving scenarios.
Figures & tables
| Subject | MMD | KL | B2T Dist | |||
|---|---|---|---|---|---|---|
| Vanilla | +RAF | Vanilla | +RAF | Vanilla | +RAF | |
| 388 | 0.254 | 0.216 | 38.948 | 25.365 | 1.398 | 0.796 |
| 422 | 0.186 | 0.128 | 50.575 | 31.425 | 1.001 | 0.560 |
| 443 | 0.131 | 0.103 | 33.078 | 19.658 | 0.763 | 0.533 |
| 445 | 0.152 | 0.089 | 56.310 | 32.658 | 0.872 | 0.499 |
| 475 | 0.213 | 0.167 | 35.662 | 21.417 | 1.023 | 0.583 |
| Method | Self-Driving | Cross-Driving | ||||||
|---|---|---|---|---|---|---|---|---|
| PSNR | SSIM | AED | APD | Emotion Sim. | AED | APD | Emotion Sim. | |
| Original | 18.038 | 0.7881 | 0.24227 | 0.13409 | 0.82016 | 0.27037 | 0.12289 | 0.78746 |
| Random Noise | 18.004 | 0.7890 | 0.24998 | 0.13023 | 0.81604 | 0.26920 | 0.12105 | 0.78770 |
| Ours (RAF) | 18.002 | 0.7917 | 0.23094 | 0.12932 | 0.85595 | 0.26164 | 0.12687 | 0.80801 |
| Method | Self-Driving | Cross-Driving | ||||||
|---|---|---|---|---|---|---|---|---|
| PSNR | SSIM | AED | APD | Emotion Sim. | AED | APD | Emotion Sim. | |
| Half bank #1 | 18.070 | 0.7926 | 0.23513 | 0.13226 | 0.85169 | 0.27079 | 0.12495 | 0.80305 |
| Half bank #2 | 18.137 | 0.7941 | 0.24693 | 0.13225 | 0.85616 | 0.26828 | 0.12295 | 0.80733 |
| RAF (top 5) | - | - | 0.23927 | 0.17058 | 0.88228 | 0.26687 | 0.15089 | 0.83073 |
| RAF | 18.002 | 0.7917 | 0.23094 | 0.12932 | 0.85595 | 0.26164 | 0.12687 | 0.80801 |
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
Human Avatar Animation
Human avatar animation builds digital people that can be reconstructed, driven by motion or speech, and rendered consistently across poses and views.
See all 91 papers