← Search

Rosie Zhao

9 accepted papers

2026

On Robustness and Chain-of-Thought Consistency of RL-Finetuned VLMs

ICML 2026poster

Reinforcement learning (RL) fine-tuning is now widely used to improve LLM reasoning, and recent work has begun extending it to vision-language models (VLMs). While RL-tuned VLMs can improve visual reasoning benchmark performance, they can still suffer from weak visual grounding, hallucinations, and …

Cited by 0SourceScholar
2026

Parameter-Efficient Reinforcement Learning using Prefix Optimization

ICLR 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) is a leading approach for tuning language models on mathematical reasoning tasks. However, it remains unclear whether RLVR's gains stem from genuine reasoning improvements or simply from steering the model toward answer formats that already appea…

Cited by 0SourcecodeScholar
2026

Using cognitive models to reveal value trade-offs in language models

ICLR 2026poster

Value trade-offs are an integral part of human decision-making and language use, however, current tools for interpreting such dynamic and multi-faceted notions of values in LLMs are limited. In cognitive science, so-called “cognitive models” provide formal accounts of such trade-offs in humans, by m…

Cited by 0SourcecodeScholar
2025

Deconstructing What Makes a Good Optimizer for Autoregressive Language Models

ICLR 2025poster

Training language models becomes increasingly expensive with scale, prompting numerous attempts to improve optimization efficiency. Despite these efforts, the Adam optimizer remains the most widely used, due to a prevailing view that it is the most effective approach. We aim to compare several optim…

Cited by 0SourcePDFScholar
2025

SOAP: Improving and Stabilizing Shampoo using Adam for Language Modeling

ICLR 2025poster

There is growing evidence of the effectiveness of Shampoo, a higher-order preconditioning method, over Adam in deep learning optimization tasks. However, Shampoo's drawbacks include additional hyperparameters and computational overhead when compared to Adam, which only updates running averages of fi…

2024

Beyond Implicit Bias: The Insignificance of SGD Noise in Online Learning

ICML 2024spotlight

The success of SGD in deep learning has been ascribed by prior works to the *implicit bias* induced by finite batch sizes (''SGD noise''). While prior works focused on *offline learning* (i.e., multiple-epoch training), we study the impact of SGD noise on *online* (i.e., single epoch) learning. Thro…

Cited by 3SourcePDFScholar
2024

Feature emergence via margin maximization: case studies in algebraic tasks

ICLR 2024spotlight

Understanding the internal representations learned by neural networks is a cornerstone challenge in the science of machine learning. While there have been significant recent strides in some cases towards understanding *how* neural networks implement specific target functions, this paper explores a c…

Cited by 15SourcePDFScholar
2022

Continuous MDP Homomorphisms and Homomorphic Policy Gradient

NeurIPS 2022accept

Abstraction has been widely studied as a way to improve the efficiency and generalization of reinforcement learning algorithms. In this paper, we study abstraction in the continuous-control setting. We extend the definition of MDP homomorphisms to encompass continuous actions in continuous state spa…