English

Representation Fr\'echet Loss for Visual Generation

Computer Vision and Pattern Recognition 2026-05-01 v1

Abstract

We show that Fr\'echet Distance (FD), long considered impractical as a training objective, can in fact be effectively optimized in the representation space. Our idea is simple: decouple the population size for FD estimation (e.g., 50k) from the batch size for gradient computation (e.g., 1024). We term this approach FD-loss. Optimizing FD-loss reveals several surprising findings. First, post-training a base generator with FD-loss in different representation spaces consistently improves visual quality. Under the Inception feature space, a one-step generator achieves0.72 FID on ImageNet 256x256. Second, the same FD-loss repurposes multi-step generators into strong one-step generators without teacher distillation, adversarial training or per-sample targets. Third, FID can misrank visual quality: modern representations can yield better samples despite worse Inception FID. This motivates FDrk^k, a multi-representation metric. We hope this work will encourage further exploration of distributional distances in diverse representation spaces as both training objectives and evaluation metrics for generative models.

Keywords

Cite

@article{arxiv.2604.28190,
  title  = {Representation Fr\'echet Loss for Visual Generation},
  author = {Jiawei Yang and Zhengyang Geng and Xuan Ju and Yonglong Tian and Yue Wang},
  journal= {arXiv preprint arXiv:2604.28190},
  year   = {2026}
}

Comments

Code and checkpoints are available at https://github.com/Jiawei-Yang/FD-loss

R2 v1 2026-07-01T12:44:08.511Z