Convolutional Sequence Modeling Revisited. - Shaojie Bai; J. Zico Kolter; Vladlen Koltun (ICLR (Workshop), 2018)

An Empirical Evaluation of Generic Convolutional and Recurrent Networks for Sequence Modeling. - Shaojie Bai; J. Zico Kolter; Vladlen Koltun (CoRR, 2018)

Multimodal Transformer for Unaligned Multimodal Language Sequences. - Yao-Hung Hubert Tsai; Shaojie Bai; Paul Pu Liang; J. Zico Kolter; Louis-Philippe Morency; Ruslan Salakhutdinov (ACL (1), 2019)

Transformer Dissection: An Unified Understanding for Transformer's Attention via the Lens of Kernel. - Yao-Hung Hubert Tsai; Shaojie Bai; Makoto Yamada; Louis-Philippe Morency; Ruslan Salakhutdinov (EMNLP/IJCNLP (1), 2019)

Trellis Networks for Sequence Modeling. - Shaojie Bai; J. Zico Kolter; Vladlen Koltun (ICLR (Poster), 2019)

Deep Equilibrium Models. - Shaojie Bai; J. Zico Kolter; Vladlen Koltun (NeurIPS, 2019)

Multiscale Deep Equilibrium Models. - Shaojie Bai; Vladlen Koltun; J. Zico Kolter (NeurIPS, 2020)

A community-powered search of machine learning strategy space to find NMR property prediction models. - Lars A. Bratholm; Will Gerrard; Brandon M. Anderson; Shaojie Bai; Sunghwan Choi; Lam Dang; Pavel Hanchar; Addison Howard; Guillaume Huard; Sanghoon Kim; J. Zico Kolter; Risi Kondor; et al. (CoRR, 2020)

AHEAD: Adaptive Hierarchical Decomposition for Range Query under Local Differential Privacy. - Linkang Du; Zhikun Zhang 001; Shaojie Bai; Changchang Liu; Shouling Ji; Peng Cheng 001; Jiming Chen 001 (CCS, 2021)

Stabilizing Equilibrium Models by Jacobian Regularization. - Shaojie Bai; Vladlen Koltun; J. Zico Kolter (ICML, 2021)

GraFin: An Applicable Graph-based Fingerprinting Approach for Robust Indoor Localization. - Han Zheng; Yan Zhang; Lan Zhang 002; Hao Xia; Shaojie Bai; Guobin Shen; Tian He 001; Xiangyang Li 001 (ICPADS, 2021)

On Training Implicit Models. - Zhengyang Geng; Xin-Yu Zhang; Shaojie Bai; Yisen Wang 001; Zhouchen Lin (NeurIPS, 2021)

Joint inference and input optimization in equilibrium networks. - Swaminathan Gurumurthy; Shaojie Bai; Zachary Manchester; J. Zico Kolter (NeurIPS, 2021)

$( textrm{Implicit})^2$: Implicit Layers for Implicit Representations. - Zhichun Huang; Shaojie Bai; J. Zico Kolter (NeurIPS, 2021)

A Note on Connecting Barlow Twins with Negative-Sample-Free Contrastive Learning. - Yao-Hung Hubert Tsai; Shaojie Bai; Louis-Philippe Morency; Ruslan Salakhutdinov (CoRR, 2021)

Deep Equilibrium Optical Flow Estimation. - Shaojie Bai; Zhengyang Geng; Yash Savani; J. Zico Kolter (CVPR, 2022)

Neural Deep Equilibrium Solvers. - Shaojie Bai; Vladlen Koltun; J. Zico Kolter (ICLR, 2022)

SHINE: SHaring the INverse Estimate from the forward pass for bi-level optimization and implicit models. - Zaccharie Ramzi; Florian Mannel; Shaojie Bai; Jean-Luc Starck; Philippe Ciuciu; Thomas Moreau 001 (ICLR, 2022)

Path Independent Equilibrium Models Can Better Exploit Test-Time Computation. - Cem Anil; Ashwini Pokle; Kaiqu Liang; Johannes Treutlein; Yuhuai Wu; Shaojie Bai; J. Zico Kolter; Roger B. Grosse (NeurIPS, 2022)

Equilibrium Approaches to Modern Deep Learning - Shaojie Bai (2022)

Stability of Weighted Majority Voting under Estimated Weights. - Shaojie Bai; Dongxia Wang 002; Tim Muller; Peng Cheng 001; Jiming Chen 001 (AAMAS, 2023)

From Audio to Photoreal Embodiment: Synthesizing Humans in Conversations. - Evonne Ng; Javier Romero 002; Timur M. Bagautdinov; Shaojie Bai; Trevor Darrell; Angjoo Kanazawa; Alexander Richard (CVPR, 2024)

Fast Registration of Photorealistic Avatars for VR Facial Animation. - Chaitanya Patel; Shaojie Bai; Te-Li Wang; Jason M. Saragih; Shih-En Wei (ECCV (62), 2024)

Codec Avatar Studio: Paired Human Captures for Complete, Driveable, and Generalizable Avatars. - Julieta Martinez 001; Emily Kim; Javier Romero 002; Timur M. Bagautdinov; Shunsuke Saito; Shoou-I Yu; Stuart Anderson; Michael Zollhöfer; Te-Li Wang; Shaojie Bai; Chenghui Li; Shih-En Wei; et al. (NeurIPS, 2024)

Differentially Private No-regret Exploration in Adversarial Markov Decision Processes. - Shaojie Bai; Lanting Zeng; Chengcheng Zhao; Xiaoming Duan; Mohammad Sadegh Talebi; Peng Cheng 001; Jiming Chen 001 (UAI, 2024)

Universal Facial Encoding of Codec Avatars from VR Headsets. - Shaojie Bai; Te-Li Wang; Chenghui Li; Akshay Venkatesh; Tomas Simon; Chen Cao 001; Gabriel Schwartz; Ryan Wrench; Jason M. Saragih; Yaser Sheikh; Shih-En Wei (CoRR, 2024)

No-regret Exploration in Shuffle Private Reinforcement Learning. - Shaojie Bai; Mohammad Sadegh Talebi; Chengcheng Zhao; Peng Cheng 001; Jiming Chen 001 (CoRR, 2024)

Reference Image-Assisted Auxiliary Feature Fusion in Image Inpainting. - Shaojie Bai; Lixia Lin; Zihan Hu; Peng Cao (IEEE Signal Process. Lett., 2024)

Universal Facial Encoding of Codec Avatars from VR Headsets. - Shaojie Bai; Te-Li Wang; Chenghui Li; Akshay Venkatesh; Tomas Simon; Chen Cao 001; Gabriel Schwartz; Jason M. Saragih; Yaser Sheikh; Shih-En Wei (ACM Trans. Graph., 2024)

REWIND: Real-Time Egocentric Whole-Body Motion Diffusion with Exemplar-Based Identity Conditioning. - Jihyun Lee; Weipeng Xu; Alexander Richard; Shih-En Wei; Shunsuke Saito; Shaojie Bai; Te-Li Wang; Minhyuk Sung; Tae-Kyun Kim 001; Jason M. Saragih (CVPR, 2025)

Generative Head-Mounted Camera Captures for Photorealistic Avatars. - Shaojie Bai; Seunghyeon Seo; Yida Wang; Chenghui Li; Owen Wang; Te-Li Wang; Tianyang Ma; Jason M. Saragih; Shih-En Wei; Nojun Kwak; Hyung Jun Kim (CoRR, 2025)