Update README.md

README.md

@@ -9,11 +9,15 @@ tags:
 - transformer
 library_name: transformers
 pipeline_tag: text-generation
+arxiv: 2602.02522
 ---
 
 # IMU-1 Base
 
-A sample-efficient 430M parameter language model trained on 72B tokens that approaches the benchmark performance of models trained on 56× more data.
+This repository contains the IMU-1 Base model, a sample-efficient 430M parameter language model introduced in the paper [IMU-1: Sample-Efficient Pre-training of Small Language Models](https://huggingface.co/papers/2602.02522).
+
+IMU-1 is trained on 72B tokens and approaches the benchmark performance of models trained on 56× more data.
+
 
 ## Model Details