Update README.md

README.md

@@ -1,3 +1,61 @@
----
-license: apache-2.0
----
+---
+license: apache-2.0
+language:
+- en
+base_model:
+- Qwen/Qwen2.5-0.5B
+---
+
+## Model Description
+
+This Memory Decoder model is trained on the Law domain and can be adapted to enhance any model in the Qwen2 and Qwen2.5 families. 
+
+**Paper:** [Memory Decoder: A Pretrained, Plug-and-Play Memory for Large Language Models](https://www.arxiv.org/abs/2508.09874)
+
+**GitHub:** [https://github.com/LUMIA-Group/MemoryDecoder](https://github.com/LUMIA-Group/MemoryDecoder/tree/main)
+
+## Training & Evaluation Data
+
+**Law Domain Dataset:** [AsyLex](https://huggingface.co/datasets/clairebarale/AsyLex)
+
+**Test Split:** [MemoryDecoder-domain-data](https://huggingface.co/datasets/Clover-Hill/MemoryDecoder-domain-data)
+
+## Performance Results
+
+### Qwen2 Family
+
+| Model | Base Model | Base + MemDec |
+|-------|------------|---------------|
+| Qwen2-0.5B | 10.23 | 4.57 |
+| Qwen2-1.5B | 7.69 | 4.32 |
+| Qwen2-7B | 5.92 | 4.00 |
+| Qwen2-72B | 4.84 | 3.69 |
+
+### Qwen2.5 Family
+
+| Model | Base Model | Base + MemDec |
+|-------|------------|---------------|
+| Qwen2.5-0.5B | 9.86 | 4.57 |
+| Qwen2.5-1.5B | 7.42 | 4.29 |
+| Qwen2.5-3B | 6.68 | 4.16 |
+| Qwen2.5-7B | 5.94 | 4.01 |
+| Qwen2.5-14B | 5.35 | 3.86 |
+| Qwen2.5-32B | 5.18 | 3.81 |
+| Qwen2.5-72B | 4.84 | 3.70 |
+
+*Perplexity scores on Law domain test set. Lower is better.*
+
+## Citation
+
+```bibtex
+@article{cao2025memory,
+  title={Memory decoder: A pretrained, plug-and-play memory for large language models},
+  author={Cao, Jiaqi and Wang, Jiarui and Wei, Rubin and Guo, Qipeng and Chen, Kai and Zhou, Bowen and Lin, Zhouhan},
+  journal={arXiv preprint arXiv:2508.09874},
+  year={2025}
+}
+```
+
+## Contact
+
+For questions and support: [email protected]