publications

* Equal Contribution

2026

1. Preprint

   Does the Proof Prove It That Way? Faithful Formalization of Elements Proofs

   Tadd Mao, Tianjun Zhong, Dhruva Arekar, and 5 more authors

   Aug 2026

   arXiv Bib

   @misc{mao2026proofproveitthatway,
     title = {Does the Proof Prove It That Way? Faithful Formalization of Elements Proofs},
     author = {Mao, Tadd and Zhong, Tianjun and Arekar, Dhruva and Feng, Yuming and An, One and Huang, Jiani and Si, Xujie and Li, Ziyang},
     year = {2026},
     month = aug,
     url = {https://arxiv.org/abs/2608.15432},
   }

2. Preprint

   SWE-Spot: Building Small Repo-Experts with Repository-Centric Learning

   Jinjun Peng^*, Magnus Saebo^*, Tianjun Zhong, and 5 more authors

   Jan 2026

   arXiv Bib

   @misc{peng2026swespotbuildingsmallrepoexperts,
     title = {SWE-Spot: Building Small Repo-Experts with Repository-Centric Learning},
     author = {Peng, Jinjun and Saebo, Magnus and Zhong, Tianjun and Cheng, Yi-Jie and Yang, Junfeng and Ray, Baishakhi and Chen, Simin and Ding, Yangruibo},
     year = {2026},
     month = jan,
     url = {https://arxiv.org/abs/2601.21649},
   }

3. Preprint

   From Chains to DAGs: Probing the Graph Structure of Reasoning in LLMs

   Tianjun Zhong, Linyang He, Ziyang Li, and 1 more author

   Jan 2026

   arXiv Bib

   @misc{zhong2026chainsdagsprobinggraph,
     title = {From Chains to DAGs: Probing the Graph Structure of Reasoning in LLMs},
     author = {Zhong, Tianjun and He, Linyang and Li, Ziyang and Mesgarani, Nima},
     year = {2026},
     month = jan,
     url = {https://arxiv.org/abs/2601.17593},
   }

2025

1. NeurIPS 2025

   Far from the Shallow: Brain-Predictive Reasoning Embedding through Residual Disentanglement

   Linyang He^*, Tianjun Zhong^*, Richard Antonello, and 3 more authors

   In The Thirty-ninth Annual Conference on Neural Information Processing Systems, Oct 2025

   arXiv OpenReview Code Bib

   @inproceedings{he2025brainpredictive,
     title = {Far from the Shallow: Brain-Predictive Reasoning Embedding through Residual Disentanglement},
     author = {He, Linyang and Zhong, Tianjun and Antonello, Richard and Mischler, Gavin and Goldblum, Micah and Mesgarani, Nima},
     booktitle = {The Thirty-ninth Annual Conference on Neural Information Processing Systems},
     year = {2025},
     month = oct,
     openreview = {https://openreview.net/forum?id=tPqBnGwTwa},
     url = {https://arxiv.org/abs/2510.22860},
   }

2. Preprint

   K2-Think: A Parameter-Efficient Reasoning System

   Zhoujun Cheng^*, Richard Fan^*, Shibo Hao^*, and 28 more authors

   Sep 2025

   arXiv Hugging Face Code Bib

   @misc{cheng2025k2thinkparameterefficientreasoning,
     title = {K2-Think: A Parameter-Efficient Reasoning System},
     author = {Cheng, Zhoujun and Fan, Richard and Hao, Shibo and Killian, Taylor W. and Li, Haonan and Sun, Suqi and Ren, Hector and Moreno, Alexander and Zhang, Daqian and Zhong, Tianjun and Xiong, Yuxin and Hu, Yuanzhe and Xie, Yutao and Han, Xudong and Wang, Yuqi and Pimpalkhute, Varad and Zhuang, Yonghao and Singh, Aaryamonvikram and Liang, Xuezhi and Xie, Anze and She, Jianshu and Fan, Desai and Gao, Chengqian and Ma, Liqun and Yurochkin, Mikhail and Maggs, John and Ma, Xuezhe and He, Guowei and Hu, Zhiting and Liu, Zhengzhong and Xing, Eric P.},
     year = {2025},
     month = sep,
     huggingface = {https://huggingface.co/LLM360/K2-Think},
     url = {https://arxiv.org/abs/2509.07604},
   }