Current Research

Data & Evaluation. Create new data engines. Create benchmarks that are formally verified. Unify scientific inverse problems into common frameworks.

Models & Architecture. Create new effective representations, such as for time series.

Inference & Search. Develop Bayesian inference methods using diffusion priors. Develop advanced LLM reasoning methods such as dynamic tree decomposition and self-training. Develop abstractions to simplify agent programming.

Selected Talks & Tutorials

Foundation Models in the Scientific Loop: Design, Measure, Intepret

2026

AI for Adaptive Experiment Design

2025

AI for Robust Decision Making

2025

Neurosymbolic AI for Safety-Critical Agile Control

2024

Controlling the Structure of Inference and Learning in Neural Networks

2024

Neurosymbolic Programming Summer School

2022

Co-taught Architecture Search Module with Swarat Chaudhuri and Jennifer Sun.

Personalized Preference Learning from Spinal Cord Stimulation to Exoskeletons

2021

Neurosymbolic Programming

2021

Imitation Learning Tutorial

2018

Co-taught with Hoang Le at ICML 2018

Selected Papers

Simple Agents Outperform Experts in Biomedical Imaging Workflow Optimization

2026

Xuefei Wang, Kai A. Horstmann, Ethan Lin, Jonathan Chen, Alexander R. Farhang, Sophia Stiles, Atharva Sehgal, Jonathan Light, David Van Valen, Yisong Yue, Jennifer J. Sun

arXiv preprint arXiv:2512.06006, 2026

Feedforward 3D Editing via Text-Steerable Image-to-3D

2025

Ziqi Ma, Hongqiao Chen, Yisong Yue, Georgia Gkioxari

arXiv preprint arXiv:2512.13678, 2025

CellSAM: a foundation model for cell segmentation

2025

Markus Marks, Uriah Israel, Rohit Dilip, Qilin Li, Changhua Yu, Emily Laubscher, Ahamed Iqbal, Elora Pradhan, Ada Ates, Martin Abt, Caitlin Brown, Edward Pao, Shenyi Li, Alexander Pearson-Goulart, Pietro Perona, Georgia Gkioxari, Ross Barnowski, Yisong Yue, David Van Valen

Nature Methods, 2025

EnCompass: Enhancing Agent Programming with Search Over Program Execution Paths

2025

Zhening Li, Armando Solar-Lezama, Yisong Yue, Stephan Zheng

Neural Information Processing Systems (NeurIPS), 2025

KALIKO: Kalman-Implicit Koopman Operator Learning For Prediction of Nonlinear Dynamical Systems

2025

Albert H Li, Ivan Dario Jimenez Rodriguez, Joel W Burdick, Yisong Yue, Aaron D Ames

arXiv preprint arXiv:2512.03256, 2025

Blade: A Derivative-free Bayesian Inversion Method using Diffusion Priors

2025

Hongkai Zheng, Austin Wang, Zihui Wu, Zhengyu Huang, Ricardo Baptista, Yisong Yue

arXiv preprint arXiv:2510.10968, 2025

STeP: A General and Scalable Framework for Solving Video Inverse Problems with Spatiotemporal Diffusion Priors

2025

Bingliang Zhang, Zihui Wu, Berthy T. Feng, Yang Song, Yisong Yue, Katherine L. Bouman

arXiv preprint arXiv:2504.07549, 2025

A Narwhal-Inspired Sensing-to-Control Framework for Small Fixed-Wing Aircraft

2025

Fengze Xie, Xiaozhou Fan, Jacob Schuster, Yisong Yue, Morteza Gharib

arXiv preprint arXiv:2510.07160, 2025

Conformal Risk Training: End-to-End Optimization of Conformal Risk Control

2025

Christopher Yeh, Nicolas Christianson, Adam Wierman, Yisong Yue

Neural Information Processing Systems (NeurIPS), 2025

Kuramoto Orientation Diffusion Models

2025

Yue Song, T Anderson Keller, Sevan Brodjian, Takeru Miyato, Yisong Yue, Pietro Perona, Max Welling

Neural Information Processing Systems (NeurIPS), 2025

DISC: Dynamic Decomposition Improves LLM Inference Scaling

2025

Jonathan Light, Wei Cheng, Benjamin Riviere, Wu Yue, Masafumi Oyamada, Mengdi Wang, Yisong Yue, Santiago Paternain, Haifeng Chen

Neural Information Processing Systems (NeurIPS), 2025

Steering Generative Models with Experimental Data for Protein Fitness Optimization

2025

Jason Yang, Wenda Chu, Daniel Khalil, Raul Astudillo, Bruce J. Wittmann, Frances H. Arnold, Yisong Yue

Neural Information Processing Systems (NeurIPS), 2025

CLEVER: A Curated Benchmark for Formally Verified Code Generation

2025

Amitayush Thakur, Jasper Lee, George Tsoukalas, Meghana Sistla, Matthew Zhao, Stefan Zetzche, Greg Durrett, Yisong Yue, Swarat Chaudhuri

Neural Information Processing Systems (NeurIPS), 2025

Find Any Part in 3D

2025

Ziqi Ma, Yisong Yue, Georgia Gkioxari

