Discrete Flows: Invertible Generative Models of Discrete Data. - Dustin Tran; Keyon Vafa; Kumar Krishna Agrawal; Laurent Dinh; Ben Poole (NeurIPS, 2019)

Text-Based Ideal Points. - Keyon Vafa; Suresh Naidu; David M. Blei (ACL, 2020)

Rationales for Sequential Predictions. - Keyon Vafa; Yuntian Deng; David M. Blei; Alexander M. Rush (EMNLP (1), 2021)

Assessing the Effects of Friend-to-Friend Texting onTurnout in the 2018 US Midterm Elections. - Aaron Schein; Keyon Vafa; Dhanya Sridhar; Victor Veitch; Jeffrey Quinn; James Moffet; David M. Blei; Donald P. Green (WWW, 2021)

Learning Transferrable Representations of Career Trajectories for Economic Prediction. - Keyon Vafa; Emil Palikot; Tianyu Du; Ayush Kanodia; Susan Athey; David M. Blei (CoRR, 2022)

An Invariant Learning Characterization of Controlled Text Generation. - Carolina Zheng; Claudia Shi; Keyon Vafa; Amir Feder; David M. Blei (ACL (1), 2023)

Use large language models to promote equity. - Emma Pierson; Divya Shanmugam; Rajiv Movva; Jon M. Kleinberg; Monica Agrawal; Mark Dredze; Kadija Ferryman; Judy Wawira Gichoya; Dan Jurafsky; Pang Wei Koh; Karen Levy; Sendhil Mullainathan; et al. (CoRR, 2023)

Revisiting Topic-Guided Language Models. - Carolina Zheng; Keyon Vafa; David M. Blei (Trans. Mach. Learn. Res., 2023)

Do Large Language Models Perform the Way People Expect? Measuring the Human Generalization Function. - Keyon Vafa; Ashesh Rambachan; Sendhil Mullainathan (ICML, 2024)

Evaluating the World Model Implicit in a Generative Model. - Keyon Vafa; Justin Y. Chen; Ashesh Rambachan; Jon M. Kleinberg; Sendhil Mullainathan (NeurIPS, 2024)

LABOR-LLM: Language-Based Occupational Representations with Large Language Models. - Tianyu Du; Ayush Kanodia; Herman Brunborg; Keyon Vafa; Susan Athey (CoRR, 2024)

Estimating Wage Disparities Using Foundation Models. - Keyon Vafa; Susan Athey; David M. Blei (CoRR, 2024)

CAREER: A Foundation Model for Labor Sequence Data. - Keyon Vafa; Emil Palikot; Tianyu Du; Ayush Kanodia; Susan Athey; David M. Blei (Trans. Mach. Learn. Res., 2024)

What's Producible May Not Be Reachable: Measuring the Steerability of Generative Models. - Keyon Vafa; Sarah Bentley; Jon M. Kleinberg; Sendhil Mullainathan (CoRR, 2025)

Critical Thinking: Which Kinds of Complexity Govern Optimal Reasoning Length? - Celine Lee; Alexander M. Rush; Keyon Vafa (CoRR, 2025)

Potemkin Understanding in Large Language Models. - Marina Mancoridis; Bec Weeks; Keyon Vafa; Sendhil Mullainathan (CoRR, 2025)

What Has a Foundation Model Found? Using Inductive Bias to Probe for World Models. - Keyon Vafa; Peter G. Chang; Ashesh Rambachan; Sendhil Mullainathan (CoRR, 2025)