From Word Embeddings To Document Distances. - Matt J. Kusner; Yu Sun 020; Nicholas I. Kolkin; Kilian Q. Weinberger (ICML, 2015)

Private Causal Inference. - Matt J. Kusner; Yu Sun 020; Karthik Sridharan; Kilian Q. Weinberger (AISTATS, 2016)

Deep Networks with Stochastic Depth. - Gao Huang 001; Yu Sun 020; Zhuang Liu 003; Daniel Sedra; Kilian Q. Weinberger (ECCV (4), 2016)

Supervised Word Mover's Distance. - Gao Huang 001; Chuan Guo 001; Matt J. Kusner; Yu Sun 020; Fei Sha; Kilian Q. Weinberger (NIPS, 2016)

On Calibration of Modern Neural Networks. - Chuan Guo 001; Geoff Pleiss; Yu Sun 020; Kilian Q. Weinberger (ICML, 2017)

An empirical study on evaluation metrics of generative adversarial networks. - Qiantong Xu; Gao Huang 001; Yang Yuan; Chuan Guo 001; Yu Sun 020; Felix Wu; Kilian Q. Weinberger (CoRR, 2018)

Adversarial Deep Averaging Networks for Cross-Lingual Sentiment Classification. - Xilun Chen 002; Yu Sun 020; Ben Athiwaratkun; Claire Cardie; Kilian Q. Weinberger (Trans. Assoc. Comput. Linguistics, 2018)

Unsupervised Domain Adaptation through Self-Supervision. - Yu Sun 020; Eric Tzeng; Trevor Darrell; Alexei A. Efros (CoRR, 2019)

Test-Time Training for Out-of-Distribution Generalization. - Yu Sun 020; Xiaolong Wang 004; Zhuang Liu 003; John Miller 001; Alexei A. Efros; Moritz Hardt (CoRR, 2019)

Test-Time Training with Self-Supervision for Generalization under Distribution Shifts. - Yu Sun 020; Xiaolong Wang 004; Zhuang Liu 003; John Miller 001; Alexei A. Efros; Moritz Hardt (ICML, 2020)

Self-Supervised Policy Adaptation during Deployment. - Nicklas Hansen 001; Yu Sun 020; Pieter Abbeel; Alexei A. Efros; Lerrel Pinto; Xiaolong Wang 004 (CoRR, 2020)

Self-Supervised Policy Adaptation during Deployment. - Nicklas Hansen 001; Rishabh Jangir; Yu Sun 020; Guillem Alenyà; Pieter Abbeel; Alexei A. Efros; Lerrel Pinto; Xiaolong Wang 004 (ICLR, 2021)

Test-Time Training with Masked Autoencoders. - Yossi Gandelsman; Yu Sun 020; Xinlei Chen; Alexei A. Efros (NeurIPS, 2022)

Test-Time Training on Video Streams. - Renhao Wang; Yu Sun 020; Yossi Gandelsman; Xinlei Chen; Alexei A. Efros; Xiaolong Wang 004 (CoRR, 2023)

Learning to (Learn at Test Time). - Yu Sun 020; Xinhao Li; Karan Dalal; Chloe Hsu; Sanmi Koyejo; Carlos Guestrin; Xiaolong Wang 004; Tatsunori Hashimoto; Xinlei Chen (CoRR, 2023)

Test-Time Training on Nearest Neighbors for Large Language Models. - Moritz Hardt; Yu Sun 020 (ICLR, 2024)

Structure-aware World Model for Probe Guidance via Large-scale Self-supervised Pre-train. - Haojun Jiang; Meng Li; Zhenguo Sun; Ning Jia; Yu Sun 020; Shaqi Luo; Shiji Song; Gao Huang 001 (ASMUS@MICCAI, 2024)

Cardiac Copilot: Automatic Probe Guidance for Echocardiography with World Model. - Haojun Jiang; Zhenguo Sun; Ning Jia; Meng Li; Yu Sun 020; Shaqi Luo; Shiji Song; Gao Huang 001 (MICCAI (1), 2024)

Learning to (Learn at Test Time): RNNs with Expressive Hidden States. - Yu Sun 020; Xinhao Li; Karan Dalal; Jiarui Xu; Arjun Vikram; Genghan Zhang; Yann Dubois; Xinlei Chen; Xiaolong Wang 004; Sanmi Koyejo; Tatsunori Hashimoto; Carlos Guestrin (CoRR, 2024)

Sequence-aware Pre-training for Echocardiography Probe Guidance. - Haojun Jiang; Zhenguo Sun; Yu Sun 020; Ning Jia; Meng Li; Shaqi Luo; Shiji Song; Gao Huang 001 (CoRR, 2024)

One-Minute Video Generation with Test-Time Training. - Karan Dalal; Daniel Koceja; Jiarui Xu; Yue Zhao 006; Shihao Han; Ka Chun Cheung; Jan Kautz; Yejin Choi 001; Yu Sun 020; Xiaolong Wang 004 (CVPR, 2025)

One-Minute Video Generation with Test-Time Training. - Karan Dalal; Daniel Koceja; Gashon Hussein; Jiarui Xu; Yue Zhao 006; Youjin Song; Shihao Han; Ka Chun Cheung; Jan Kautz; Carlos Guestrin; Tatsunori Hashimoto; Sanmi Koyejo; et al. (CoRR, 2025)