Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Stanley Wei

dblp:146/2234 · DBLP profile ↗
← Back
4ranked-venue papers
1as first author
4since 2021 · last 2025
—ORCID · none

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 4 · 1 first-author · 4 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
4 papers
Language models and text generation · 21% Trustworthy machine learning · 14% Information extraction and text analysis · 14%
Interdisciplinary, comprehensive, and emerging computing
1 paper
Computing education · 100%

Topics — the 12 heaviest of 12, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Natural language and speech › Language models and text generation
code generation
0.912025
LiveCodeBench Pro: How Do Olympiad Medalists Judge LLMs in Competitive Programming? · NeurIPS 2025
Machine learning › Trustworthy machine learning
machine unlearning
0.912025
Provable unlearning in topic modeling and downstream tasks · ICLR 2025
Machine learning › Optimization for machine learning
optimization landscape
0.912025
What Makes a Reward Model a Good Teacher? An Optimization Perspective · NeurIPS 2025
Machine learning › Reinforcement learning
reinforcement learning from human feedback
0.912025
What Makes a Reward Model a Good Teacher? An Optimization Perspective · NeurIPS 2025
Natural language and speech › Information extraction and text analysis
topic model
0.912025
Provable unlearning in topic modeling and downstream tasks · ICLR 2025
Computing education › programming education
competitive programming
0.912025
LiveCodeBench Pro: How Do Olympiad Medalists Judge LLMs in Competitive Programming? · NeurIPS 2025
Computing education
large language model evaluation
0.912025
LiveCodeBench Pro: How Do Olympiad Medalists Judge LLMs in Competitive Programming? · NeurIPS 2025
Computing education
programming education
0.912025
LiveCodeBench Pro: How Do Olympiad Medalists Judge LLMs in Competitive Programming? · NeurIPS 2025
Machine learning › Learning theory › neural network theory
expressiveness and learnability
0.812024
Transformers Provably Learn Sparse Token Selection While Fully-Connected Nets Cannot · ICML 2024
Machine learning › Deep learning architectures and training
transformer
0.812024
Transformers Provably Learn Sparse Token Selection While Fully-Connected Nets Cannot · ICML 2024
Natural language and speech › Language models and text generation
alignment
0.312025
What Makes a Reward Model a Good Teacher? An Optimization Perspective · NeurIPS 2025
Natural language and speech › Language models and text generation
in-context learning
0.212024
Transformers Provably Learn Sparse Token Selection While Fully-Connected Nets Cannot · ICML 2024

Methods — techniques the papers use, named apart from their topics

line-by-line failure analysis · 1.7benchmark annotation · 1.7topic modeling · 0.9reward variance analysis · 0.9optimization analysis · 0.9linear head adaptation · 0.9fine-tuning · 0.9gradient descent · 0.8average-case lower bound analysis · 0.8
YearPublicationVenuePosition
2025 Provable unlearning in topic modeling and downstream tasks
abstract
Machine unlearning algorithms are increasingly important as legal concerns arise around the provenance of training data, but verifying the success of unlearning is often difficult. Provable guarantees for unlearning are often limited to supervised learning settings. In this paper, we provide the first theoretical guarantees for unlearning in the pre-training and fine-tuning paradigm by studying topic models, simple bag-of-words language models that can be adapted to solve downstream tasks like retrieval and classification. First, we design a provably effective unlearning algorithm for topic models that incurs a computational overhead independent of the size of the original dataset. Our analysis additionally quantifies the deletion capacity of the model -- i.e., the number of examples that can be unlearned without incurring a significant cost in model performance. Finally, we formally extend our analyses to account for adaptation to a given downstream task. In particular, we design an efficient algorithm to perform unlearning after fine-tuning the topic model via a linear head. Notably, we show that it is easier to unlearn pre-training data from models that have been fine-tuned to a particular task, and one can unlearn this data without modifying the base model.
Stanley Wei, Sadhika Malladi, Sanjeev Arora, Amartya Sanyal
ICLR1
2025 What Makes a Reward Model a Good Teacher? An Optimization Perspective
abstract
The success of Reinforcement Learning from Human Feedback (RLHF) critically depends on the quality of the reward model. However, while this quality is primarily evaluated through accuracy, it remains unclear whether accuracy fully captures what makes a reward model an effective teacher. We address this question from an optimization perspective. First, we prove that regardless of how accurate a reward model is, if it induces low reward variance, then the RLHF objective suffers from a flat landscape. Consequently, even a perfectly accurate reward model can lead to extremely slow optimization, underperforming less accurate models that induce higher reward variance. We additionally show that a reward model that works well for one language model can induce low reward variance, and thus a flat objective landscape, for another. These results establish a fundamental limitation of evaluating reward models solely based on accuracy or independently of the language model they guide. Experiments using models of up to 8B parameters corroborate our theory, demonstrating the interplay between reward variance, accuracy, and reward maximization rate. Overall, our findings highlight that beyond accuracy, a reward model needs to induce sufficient variance for efficient optimization.
Noam Razin, Zixuan Wang 0029, Hubert Strauss, Stanley Wei, Jason D. Lee, Sanjeev Arora
NeurIPS4
2025 LiveCodeBench Pro: How Do Olympiad Medalists Judge LLMs in Competitive Programming?
abstract
Recent reports claim that large language models (LLMs) now outperform elite humans in competitive programming. Drawing on knowledge from a group of medalists in international algorithmic contests, we revisit this claim, examining how LLMs differ from human experts and where limitations still remain. We introduce LiveCodeBench Pro, a benchmark composed of problems from Codeforces, ICPC, and IOI that are continuously updated to reduce the likelihood of data contamination. A team of Olympiad medalists annotates every problem for algorithmic categories and conducts a line-by-line analysis of failed model-generated submissions. Using this new data and benchmark, we find that frontier models still have significant limitations: without external tools, the best model achieves only 53\% pass@1 on medium-difficulty problems and 0\% on hard problems, domains where expert humans still excel. We also find that LLMs succeed at implementation-heavy problems but struggle with nuanced algorithmic reasoning and complex case analysis, often generating confidently incorrect justifications. High performance appears largely driven by implementation precision and tool augmentation, not superior reasoning. LiveCodeBench Pro thus highlights the significant gap to human grandmaster levels, while offering fine-grained diagnostics to steer future improvements in code-centric LLM reasoning.
Zihan Zheng, Zerui Cheng, Shang Zhou, Hansen He, Dongruixuan Li, Stanley Wei, Hangyi Hao, Jianzhu Yao, Peiyao Sheng, Zixuan Wang 0029, Wenhao Chai, Aleksandra Korolova, Peter Henderson 0002, Sanjeev Arora, Pramod Viswanath, Jingbo Shang, Saining Xie
NeurIPS8
2024 Transformers Provably Learn Sparse Token Selection While Fully-Connected Nets Cannot
abstract
The transformer architecture has prevailed in various deep learning settings due to its exceptional capabilities to select and compose structural information. Motivated by these capabilities, Sanford et al. (2023) proposed the *sparse token selection* task, in which transformers excel while fully-connected networks (FCNs) fail in the worst case. Building upon that, we strengthen the FCN lower bound to an average-case setting and establish an algorithmic separation of transformers over FCNs. Specifically, a one-layer transformer trained with gradient descent provably learns the sparse token selection task and, surprisingly, exhibits strong out-of-distribution length generalization. We provide empirical simulations to justify our theoretical findings.
Zixuan Wang 0029, Stanley Wei, Daniel Hsu 0001, Jason D. Lee
ICML2