Zhai, Shuangfei

42 publications

AISTATS 2025 Composition and Control with Distilled Energy Diffusion Models and Sequential Monte Carlo James Thornton, Louis Béthune, Ruixiang Zhang, Arwen Bradley, Preetum Nakkiran, Shuangfei Zhai

ICLR 2025 Denoising Autoregressive Transformers for Scalable Text-to-Image Generation Jiatao Gu, Yuyang Wang, Yizhe Zhang, Qihang Zhang, Dinghuai Zhang, Navdeep Jaitly, Joshua M. Susskind, Shuangfei Zhai

NeurIPS 2025 Flexible Language Modeling in Continuous Space with Transformer-Based Autoregressive Flows Ruixiang Zhang, Shuangfei Zhai, Jiatao Gu, Yizhe Zhang, Huangjie Zheng, Tianrong Chen, Miguel Ángel Bautista, Joshua M. Susskind, Navdeep Jaitly

TMLR 2025 Improving GFlowNets for Text-to-Image Diffusion Alignment Dinghuai Zhang, Yizhe Zhang, Jiatao Gu, Ruixiang Zhang, Joshua M. Susskind, Navdeep Jaitly, Shuangfei Zhai

ICML 2025 Normalizing Flows Are Capable Generative Models Shuangfei Zhai, Ruixiang Zhang, Preetum Nakkiran, David Berthelot, Jiatao Gu, Huangjie Zheng, Tianrong Chen, Miguel Ángel Bautista, Navdeep Jaitly, Joshua M. Susskind

NeurIPS 2025 STARFlow: Scaling Latent Normalizing Flows for High-Resolution Image Synthesis Jiatao Gu, Tianrong Chen, David Berthelot, Huangjie Zheng, Yuyang Wang, Ruixiang Zhang, Laurent Dinh, Miguel Ángel Bautista, Joshua M. Susskind, Shuangfei Zhai

NeurIPS 2025 TADA: Improved Diffusion Sampling with Training-Free Augmented DynAmics Tianrong Chen, Huangjie Zheng, David Berthelot, Jiatao Gu, Joshua M. Susskind, Shuangfei Zhai

ICML 2025 Target Concrete Score Matching: A Holistic Framework for Discrete Diffusion Ruixiang Zhang, Shuangfei Zhai, Yizhe Zhang, James Thornton, Zijing Ou, Joshua M. Susskind, Navdeep Jaitly

CVPR 2025 World-Consistent Video Diffusion with Explicit 3D Modeling Qihang Zhang, Shuangfei Zhai, Miguel Ángel Bautista Martin, Kevin Miao, Alexander Toshev, Joshua Susskind, Jiatao Gu

ICML 2024 Data-Free Distillation of Diffusion Models with Bootstrapping Jiatao Gu, Chen Wang, Shuangfei Zhai, Yizhe Zhang, Lingjie Liu, Joshua M. Susskind

ICLR 2024 Generative Modeling with Phase Stochastic Bridge Tianrong Chen, Jiatao Gu, Laurent Dinh, Evangelos Theodorou, Joshua M. Susskind, Shuangfei Zhai

ICLRW 2024 How Far Are We from Intelligent Visual Deductive Reasoning? Yizhe Zhang, He Bai, Ruixiang Zhang, Jiatao Gu, Shuangfei Zhai, Joshua M. Susskind, Navdeep Jaitly

ICMLW 2024 Improving GFlowNets for Text-to-Image Diffusion Alignment Dinghuai Zhang, Yizhe Zhang, Jiatao Gu, Ruixiang Zhang, Joshua M. Susskind, Navdeep Jaitly, Shuangfei Zhai

ICMLW 2024 Improving GFlowNets for Text-to-Image Diffusion Alignment Dinghuai Zhang, Yizhe Zhang, Jiatao Gu, Ruixiang Zhang, Joshua M. Susskind, Navdeep Jaitly, Shuangfei Zhai

NeurIPS 2024 Kaleido Diffusion: Improving Conditional Diffusion Models with Autoregressive Latent Modeling Jiatao Gu, Ying Shen, Shuangfei Zhai, Yizhe Zhang, Navdeep Jaitly, Josh Susskind

ICMLW 2024 Kaleido Diffusion: Improving Conditional Diffusion Models with Autoregressive Latent Modeling Jiatao Gu, Ying Shen, Shuangfei Zhai, Yizhe Zhang, Navdeep Jaitly, Joshua M. Susskind

ICMLW 2024 Many-to-Many Image Generation with Auto-Regressive Diffusion Models Ying Shen, Yizhe Zhang, Shuangfei Zhai, Lifu Huang, Joshua M. Susskind, Jiatao Gu

ICLR 2024 Matryoshka Diffusion Models Jiatao Gu, Shuangfei Zhai, Yizhe Zhang, Joshua M. Susskind, Navdeep Jaitly

ICML 2024 Scalable Pre-Training of Large Autoregressive Image Models Alaaeldin El-Nouby, Michal Klein, Shuangfei Zhai, Miguel Ángel Bautista, Vaishaal Shankar, Alexander T Toshev, Joshua M. Susskind, Armand Joulin

TMLR 2024 The Slingshot Effect: A Late-Stage Optimization Anomaly in Adaptive Gradient Methods Vimal Thilak, Etai Littwin, Shuangfei Zhai, Omid Saremi, Roni Paiss, Joshua M. Susskind

