Update README.md

README.md

@@ -22,9 +22,9 @@ tags:
   <a href="https://discord.gg/P2yYH95N" target="_blank" style="margin: 2px;">
     <img alt="Discord" src="https://img.shields.io/badge/Discord-Small%20Doges-7289da?logo=discord&logoColor=white&color=7289da" style="display: inline-block; vertical-align: middle;"/>
   </a>
-  <a href="https://arxiv.org/abs/2412.11834" target="_blank" style="margin: 2px;">
+  <!-- <a href="https://arxiv.org/abs/2412.11834" target="_blank" style="margin: 2px;">
     <img alt="arXiv" src="https://img.shields.io/static/v1?label=arXiv&message=2412.11834&color=B31B1B&logo=arXiv" style="display: inline-block; vertical-align: middle;"/>
-  </a>
+  </a> -->
   <a href="https://github.com/SmallDoges/small-doge" target="_blank" style="margin: 2px;">
     <img alt="GitHub" src="https://img.shields.io/badge/GitHub-SmallDoge-181717?logo=github" style="display: inline-block; vertical-align: middle;"/>
   </a>
@@ -33,7 +33,8 @@ tags:
   </a>
 </div>
 
-Doge uses Dynamic Mask Attention as sequence transformation and can use Multi-Layer Perceptron or Cross Domain Mixture of Experts as state transformation. Dynamic Mask Attention allows the Transformer to use self-attention during training and state space during inference, and Cross Domain Mixture of Experts can directly inherit the weights of Multi-Layer Perceptron for further training. This model is trained by [SmallDoge](https://huggingface.co/SmallDoge) community, for detailed algorithm and model architecture, please refer to [Wonderful Matrices](https://arxiv.org/abs/2412.11834), all training details and code are publicly available on the [small-doge](https://github.com/SmallDoges/small-doge) repository.
+
+Doge uses Dynamic Mask Attention as sequence transformation and can use Multi-Layer Perceptron or Cross Domain Mixture of Experts as state transformation. Dynamic Mask Attention allows the Transformer to use self-attention during training and state space during inference, and Cross Domain Mixture of Experts can directly inherit the weights of Multi-Layer Perceptron for further training. This model is trained by [SmallDoge](https://huggingface.co/SmallDoge) community, for detailed algorithm and model architecture, paper coming soon, all training details and code are available in the [small-doge](https://github.com/SmallDoges/small-doge) repository.
 
 
 ## Uses
@@ -70,8 +71,8 @@ We build the Doge by doing Per-Training on [Smollm-Corpus](https://huggingface.c
 |---|---|---|---|---|---|---|---|---|
 | [Doge-20M](https://huggingface.co/SmallDoge/Doge-20M) | 25.4 | 0.03 | 29.8 | 58.4 | 27.3 | 25.6 | 50.2 | 142 |
 | [Doge-60M](https://huggingface.co/SmallDoge/Doge-60M) | 26.4 | 0.2 | 37.9 | 61.4 | 31.5 | 28.0 | 50.8 | 62 |
-| [Doge-160M](https://huggingface.co/SmallDoge/Doge-160M) | 29.2 | 4.8 | 44.4 | 66.3 | 38.7 | 34.4 | 52.2 | 28 |
-| [Doge-320M](https://huggingface.co/SmallDoge/Doge-320M) | 33.8 | 9.4 | 52.1 | 69.9 | 46.5 | 37.9 | 55.0 | 16 |
+| [Doge-160M](https://huggingface.co/SmallDoge/Doge-160M) | 29.2 | 4.8 | 44.4 | 70.1 | 43.4 | 34.4 | 52.2 | 28 |
+| [Doge-320M](https://huggingface.co/SmallDoge/Doge-320M) | 33.8 | 9.4 | 52.1 | 73.9 | 52.7 | 37.9 | 55.0 | 16 |
 
 > [!NOTE]
 > All evaluations are done using five-shot settings, without additional training on the benchmarks.
@@ -91,13 +92,10 @@ We build the Doge by doing Per-Training on [Smollm-Corpus](https://huggingface.c
 ## Citation
 
 ```bibtex
-@misc{shi2024wonderfulmatrices,
-      title={Wonderful Matrices: Combining for a More Efficient and Effective Foundation Model Architecture}, 
-      author={Jingze Shi and Bingheng Wu},
-      year={2024},
-      eprint={2412.11834},
-      archivePrefix={arXiv},
-      primaryClass={cs.LG},
-      url={https://arxiv.org/abs/2412.11834}, 
+@misc{smalldoges,
+  title={SmallDoges},
+  author={SmallDoge Team and Jingze, Shi and Yifan, Wu and Bingheng, Wu},
+  year={2025},
+  month={March}, 
 }
 ```