Adversarial Grasp Objects. - David Wang; David Tseng; Pusong Li; Yiding Jiang; Menglong Guo; Michael Danielczuk; Jeffrey Mahler; Jeffrey Ichnowski; Ken Goldberg (CASE, 2019)

Predicting the Generalization Gap in Deep Networks with Margin Distributions. - Yiding Jiang; Dilip Krishnan; Hossein Mobahi; Samy Bengio (ICLR (Poster), 2019)

Language as an Abstraction for Hierarchical Deep Reinforcement Learning. - Yiding Jiang; Shixiang Gu; Kevin Murphy; Chelsea Finn (NeurIPS, 2019)

Fantastic Generalization Measures and Where to Find Them. - Yiding Jiang; Behnam Neyshabur; Hossein Mobahi; Dilip Krishnan; Samy Bengio (ICLR, 2020)

Observational Overfitting in Reinforcement Learning. - Xingyou Song; Yiding Jiang; Stephen Tu; Yilun Du; Behnam Neyshabur (ICLR, 2020)

Methods and Analysis of The First Competition in Predicting Generalization of Deep Learning. - Yiding Jiang; Parth Natekar; Manik Sharma; Sumukh K. Aithal; Dhruva Kashyap; Natarajan Subramanyam; Carlos Lassance; Daniel M. Roy 001; Gintare Karolina Dziugaite; Suriya Gunasekar; Isabelle Guyon; Pierre Foret; et al. (NeurIPS (Competition and Demos), 2020)

NeurIPS 2020 Competition: Predicting Generalization in Deep Learning. - Yiding Jiang; Pierre Foret; Scott Yak; Daniel M. Roy 001; Hossein Mobahi; Gintare Karolina Dziugaite; Samy Bengio; Suriya Gunasekar; Isabelle Guyon; Behnam Neyshabur (CoRR, 2020)

Ask & Explore: Grounded Question Answering for Curiosity-Driven Exploration. - Jivat Neet Kaur; Yiding Jiang; Paul Pu Liang (CoRR, 2021)

Assessing Generalization of SGD via Disagreement. - Yiding Jiang; Vaishnavh Nagarajan; Christina Baek; J. Zico Kolter (ICLR, 2022)

Agreement-on-the-line: Predicting the Performance of Neural Networks under Distribution Shift. - Christina Baek; Yiding Jiang; Aditi Raghunathan; J. Zico Kolter (NeurIPS, 2022)

Learning Options via Compression. - Yiding Jiang; Evan Zheran Liu; Benjamin Eysenbach; J. Zico Kolter; Chelsea Finn (NeurIPS, 2022)

On the Importance of Exploration for Generalization in Reinforcement Learning. - Yiding Jiang; J. Zico Kolter; Roberta Raileanu (NeurIPS, 2023)

Language Models are Weak Learners. - Hariharan Manikandan; Yiding Jiang; J. Zico Kolter (NeurIPS, 2023)

Permutation Equivariant Neural Functionals. - Allan Zhou; Kaien Yang; Kaylee Burns; Adriano Cardace; Yiding Jiang; Samuel Sokota; J. Zico Kolter; Chelsea Finn (NeurIPS, 2023)

Neural Functional Transformers. - Allan Zhou; Kaien Yang; Yiding Jiang; Kaylee Burns; Winnie Xu; Samuel Sokota; J. Zico Kolter; Chelsea Finn (NeurIPS, 2023)

A Wavelet Decomposition Network Based Edge Monitoring System for Classification of Electrical Equipment. - Xianchen Wang; Di Cao; Yiding Jiang; Yanjing Lei; Linghao Ying (SMC, 2023)

Permutation Equivariant Neural Functionals. - Allan Zhou; Kaien Yang; Kaylee Burns; Yiding Jiang; Samuel Sokota; J. Zico Kolter; Chelsea Finn (CoRR, 2023)

Understanding prompt engineering may not require rethinking generalization. - Victor Akinwande; Yiding Jiang; Dylan Sam; J. Zico Kolter (ICLR, 2024)

On the Joint Interaction of Models, Data, and Features. - Yiding Jiang; Christina Baek; J. Zico Kolter (ICLR, 2024)

Improving Generalization on the ProcGen Benchmark with Simple Architectural Changes and Scale. - Andrew Jesson; Yiding Jiang (CoRR, 2024)

Adaptive Data Optimization: Dynamic Sample Selection with Scaling Laws. - Yiding Jiang; Allan Zhou; Zhili Feng; Sadhika Malladi; J. Zico Kolter (ICLR, 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 (Trans. Mach. Learn. Res., 2025)

Training a Generally Curious Agent. - Fahim Tajwar; Yiding Jiang; Abitha Thankaraj; Sumaita Sadia Rahman; J. Zico Kolter; Jeff Schneider 001; Ruslan Salakhutdinov (CoRR, 2025)

Looking beyond the next token. - Abitha Thankaraj; Yiding Jiang; J. Zico Kolter; Yonatan Bisk (CoRR, 2025)

Safety Pretraining: Toward the Next Generation of Safe AI. - Pratyush Maini; Sachin Goyal; Dylan Sam; Alexander Robey; Yash Savani; Yiding Jiang; Andy Zou; Zacharcy C. Lipton; J. Zico Kolter (CoRR, 2025)