← Search

Zixin Wen

7 accepted papers

2026

On the Learning Dynamics of RLVR at the Edge of Competence

ICML 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) has been a main driver of recent breakthroughs in large reasoning models. Yet it remains a mystery how rewards based solely on final outcomes can help overcome the long-horizon barrier to extended reasoning. To understand this, we develop a theor…

Cited by 0SourceScholar
2025

A Theoretical Analysis of Self-Supervised Learning for Vision Transformers

ICLR 2025poster

Self-supervised learning has become a cornerstone in computer vision, primarily divided into reconstruction-based methods like masked autoencoders (MAE) and discriminative methods such as contrastive learning (CL). Recent empirical observations reveal that MAE and CL capture different types of repr…

Cited by 0SourcePDFScholar
2025

Faster WIND: Accelerating Iterative Best-of-$N$ Distillation for LLM Alignment

AISTATS 2025poster

Recent advances in aligning large language models with human preferences have corroborated the growing importance of best-of-$N$ distillation (BOND). However, the iterative BOND algorithm is prohibitively expensive in practice due to the sample and computation inefficiency. This paper addresses the…

Cited by 0SourceScholar
2025

Transformers Provably Learn Chain-of-Thought Reasoning with Length Generalization

NeurIPS 2025poster

The ability to reason lies at the core of artificial intelligence (AI), and challenging problems usually call for deeper and longer reasoning to tackle. A crucial question about AI reasoning is whether models can extrapolate learned reasoning patterns to solve harder tasks that require longer chain…

Cited by 0SourceScholar
2024

Revisiting Disentanglement in Downstream Tasks: A Study on Its Necessity for Abstract Visual Reasoning

AAAI 2024technical

In representation learning, a disentangled representation is highly desirable as it encodes generative factors of data in a separable and compact pattern. Researchers have advocated leveraging disentangled representations to complete downstream tasks with encouraging empirical evidence. This paper f…

2021

Toward Understanding the Feature Learning Process of Self-supervised Contrastive Learning

ICML 2021spotlight

We formally study how contrastive learning learns the feature representations for neural networks by investigating its feature learning process. We consider the case where our data are comprised of two types of features: the sparse features which we want to learn from, and the dense features we want…

Cited by 170SourcePDFScholar