← Search

Zheyang Xiong

5 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

From Artificial Needles to Real Haystacks: Improving Retrieval Capabilities in LLMs by Finetuning on Synthetic Data

ICLR 2025poster

Recent studies have shown that Large Language Models (LLMs) struggle to accurately retrieve information and maintain reasoning capabilities when processing long-context inputs. To address these limitations, we propose a finetuning approach utilizing a carefully designed synthetic dataset comprising…

2025

OmniEdit: Building Image Editing Generalist Models Through Specialist Supervision

ICLR 2025poster

Instruction-guided image editing methods have demonstrated significant potential by training diffusion models on automatically synthesized or manually annotated image editing pairs. However, these methods remain far from practical, real-life applications. We identify three primary challenges contrib…

Cited by 10SourcePDFScholar
2024

Can Mamba Learn How To Learn? A Comparative Study on In-Context Learning Tasks

ICML 2024poster

State-space models (SSMs), such as Mamba (Gu & Dao, 2023), have been proposed as alternatives to Transformer networks in language modeling, incorporating gating, convolutions, and input-dependent token selection to mitigate the quadratic cost of multi-head attention. Although SSMs exhibit competitiv…

2023

Strong Lottery Ticket Hypothesis with $\varepsilon$–perturbation

AISTATS 2023poster

The strong Lottery Ticket Hypothesis (LTH) (Ramanujan et al., 2019; Zhou et al., 2019) claims the existence of a subnetwork in a sufficiently large, randomly initialized neural network that approximates some target neural network without the need of training. We extend the theoretical guarantee of t…

Cited by 0SourcePDFScholar