Zitao Song (宋子韬)
Research Associate
School of Computer Science and Engineering,
Nanyang Technological University,
Singapore.
Email: song903 (at) purdue (dot) edu
![Portrait](pic/personal.jpg)
Short Bio
I am an incoming Ph.D. student at Department of Computer Science - Purdue University and I am fortunate to be awarded by Lynn Fellowship. I am currently a Research Associate at the Agent Mediated Intelligence (AMI) Research Group under the supervision of Prof. Bo An at Nanyang Technological University (NTU). I obtained my master's degree in Data Science at The Chinese University of Hong Kong, Shenzhen (CUHK-Shenzhen) under the supervision of Prof. Shuang Li. I obtained my bachelor's degree in Applied Mathematics at Xi'an Jiaotong-Liverpool University (XJTLU) under the supervision of Prof. Jia Meng and Prof. Jionglong Su.
Research
I have broad interests in Sequential Decision Making, Machine Learning, and LLMs. Especially, I focus on the following directions:
- Large-scale optimization for scientific discovery, i.e., biological sequence optimization, material discovery and perference learning.
- RL as Probabilistic Inference, scale and accelerate probabilistic sampling algorithms, i.e., SMC, MCMC, by leveraging world/prior knowledge in LLM/VLMs and diffusion models.
Selected Publications
(* indicates equal contribution, see full list in Google Scholar)
2021 - Present
![Publication 1](thumbnail/Latee.png)
Zitao Song, Chao Yang, Chaojie Wang, Bo An, Shuang Li.
The Forty-first International Conference on Machine Learning [ICML 2024]
![Publication 2](thumbnail/ANI.png)
Zitao Song, Wendi Ren, Shuang Li.
The Twelfth International Conference on Learning Representations [ICLR 2024]
![Publication 3](thumbnail/MultiRM.png)
Zitao Song, Daiyun Huang, Bowen Song, Kunqi Chen, Yiyou Song, Gang Liu, Jionglong Su, João Pedro de Magalhães, Daniel J Rigden, Jia Meng.
Nature Communications [Nat. Commun 12(4011), 2021]