About

I'm a fully-funded master's student at Princeton Computer Science, advised by Prof. Sanjeev Arora. My research interests lie in large language models and understanding learning algorithms. I graduated from Princeton class of 2024, where I had the fortune to work with Prof. Danqi Chen and Prof. Kai Li. During college I co-founded Princeton's biggest AI student organization with awesome people.

I can be reached at jiatongy [at] princeton.edu.

Publications

  1. On the Impossibility of Retrain Equivalence in Machine Unlearning — Jiatong Yu, Yinghui He, Anirudh Goyal, Sanjeev Arora. Under Review
    [paper] [website]
  2. AI-Assisted Generation of Difficult Math Questions — Vedant Shah, Dingli Yu, Kaifeng Lyu, Simon Park, Jiatong Yu, Yinghui He, Nan Rosemary Ke, Michael Mozer, Yoshua Bengio, Sanjeev Arora, Anirudh Goyal. Math-AI Workshop at NeurIPS 2024
    [paper]
  3. Evaluating large language models at evaluating instruction following — Zhiyuan Zeng, Jiatong Yu, Tianyu Gao, Yu Meng, Tanya Goyal, Danqi Chen. ICLR 2024
    [paper]
  4. Enabling large language models to generate text with citations — Tianyu Gao, Howard Yen, Jiatong Yu, Danqi Chen. EMNLP 2023
    [paper]

Teaching

Algorithms and Data Structures
Fall 2025, Fall2024 [Syllabus]
Introduction to Machine Learning
Spring 2025 [Syllabus]
Natural Language Processing
Spring 2023 [Syllabus]