Update README.md

README.md

@@ -36,4 +36,37 @@ This repository hosts the official models and datasets released as part of the H
 | [Human Behavior Atlas (Parquet)](https://huggingface.co/datasets/HumanBehaviorAtlas/human_behavior_atlas) | Primary HBA release optimized for Hugging Face `datasets` workflows. |
 | [Human Behavior Atlas (TAR)](https://huggingface.co/datasets/HumanBehaviorAtlas/human_behavior_atlas_tar) | TAR-based release for custom preprocessing and large-scale storage workflows. |
 
-The version distributed in **Parquet format**, is generally easier and faster to use with the Hugging Face `datasets` library and most downstream training pipelines. Although, we also provide the tar version to allow for flexible use. 
+The version distributed in **Parquet format**, is generally easier and faster to use with the Hugging Face `datasets` library and most downstream training pipelines. Although, we also provide the tar version to allow for flexible use. 
+
+
+---
+
+# Citation
+
+If you use Human Behavior Atlas or OmniSapiens in your research, please cite the corresponding work below.
+
+## Human Behavior Atlas
+
+```bibtex
+@inproceedings{
+ong2026human,
+title={Human Behavior Atlas: Benchmarking Unified Psychological And Social Behavior Understanding},
+author={Keane Ong and Wei Dai and Carol Li and Dewei Feng and Hengzhi Li and Jingyao Wu and Jiaee Cheong and Rui Mao and Gianmarco Mengaldo and Erik Cambria and Paul Pu Liang},
+booktitle={The Fourteenth International Conference on Learning Representations},
+year={2026},
+url={https://openreview.net/forum?id=ZKE23BBvlQ}
+}
+```
+
+## OmniSapiens 2.0
+
+```bibtex
+@article{ong2026omnisapiens,
+  title={Omnisapiens: A foundation model for social behavior processing via heterogeneity-aware relative policy optimization},
+  author={Ong, Keane and Boughorbel, Sabri and Xiao, Luwei and Ekbote, Chanakya and Dai, Wei and Qu, Ao and Wu, Jingyao and Mao, Rui and Hoque, Ehsan and Cambria, Erik and others},
+  journal={arXiv preprint arXiv:2602.10635},
+  year={2026}
+}
+```
+
+