Improve model card: Add pipeline tag, paper, code links, and citation

README.md

@@ -1,13 +1,15 @@
 ---
-license: cc-by-nc-sa-4.0
 datasets:
 - HorizonTEL/AIGIBench
+license: cc-by-nc-sa-4.0
+pipeline_tag: image-classification
 ---
+
 <div align="center">
  <br>
 <h1>Is Artificial Intelligence Generated Image Detection a Solved Problem?</h1>
  
-[Ziqiang Li](https://scholar.google.com/citations?user=mj5a8WgAAAAJ&hl=zh-CN)<sup>1</sup>, [Jiazhen Yan](https://scholar.google.com/citations?user=QkURh8EAAAAJ&hl=zh-CN)<sup>1</sup>, [Ziwen He](https://scholar.google.com/citations?user=PjkDK9cAAAAJ&hl=zh-CN)<sup>1</sup>, [Kai Zeng](https://scholar.google.com.hk/citations?user=TsI93SIAAAAJ&hl=zh-CN)<sup>2</sup>, [Weiwei Jiang](https://scholar.google.co.jp/citations?user=mbPN0hgAAAAJ&hl=zh-CN)<sup>1</sup>, [Lizhi Xiong](https://scholar.google.com/citations?user=-FzrEP4AAAAJ&hl=zh-CN)<sup>1</sup>, [Zhangjie Fu](https://scholar.google.com/citations?user=fO9NmagAAAAJ&hl=zh-CN)<sup>1‡</sup>
+[Ziqiang Li](https://scholar.google.com/citations?user=mj5a8WgAAAAJ&hl=zh-CN)<sup>1</sup>, [Jiazhen Yan](https://scholar.google.com/citations?user=QkURh8EAAAAJ&hl=zh-CN)<sup>1</sup>, [Ziwen He](https://scholar.google.com/citations?user=PjkDK9cAAAAJ&hl=zh-CN)<sup>1</sup>, [Kai Zeng](https://scholar.google.com.hk/citations?user=TsI93SIAAAAJ&hl=zh-CN)<sup>2</sup>, [Weiwei Jiang](https://scholar.co.jp/citations?user=mbPN0hgAAAAJ&hl=zh-CN)<sup>1</sup>, [Lizhi Xiong](https://scholar.google.com/citations?user=-FzrEP4AAAAJ&hl=zh-CN)<sup>1</sup>, [Zhangjie Fu](https://scholar.google.com/citations?user=fO9NmagAAAAJ&hl=zh-CN)<sup>1‡</sup>
 
 
 <div class="is-size-6 publication-authors">
@@ -22,7 +24,20 @@ datasets:
 
 **This repository is the official pre-trained checkpoints of the AIGIBench in Setting-II**: Training on 144K images generated by both SD-v1.4 and ProGAN, covering the same four object categories.
 
+This work was presented in the paper [Is Artificial Intelligence Generated Image Detection a Solved Problem?](https://huggingface.co/papers/2505.12335), also available on [arXiv](https://arxiv.org/abs/2505.12335).
+The official code can be found on [GitHub](https://github.com/HorizonTEL/AIGIBench).
+
 If this project helps you, please fork, watch, and give a star to this repository.  
 
+## Citation
+```bibtex
+@inproceedings{li2025artificial,
+  title={Is Artificial Intelligence Generated Image Detection a Solved Problem?},
+  author={Li, Ziqiang and Yan, Jiazhen and He, Ziwen and Zeng, Kai and Jiang, Weiwei and Xiong, Lizhi and Fu, Zhangjie},
+  booktitle={Advances in Neural Information Processing Systems},
+  year={2025}
+}
+```
+
 ## Contact
 If you have any question about this project, please feel free to contact 247918horizon@gmail.com