Update README.md

README.md

@@ -10,6 +10,21 @@ library_name: pytorch
 
 # Model Card
 
+## Citation
+
+If you find the models useful, please cite our paper.
+
+```BibTeX
+@article{zhang2025overcoming,
+  title={Overcoming Knowledge Barriers: Online Imitation Learning from Visual Observation with Pretrained World Models},
+  author={Xingyuan Zhang and Philip Becker-Ehmck and Patrick van der Smagt and Maximilian Karl},
+  journal={Transactions on Machine Learning Research},
+  issn={2835-8856},
+  year={2025},
+  url={https://openreview.net/forum?id=BaRD2Nfj41},
+}
+```
+
 ## Model Details
 
 ### Model Description
@@ -100,23 +115,6 @@ For example, to get the `walker-plan2explore-buffer` model, you can run `python
 
 Besides the details we discuss above, please checkout the general config at [here](https://github.com/IcarusWizard/AIME-NoB/blob/main/aime_nob/configs/model-only.yaml) and model configs for [RSSMO](https://github.com/IcarusWizard/AIME-NoB/blob/main/aime_nob/configs/world_model/rssmo.yaml).
 
-## Citation
-
-<!-- If there is a paper or blog post introducing the model, Bibtex information for that should go in this section. -->
-
-If you find the models useful, please cite our paper.
-
-```BibTeX
-@misc{zhang2024overcoming,
-    title={Overcoming Knowledge Barriers: Online Imitation Learning from Visual Observation with Pretrained World Models}, 
-    author={Xingyuan Zhang and Philip Becker-Ehmck and Patrick van der Smagt and Maximilian Karl},
-    year={2024},
-    eprint={2404.18896},
-    archivePrefix={arXiv},
-    primaryClass={cs.LG}
-}
-```
-
 ## Model Card Authors and Contact
 
 Xingyuan Zhang with wizardicarus@gmail.com.