Add arXiv badge

README.md

@@ -11,10 +11,15 @@ datasets:
   - imagenet-1k
 ---
 
-# Decoupled Fréchet Loss Improves Visual Generation
+# Representation Fréchet Loss for Visual Generation
+
+[![arXiv](https://img.shields.io/badge/arXiv-2604.28190-b31b1b.svg)](https://arxiv.org/abs/2604.28190)
+[![GitHub](https://img.shields.io/badge/GitHub-code-181717.svg)](https://github.com/Jiawei-Yang/FD-Loss)
 
 This repository hosts the released checkpoints and reference data for
-**Decoupled Fréchet Loss Improves Visual Generation**.
+**Representation Fréchet Loss for Visual Generation**.
+
+Paper: [Representation Fréchet Loss for Visual Generation](https://arxiv.org/abs/2604.28190).
 
 Code, training scripts, and evaluation utilities are available at:
 [github.com/Jiawei-Yang/FD-Loss](https://github.com/Jiawei-Yang/FD-Loss).
@@ -149,8 +154,10 @@ torchrun --nproc_per_node=8 scripts/compute_valfd.py \
 
 ```bibtex
 @article{yang2026fdloss,
-  title={Decoupled Fréchet Loss Improves Visual Generation},
+  title={Representation Fréchet Loss for Visual Generation},
   author={Yang, Jiawei and Geng, Zhengyang and Ju, Xuan and Tian, Yonglong and Wang, Yue},
+  journal={arXiv:2604.28190},
+  url={https://arxiv.org/abs/2604.28190},
   year={2026}
 }
 ```