diff --git a/basicsr/models/archs/NAFNet_arch.py b/basicsr/models/archs/NAFNet_arch.py
index 638a0bd..119786b 100644
--- a/basicsr/models/archs/NAFNet_arch.py
+++ b/basicsr/models/archs/NAFNet_arch.py
@@ -5,6 +5,12 @@
 '''
 Simple Baselines for Image Restoration
 
+@article{chen2022simple,
+  title={Simple Baselines for Image Restoration},
+  author={Chen, Liangyu and Chu, Xiaojie and Zhang, Xiangyu and Sun, Jian},
+  journal={arXiv preprint arXiv:2204.04676},
+  year={2022}
+}
 '''
 
 import torch