← Search

Ziyang Cai

4 accepted papers

2025

Everything Everywhere All at Once: LLMs can In-Context Learn Multiple Tasks in Superposition

ICML 2025spotlight

Large Language Models (LLMs) have demonstrated remarkable in-context learning (ICL) capabilities. In this study, we explore a surprising phenomenon related to ICL: LLMs can perform multiple, computationally distinct ICL tasks simultaneously, during a single inference call, a capability we term task…

Cited by 3SourcePDFScholar
2025

Extrapolation by Association: Length Generalization Transfer In Transformers

NeurIPS 2025spotlight

Transformer language models have demonstrated impressive generalization capabilities in natural language domains, yet we lack a fine-grained understanding of how such generalization arises. In this paper, we investigate length generalization—the ability to extrapolate from shorter to longer inputs—t…

Cited by 0SourceScholar
2025

Self-Improving Transformers Overcome Easy-to-Hard and Length Generalization Challenges

ICML 2025poster

Large language models often struggle with length generalization and solving complex problem instances beyond their training distribution. We present a self-improvement approach where models iteratively generate and learn from their own solutions, progressively tackling harder problems while maintain…

Cited by 2SourcePDFScholar
2022

Delving into Out-of-Distribution Detection with Vision-Language Representations

NeurIPS 2022accept

Recognizing out-of-distribution (OOD) samples is critical for machine learning systems deployed in the open world. The vast majority of OOD detection methods are driven by a single modality (e.g., either vision or language), leaving the rich information in multi-modal representations untapped. Inspi…