avatar

Jiachen Yao

Ph.D. Student
Caltech CMS
firstname (at) caltech.edu


About Me

Hello! This is Jiachen. ▶ More about my name… Jiachen is pronounced “chia-ch’en”. Feel free to call me James!

I’m a Ph.D. student in Computing and Mathematical Sciences at Caltech, where I am fortunate to be advised by Professor Anima Anandkumar. My research lies at the intersection of generative modeling and scientific computing.

Prior to Caltech, I completed my undergraduate studies at Tsinghua University, where I had the privilege of working with Professor Jun Zhu and Professor Hang Su on physics-informed machine learning. I am also grateful for the opportunity to pursue research at Stanford University, where I worked with Professor James Landay and Professor Monica Lam on human-AI interaction.

Research Interests

My long term goal is to develop a unified framework that learns and uses efficient representations of data, domain knowledge, and available observations to model complex systems. I am particularly interested in generative approaches to achieve this objective.

I work in the intersection of generative modeling and physical sciences. Scientific (inverse) problems are often ill-posed and non-unique, which motivates probabilistic sampling approaches. The strong priors offered by generative models can significantly enhance robustness and efficiency, while physics laws and principles, if incorporated properly, can further improve the quality and stability of generated solutions. Therefore, I am developing scalable, uncertainty-aware methods for generative models + scientific inverse problems.

News

Publications

  1. Jiachen Yao, Xiyuan Zhang, Annan Yu, Boran Han, Abdul Fatir Ansari, Shuai Zhang, Michael Bohlke-Schneider, and Bernie Wang.

  2. NeurIPS
    Jiachen Yao*, Zi-Siang Hsu*, Xi Deng*, Aditi Gupta, Xin Ju, Sally M. Benson, Gege Wen, and Anima Anandkumar.
    Neural Information Processing Systems (NeurIPS), 2026.

  3. NeurIPS
    Shaoxiang Qin, Xiongye Xiao, Yucheng Zhao, Fuyuan Lyu, Di Zhou, Jiachen Yao, Xue Liu, Anima Anandkumar, and Liangzhu Leon Wang.
    Neural Information Processing Systems (NeurIPS), 2026 (Spotlight).

  4. NeurIPSW
    Lufang Chiang*, Jiachen Yao*, Thomas Y.L. Lin, and Anima Anandkumar.
    NeurIPS Workshop on AI for Stochastic Dynamics, 2026.

  5. ICLRW
    Xin Ju, Jiachen Yao, Anima Anandkumar, Sally M. Benson, and Gege Wen.
    ICLR Workshop on AI&PDE, 2026.

  6. ICLRW
    Thomas Lin*, Jiachen Yao*, Lufang Chiang, Julius Berner, Anima Anandkumar.
    ICLR Workshop on AI&PDE, 2026 (Oral).

  7. TMLR
    Duy Nguyen*, Jiachen Yao*, Jiayun Wang*, Julius Berner, Anima Anandkumar.
    Transactions on Machine Learning Research, 2026.

  8. TMLR
    Bahareh Tolooshams*, Aditi Chandrashekar*, Rayhan Zirvi*, Abbas Mammadov, Jiachen Yao, Chuwei Wang, Anima Anandkumar.
    Transactions on Machine Learning Research, 2026.

  9. NeurIPS
    Jiachen Yao*, Abbas Mammadov*, Julius Berner, Gavin Kerrigan, Jong Chul Ye, Kamyar Azizzadenesheli, Anima Anandkumar.
    Neural Information Processing Systems (NeurIPS), 2025.

  10. NeurIPS
    Zhongkai Hao*, Jiachen Yao*, Chang Su*, Hang Su, Ziao Wang, Fanzhi Lu, Zeyu Xia, Yichi Zhang, Songming Liu, Lu Lu, and Jun Zhu.
    Neural Information Processing Systems (NeurIPS), 2024.

  11. Jackie Yang*, Jiachen Yao*, Yingtian Shi, Monica Lam, and James Landay.

  12. Songming Liu, Chang Su*, Jiachen Yao*, Zhongkai Hao, Hang Su, Youjia Wu, and Jun Zhu.

  13. ICML
    Jiachen Yao*, Chang Su*, Zhongkai Hao, Songming Liu, Hang Su, and Jun Zhu.
    International Conference on Machine Learning (ICML), 2023.

  14. CHI
    Yue Qin, Chun Yu, Wentao Yao, Jiachen Yao, Chen Liang, Yueting Weng, Yukang Yan,and Yuanchun Shi.
    Conference on Human Factors in Computing Systems (CHI), 2023.

  15. * denotes equal contribution

Services

As a reviewer

Invited talks

Honors and Awards

Graduate Honors

Undergrad Honors

Personal Interests

Hobbies

Technical Skills


Powered by Jekyll and Minimal Light theme.