activity
20122023
most citedGo for a Walk and Arrive at the Answer: Reasoning Over Paths in Knowledge Bases using Reinforcement Learning

398 citations · 1.7k across the 67 of their papers we have counts for

collaborators

78 papers

cs.LG2023

Oracle-Efficient Pessimism: Offline Policy Optimization in Contextual Bandits

Lequn Wang, Akshay Krishnamurthy, Aleksandrs Slivkins

We consider offline policy optimization (OPO) in contextual bandits, where one is given a fixed dataset of logged interactions. While pessimistic regularizers are typically used to…

cs.LG2023★ 9 cited

Exposing Attention Glitches with Flip-Flop Language Modeling

Bingbin Liu, Jordan T. Ash, Surbhi Goel +2

Why do large language models sometimes output factual inaccuracies and exhibit erroneous reasoning? The brittleness of these models, particularly when executing long chains of reas…

cs.LG2023★ 2 cited

Streaming Active Learning with Deep Neural Networks

Akanksha Saran, Safoora Yousefi, Akshay Krishnamurthy +2

Active learning is perhaps most naturally posed as an online learning problem. However, prior active learning approaches with deep neural networks assume offline access to the enti…

cs.LG2023

Learning Hidden Markov Models Using Conditional Samples

Sham M. Kakade, Akshay Krishnamurthy, Gaurav Mahajan +1

This paper is concerned with the computational complexity of learning the Hidden Markov Model (HMM). Although HMMs are some of the most widely used tools in sequential and time ser…

cs.LG2023★ 1 cited

Statistical Learning under Heterogeneous Distribution Shift

Max Simchowitz, Anurag Ajay, Pulkit Agrawal +1

This paper studies the prediction of a target from a pair of random variables , where the ground-truth predictor is additive $\mathbb{E}[\math…

cs.LG2022★ 11 cited

Transformers Learn Shortcuts to Automata

Bingbin Liu, Jordan T. Ash, Surbhi Goel +2

Algorithmic reasoning requires capabilities which are most naturally understood through recurrent models of computation, like the Turing machine. However, Transformer models, while…