← Search

Keith W. Ross

8 accepted papers

2025

Warm Up Before You Train: Unlocking General Reasoning in Resource-Constrained Settings

EMNLP 2025

Designing effective reasoning-capable LLMs typically requires training using Reinforcement Learning with Verifiable Rewards (RLVR) or distillation with carefully curated Long Chain of Thoughts (CoT), both of which depend heavily on extensive training data. This creates a major challenge when the amo

2024

Pre-training with Synthetic Data Helps Offline Reinforcement Learning

ICLR 2024poster

Recently, it has been shown that for offline deep reinforcement learning (DRL), pre-training Decision Transformer with a large language corpus can improve downstream performance (Reid et al., 2022). A natural question to ask is whether this performance gain can only be achieved with language pre-tra…

2024

The Prevalence of Neural Collapse in Neural Multivariate Regression

NeurIPS 2024poster

Recently it has been observed that neural networks exhibit Neural Collapse (NC) during the final stage of training for the classification problem. We empirically show that multivariate regression, as employed in imitation learning and other applications, exhibits Neural Regression Collapse (NRC), a…

Cited by 4SourcePDFScholar
2022

On the Convergence of the Monte Carlo Exploring Starts Algorithm for Reinforcement Learning

ICLR 2022poster

A simple and natural algorithm for reinforcement learning (RL) is Monte Carlo Exploring Starts (MCES), where the Q-function is estimated by averaging the Monte Carlo returns, and the policy is improved by choosing actions that maximize the current estimate of the Q-function. Exploration is performed…

Cited by 28SourcePDFScholar
2022

VRL3: A Data-Driven Framework for Visual Deep Reinforcement Learning

NeurIPS 2022accept

We propose VRL3, a powerful data-driven framework with a simple design for solving challenging visual deep reinforcement learning (DRL) tasks. We analyze a number of major obstacles in taking a data-driven approach, and present a suite of design principles, novel findings, and critical insights abou…

2021

Randomized Ensembled Double Q-Learning: Learning Fast Without a Model

ICLR 2021poster

Using a high Update-To-Data (UTD) ratio, model-based methods have recently achieved much higher sample efficiency than previous model-free methods for continuous-action DRL benchmarks. In this paper, we introduce a simple model-free algorithm, Randomized Ensembled Double Q-Learning (REDQ), and show…