International Conference on Computer Vision (ICCV), 2025Spotlight

Morphological-Symmetry-Equivariant Heterogeneous Graph Neural Network for Robotic Dynamics Learning

2025

Fengze Xie, Sizhe Wei, Yue Song, Yisong Yue, Lu Gan

Conference on Learning for Dynamics and Control (L4DC), 2025

Population Transformer: Learning Population-level Representations of Neural Activity

2025

Geeling Chau, Christopher Wang, Sabera Talukder, Vighnesh Subramaniam, Saraswati Soedarmadji, Yisong Yue, Boris Katz, Andrei Barbu

International Conference on Learning Representations (ICLR), 2025Oral

InverseBench: Benchmarking Plug-and-Play Diffusion Models for Inverse Problems in Physical Sciences

2025

Hongkai Zheng, Wenda Chu, Bingliang Zhang, Zihui Wu, Austin Wang, Berthy T. Feng, Caifeng Zou, Yu Sun, Nikola Kovachki, Zachary E. Ross, Katherine L. Bouman, Yisong Yue

International Conference on Learning Representations (ICLR), 2025Spotlight

Active Learning-Assisted Directed Evolution

2025

Jason Yang, Ravi G. Lal, James C. Bowden, Raul Astudillo, Mikhail A. Hameedi, Sukhvinder Kaur, Matthew Hill, Yisong Yue, Frances H. Arnold

Nature Communications, 2025

Principled Probabilistic Imaging using Diffusion Models as Plug-and-Play Priors

2024

Zihui Wu, Yu Sun, Yifan Chen, Bingliang Zhang, Yisong Yue, Katherine L. Bouman

Neural Information Processing Systems (NeurIPS), 2024

Practical Bayesian Algorithm Execution via Posterior Sampling

2024

Chu Xin Cheng, Raul Astudillo, Thomas Desautels, Yisong Yue

Neural Information Processing Systems (NeurIPS), 2024

ReST-MCTS*: LLM Self-Training via Process Reward Guided Tree Search

2024

Dan Zhang, Sining Zhoubian, Ziniu Hu, Yisong Yue, Yuxiao Dong, Jie Tang

Neural Information Processing Systems (NeurIPS), 2024

TOTEM: TOkenized Time Series EMbeddings for General Time Series Analysis

2024

Sabera Talukder, Yisong Yue, Georgia Gkioxari

Transactions of Machine Learning Research (TMLR), 2024

SceneCraft: An LLM Agent for Synthesizing 3D Scene as Blender Code

2024

Ziniu Hu, Ahmet Iscen, Aashi Jain, Thomas Kipf, Yisong Yue, David A. Ross, Cordelia Schmid, Alireza Fathi

International Conference on Machine Learning (ICML), 2024Oral

Neural-Fly Enables Rapid Learning for Agile Flight in Strong Winds

2022

Michael O'Connell, Guanya Shi, Xichen Shi, Kamyar Azizzadenesheli, Anima Anandkumar, Yisong Yue, Soon-Jo Chung

Science Robotics, 2022

MLNav: Learning to Safely Navigate on Martian Terrains

2022

Shreyansh Daftry, Neil Abcouwer, Tyler Del Sesto, Siddarth Venkatraman, Jialin Song, Lucas Igel, Amos Byon, Ugo Rosolia, Yisong Yue, Masahiro Ono

IEEE Robotics and Automation Letters (RA-L), 2022

Neurosymbolic Programming

2021

Swarat Chaudhuri, Kevin Ellis, Oleksandr Polozov, Rishabh Singh, Armando Solar-Lezama, Yisong Yue

Foundations and Trends in Programming Languages, 2021

Empirical Study of Off-Policy Policy Evaluation for Reinforcement Learning

2021

Cameron Voloshin, Hoang M. Le, Nan Jiang, Yisong Yue

Neural Information Processing Systems (NeurIPS), 2021

Task Programming: Learning Data Efficient Behavior Representations

2021

Jennifer J. Sun, Ann Kennedy, Eric Zhan, David J. Anderson, Yisong Yue, Pietro Perona

IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2021Best Student Paper

Preference-Based Learning for Exoskeleton Gait Optimization

2020

Maegan Tucker, Ellen Novoseller, Claudia Kann, Yanan Sui, Yisong Yue, Joel Burdick, Aaron D. Ames

International Conference on Robotics and Automation (ICRA), 2020Best Paper

Batch Policy Learning under Constraints

2019

Hoang M. Le, Cameron Voloshin, Yisong Yue

International Conference on Machine Learning (ICML), 2019Oral

Hierarchical Imitation and Reinforcement Learning

2018

Hoang M. Le, Nan Jiang, Alekh Agarwal, Miroslav Dudík, Yisong Yue, Hal Daumé III

International Conference on Machine Learning (ICML), 2018

Multi-dueling Bandits with Dependent Arms

2017

Yanan Sui, Vincent Zhuang, Joel Burdick, Yisong Yue

Conference on Uncertainty in Artificial Intelligence (UAI), 2017

The K-armed Dueling Bandits Problem

2009

Yisong Yue, Josef Broder, Robert Kleinberg, Thorsten Joachims

Conference on Learning Theory (COLT), 2009