Add metadata and improve model card

README.md

@@ -1,21 +1,36 @@
+---
+pipeline_tag: other
+tags:
+- recommendation
+- generative-recommender
+---
+
 # MemGen Checkpoints
 
-The model checkpoints and artifacts for **On the Memorization and Generalization of Generative Recommendation**.
+This repository contains model checkpoints and artifacts for the paper **How Well Does Generative Recommendation Generalize?**.
 
 <a href="https://huggingface.co/papers/2603.19809"><img src="https://img.shields.io/badge/Paper-ArXiv-red"></a>
 <a href="https://github.com/Jamesding000/MemGen-GR"><img src="https://img.shields.io/badge/Code-GitHub-black"></a>
 <a href="https://huggingface.co/datasets/jamesding0302/memgen-annotations"><img src="https://img.shields.io/badge/Data-Hugging%20Face-yellow"></a>
 
+## Overview
+
+A widely held hypothesis for why generative recommendation (GR) models outperform conventional item ID-based models is that they generalize better. This work provides a systematic way to verify this hypothesis by categorizing data instances into those requiring **memorization** (reusing item transition patterns observed during training) and those requiring **generalization** (composing known patterns to predict unseen item transitions). 
+
+The study shows that GR models perform better on instances that require generalization, whereas item ID-based models perform better when memorization is more important. The authors propose a simple memorization-aware indicator that adaptively combines both paradigms to improve overall performance.
+
 ## Folder Structure
 
 - `SASRec/`: SASRec checkpoints (`.pth`)
 - `TIGER/`: TIGER checkpoints (`.pth`)
 - `semantic_ids/`: semantic ID files (`.sem_ids`) used to train the corresponding TIGER checkpoints
 
-All checkpoints are trained with the **default configs** in the code repo. If you are unsure about model size / architecture, please refer to the configs in [Code].
+All checkpoints are trained with the **default configs** in the code repo. If you are unsure about model size / architecture, please refer to the configs in the [GitHub repository](https://github.com/Jamesding000/MemGen-GR).
 
 ## Download (example)
 
+You can use the `huggingface_hub` library to download specific checkpoints and semantic ID files:
+
 ```python
 from huggingface_hub import hf_hub_download
 
@@ -37,8 +52,9 @@ sem_ids_path = hf_hub_download(
 
 ## Evaluation
 
-You can run fine-grained evaluation using the saved `checkpoint_path` and `sem_ids_path`:
-```
+You can run fine-grained evaluation using the saved `checkpoint_path` and `sem_ids_path` with the scripts provided in the official repository:
+
+```bash
 CUDA_VISIBLE_DEVICES=0 python mem_gen_evaluation.py \
     --model=TIGER \
     --dataset=AmazonReviews2014 \
@@ -47,4 +63,15 @@ CUDA_VISIBLE_DEVICES=0 python mem_gen_evaluation.py \
     --sem_ids_path=path/to/semantic_ids.sem_ids \
     --eval=test \
     --save_inference
+```
+
+## Citation
+
+```bibtex
+@article{ding2026how,
+  title={How Well Does Generative Recommendation Generalize?},
+  author={Ding, Yijie and Guo, Zitian and Li, Jiacheng and Peng, Letian and Shao, Shuai and Shao, Wei and Luo, Xiaoqiang and Simon, Luke and Shang, Jingbo and McAuley, Julian and Hou, Yupeng},
+  journal={arXiv preprint arXiv:2603.19809},
+  year={2026}
+}
 ```