Cai, Ziyang

5 publications

ICML 2025 Everything Everywhere All at Once: LLMs Can In-Context Learn Multiple Tasks in Superposition Zheyang Xiong, Ziyang Cai, John Cooper, Albert Ge, Vasilis Papageorgiou, Zack Sifakis, Angeliki Giannou, Ziqian Lin, Liu Yang, Saurabh Agarwal, Grigorios Chrysos, Samet Oymak, Kangwook Lee, Dimitris Papailiopoulos

NeurIPS 2025 Extrapolation by Association: Length Generalization Transfer in Transformers Ziyang Cai, Nayoung Lee, Avi Schwarzschild, Samet Oymak, Dimitris Papailiopoulos

ICML 2025 Self-Improving Transformers Overcome Easy-to-Hard and Length Generalization Challenges Nayoung Lee, Ziyang Cai, Avi Schwarzschild, Kangwook Lee, Dimitris Papailiopoulos

ICLRW 2025 Self-Improving Transformers Overcome Easy-to-Hard and Length Generalization Challenges Nayoung Lee, Ziyang Cai, Avi Schwarzschild, Kangwook Lee, Dimitris Papailiopoulos

NeurIPS 2022 Delving into Out-of-Distribution Detection with Vision-Language Representations Yifei Ming, Ziyang Cai, Jiuxiang Gu, Yiyou Sun, Wei Li, Yixuan Li