He, Yutong

17 publications

ICLR 2026 Accelerating Diffusion Planners in Offline RL via Reward-Aware Consistency Trajectory Distillation Xintong Duan, Yutong He, Fahim Tajwar, Ruslan Salakhutdinov, J Zico Kolter, Jeff Schneider

ICLR 2026 Joint Distillation for Fast Likelihood Evaluation and Sampling in Flow-Based Models Xinyue Ai, Yutong He, Albert Gu, Ruslan Salakhutdinov, J Zico Kolter, Nicholas Matthew Boffi, Max Simchowitz

TMLR 2025 Automated Black-Box Prompt Engineering for Personalized Text-to-Image Generation Yutong He, Alexander Robey, Naoki Murata, Yiding Jiang, Joshua Nathaniel Williams, George J. Pappas, Hamed Hassani, Yuki Mitsufuji, Ruslan Salakhutdinov, J Zico Kolter

NeurIPS 2025 MISA: Memory-Efficient LLMs Optimization with Module-Wise Importance Sampling Yuxi Liu, Renjia Deng, Yutong He, Xue Wang, Tao Yao, Kun Yuan

NeurIPS 2025 MeCeFO: Enhancing LLM Training Robustness via Fault-Tolerant Optimization Rizhen Hu, Yutong He, Ran Yan, Mou Sun, Binhang Yuan, Kun Yuan

TMLR 2025 State Combinatorial Generalization in Decision Making with Conditional Diffusion Models Xintong Duan, Yutong He, Fahim Tajwar, Wentse Chen, Ruslan Salakhutdinov, Jeff Schneider

ICML 2025 Subspace Optimization for Large Language Models with Convergence Guarantees Yutong He, Pengrui Li, Yipeng Hu, Chuyan Chen, Kun Yuan

ICLR 2024 Consistency Trajectory Models: Learning Probability Flow ODE Trajectory of Diffusion Dongjun Kim, Chieh-Hsin Lai, Wei-Hsiang Liao, Naoki Murata, Yuhta Takida, Toshimitsu Uesaka, Yutong He, Yuki Mitsufuji, Stefano Ermon

ICML 2024 Distributed Bilevel Optimization with Communication Compression Yutong He, Jie Hu, Xinmeng Huang, Songtao Lu, Bin Wang, Kun Yuan

ICLR 2024 Manifold Preserving Guided Diffusion Yutong He, Naoki Murata, Chieh-Hsin Lai, Yuhta Takida, Toshimitsu Uesaka, Dongjun Kim, Wei-Hsiang Liao, Yuki Mitsufuji, J Zico Kolter, Ruslan Salakhutdinov, Stefano Ermon

ICML 2023 CSP: Self-Supervised Contrastive Spatial Pre-Training for Geospatial-Visual Representations Gengchen Mai, Ni Lao, Yutong He, Jiaming Song, Stefano Ermon

NeurIPS 2023 Unbiased Compression Saves Communication in Distributed Optimization: When and How Much? Yutong He, Xinmeng Huang, Kun Yuan

ICLR 2022 Comparing Distributions by Measuring Differences That Affect Decision Making Shengjia Zhao, Abhishek Sinha, Yutong He, Aidan Perreault, Jiaming Song, Stefano Ermon

ICLR 2022 SDEdit: Guided Image Synthesis and Editing with Stochastic Differential Equations Chenlin Meng, Yutong He, Yang Song, Jiaming Song, Jiajun Wu, Jun-Yan Zhu, Stefano Ermon

NeurIPS 2022 SatMAE: Pre-Training Transformers for Temporal and Multi-Spectral Satellite Imagery Yezhen Cong, Samar Khanna, Chenlin Meng, Patrick Liu, Erik Rozi, Yutong He, Marshall Burke, David Lobell, Stefano Ermon

NeurIPS 2021 Spatial-Temporal Super-Resolution of Satellite Imagery via Conditional Pixel Synthesis Yutong He, Dingjie Wang, Nicholas Lai, William Zhang, Chenlin Meng, Marshall Burke, David Lobell, Stefano Ermon

CVPR 2020 Fine-Grained Image-to-Image Transformation Towards Visual Recognition Wei Xiong, Yutong He, Yixuan Zhang, Wenhan Luo, Lin Ma, Jiebo Luo