CVPR 2023 AutoFocusFormer: Image Segmentation Off the Grid Chen Ziwen, Kaushik Patnaik, Shuangfei Zhai, Alvin Wan, Zhile Ren, Alexander G. Schwing, Alex Colburn, Li Fuxin

ICMLW 2023 BOOT: Data-Free Distillation of Denoising Diffusion Models with Bootstrapping Jiatao Gu, Shuangfei Zhai, Yizhe Zhang, Lingjie Liu, Joshua M. Susskind

ICLR 2023 F-DM: A Multi-Stage Diffusion Model via Progressive Signal Transformation Jiatao Gu, Shuangfei Zhai, Yizhe Zhang, Miguel Ángel Bautista, Joshua M. Susskind

NeurIPS 2023 PLANNER: Generating Diversified Paragraph via Latent Language Diffusion Model Yizhe Zhang, Jiatao Gu, Zhuofeng Wu, Shuangfei Zhai, Joshua Susskind, Navdeep Jaitly

ICML 2023 Stabilizing Transformer Training by Preventing Attention Entropy Collapse Shuangfei Zhai, Tatiana Likhomanenko, Etai Littwin, Dan Busbridge, Jason Ramapuram, Yizhe Zhang, Jiatao Gu, Joshua M. Susskind

NeurIPS 2022 GAUDI: A Neural Architect for Immersive 3D Scene Generation Miguel Angel Bautista, Pengsheng Guo, Samira Abnar, Walter Talbott, Alexander Toshev, Zhuoyuan Chen, Laurent Dinh, Shuangfei Zhai, Hanlin Goh, Daniel Ulbricht, Afshin Dehghan, Joshua Susskind

ICLR 2022 Learning Representation from Neural Fisher Kernel with Low-Rank Approximation Ruixiang Zhang, Shuangfei Zhai, Etai Littwin, Joshua M. Susskind

ICML 2022 Position Prediction as an Effective Pretraining Strategy Shuangfei Zhai, Navdeep Jaitly, Jason Ramapuram, Dan Busbridge, Tatiana Likhomanenko, Joseph Y Cheng, Walter Talbott, Chen Huang, Hanlin Goh, Joshua M Susskind

NeurIPSW 2022 The Slingshot Mechanism: An Empirical Study of Adaptive Optimizers and the \emph{Grokking Phenomenon} Vimal Thilak, Etai Littwin, Shuangfei Zhai, Omid Saremi, Roni Paiss, Joshua M. Susskind

CVPR 2021 MetricOpt: Learning to Optimize Black-Box Evaluation Metrics Chen Huang, Shuangfei Zhai, Pengsheng Guo, Josh Susskind

WACV 2021 On the Generalization of Learning-Based 3D Reconstruction Miguel Angel Bautista, Walter Talbott, Shuangfei Zhai, Nitish Srivastava, Joshua M. Susskind

NeurIPSW 2021 Robust Robotic Control from Pixels Using Contrastive Recurrent State-Space Models Nitish Srivastava, Walter Talbott, Martin Bertran Lopez, Shuangfei Zhai, Joshua M. Susskind

ICML 2021 Uncertainty Weighted Actor-Critic for Offline Reinforcement Learning Yue Wu, Shuangfei Zhai, Nitish Srivastava, Joshua M Susskind, Jian Zhang, Ruslan Salakhutdinov, Hanlin Goh

NeurIPS 2020 Collegial Ensembles Etai Littwin, Ben Myara, Sima Sabah, Joshua Susskind, Shuangfei Zhai, Oren Golan

ICML 2019 Addressing the Loss-Metric Mismatch with Adaptive Loss Alignment Chen Huang, Shuangfei Zhai, Walter Talbott, Miguel Bautista Martin, Shih-Yu Sun, Carlos Guestrin, Josh Susskind

NeurIPS 2019 Adversarial Fisher Vectors for Unsupervised Representation Learning Shuangfei Zhai, Walter Talbott, Carlos Guestrin, Joshua Susskind

CVPR 2017 Fully-Adaptive Feature Sharing in Multi-Task Networks with Applications in Person Attribute Classification Yongxi Lu, Abhishek Kumar, Shuangfei Zhai, Yu Cheng, Tara Javidi, Rogerio Feris

CVPR 2017 S3Pool: Pooling with Stochastic Spatial Sampling Shuangfei Zhai, Hui Wu, Abhishek Kumar, Yu Cheng, Yongxi Lu, Zhongfei Zhang, Rogerio Feris

AAAI 2017 Structural Correspondence Learning for Cross-Lingual Sentiment Classification with One-to-Many Mappings Nana Li, Shuangfei Zhai, Zhongfei Zhang, Boying Liu

ICML 2016 Deep Structured Energy Based Models for Anomaly Detection Shuangfei Zhai, Yu Cheng, Weining Lu, Zhongfei Zhang

NeurIPS 2016 Doubly Convolutional Neural Networks Shuangfei Zhai, Yu Cheng, Zhongfei Zhang, Weining Lu

AAAI 2016 Semisupervised Autoencoder for Sentiment Analysis Shuangfei Zhai, Zhongfei (Mark) Zhang