← Search

Allen Nie

13 accepted papers

2026

Assessing the Quality of AI-Generated Exams: A Large-Scale Field Study

AAAI 2026technical

While large language models (LLMs) challenge conventional methods of teaching and learning, they present an exciting opportunity to improve efficiency and scale high-quality instruction. One promising application is the generation of customized exams, tailored to specific course content. There has b

Cited by 0SourcePDFScholar
2026

Formalizing Learning from Language Feedback with Provable Guarantees

ICML 2026poster

Interactively learning from observation and language feedback is an increasingly studied area driven by the emergence of large language model (LLM) agents. While impressive empirical demonstrations have been shown, so far a principled framing of these decision problems remains lacking. In this paper…

Cited by 0SourceScholar
2025

EVOLvE: Evaluating and Optimizing LLMs For In-Context Exploration

ICML 2025poster

Despite their success in many domains, large language models (LLMs) remain under-studied in scenarios requiring optimal decision-making under uncertainty. This is crucial as many real-world applications, ranging from personalized recommendations to healthcare interventions, demand that LLMs not only…

Cited by 5SourcePDFScholar
2025

Improving Parallel Program Performance with LLM Optimizers via Agent-System Interfaces

ICML 2025poster

Modern scientific discovery increasingly relies on high-performance computing for complex modeling and simulation. A key challenge in improving parallel program performance is efficiently mapping tasks to processors and data to memory, a process dictated by intricate, low-level system code known as…

Cited by 0SourcePDFScholar
2024

OPERA: Automatic Offline Policy Evaluation with Re-weighted Aggregates of Multiple Estimators

NeurIPS 2024poster

Offline policy evaluation (OPE) allows us to evaluate and estimate a new sequential decision-making policy's performance by leveraging historical interaction data collected from other policies. Evaluating a new policy online without a confident estimate of its performance can lead to costly, unsafe,…

Cited by 0SourcePDFScholar
2024

Trace is the Next AutoDiff: Generative Optimization with Rich Feedback, Execution Traces, and LLMs

NeurIPS 2024poster

We study a class of optimization problems motivated by automating the design and update of AI systems like coding assistants, robots, and copilots. AutoDiff frameworks, like PyTorch, enable efficient end-to-end optimization of differentiable systems. However, general computational workflows can be n…

2023

MoCa: Measuring Human-Language Model Alignment on Causal and Moral Judgment Tasks

NeurIPS 2023poster

Human commonsense understanding of the physical and social world is organized around intuitive theories. These theories support making causal and moral judgments. When something bad happens, we naturally ask: who did what, and why? A rich literature in cognitive science has studied people's causal a…

Cited by 40SourcePDFScholar
2023

Model-Based Offline Reinforcement Learning with Local Misspecification

AAAI 2023technical

We present a model-based offline reinforcement learning policy performance lower bound that explicitly captures dynamics model misspecification and distribution mismatch and we propose an empirical algorithm for optimal offline policy selection. Theoretically, we prove a novel safe policy improvemen…

Cited by 4SourcePDFScholar
2023

Waypoint Transformer: Reinforcement Learning via Supervised Learning with Intermediate Targets

NeurIPS 2023poster

Despite the recent advancements in offline reinforcement learning via supervised learning (RvS) and the success of the decision transformer (DT) architecture in various domains, DTs have fallen short in several challenging benchmarks. The root cause of this underperformance lies in their inability t…

Cited by 21SourcePDFScholar
2022

Data-Efficient Pipeline for Offline Reinforcement Learning with Limited Data

NeurIPS 2022accept

Offline reinforcement learning (RL) can be used to improve future performance by leveraging historical data. There exist many different algorithms for offline RL, and it is well recognized that these algorithms, and their hyperparameter settings, can lead to decision policies with substantially diff…

Cited by 20SourcePDFScholar
2022

Giving Feedback on Interactive Student Programs with Meta-Exploration

NeurIPS 2022accept

Developing interactive software, such as websites or games, is a particularly engaging way to learn computer science. However, teaching and giving feedback on such software is time-consuming — standard approaches require instructors to manually grade student-implemented interactive programs. As a re…

2021

Play to Grade: Testing Coding Games as Classifying Markov Decision Process

NeurIPS 2021poster

Contemporary coding education often presents students with the task of developing programs that have user interaction and complex dynamic systems, such as mouse based games. While pedagogically compelling, there are no contemporary autonomous methods for providing feedback. Notably, interactive prog…