activity
20202023
most citedMastering Atari Games with Limited Data

40 citations · 79 across the 14 of their papers we have counts for

collaborators
Showing 2021Show all

7 papers · 1 filter

cs.LG2021★ 40 cited

Mastering Atari Games with Limited Data

Weirui Ye, Shaohuai Liu, Thanard Kurutach +2

Reinforcement learning has achieved great success in many applications. However, sample efficiency remains a key challenge, with prominent methods requiring millions (or even billi…

cs.AI2021★ 5 cited

Maximum Entropy Population-Based Training for Zero-Shot Human-AI Coordination

Rui Zhao, Jinming Song, Yufeng Yuan +5

We study the problem of training a Reinforcement Learning (RL) agent that is collaborative with humans without using any human data. Although such agents can be obtained through se…

cs.LG2021

Differentiable Architecture Pruning for Transfer Learning

Nicolo Colombo, Yang Gao

We propose a new gradient-based approach for extracting sub-architectures from a given large model. Contrarily to existing pruning methods, which are unable to disentangle the netw…

eess.AS2021

Do You Listen with One or Two Microphones? A Unified ASR Model for Single and Multi-Channel Audio

Gokce Keskin, Minhua Wu, Brian King +5

Automatic speech recognition (ASR) models are typically designed to operate on a single input data type, e.g. a single or multi-channel audio streamed from a device. This design de…

eess.AS2021★ 1 cited

Attention-based Neural Beamforming Layers for Multi-channel Speech Recognition

Bhargav Pulugundla, Yang Gao, Brian King +5

Attention-based beamformers have recently been shown to be effective for multi-channel speech recognition. However, they are less capable at capturing local information. In this wo…

cs.LG2021★ 10 cited

Adapting by Pruning: A Case Study on BERT

Yang Gao, Nicolo Colombo, Wei Wang

Adapting pre-trained neural models to downstream tasks has become the standard practice for obtaining high-quality models. In this work, we propose a novel model adaptation paradig…

cs.LG2021★ 3 cited

Mutual Information State Intrinsic Control

Rui Zhao, Yang Gao, Pieter Abbeel +2

Reinforcement learning has been shown to be highly successful at many challenging tasks. However, success heavily relies on well-shaped rewards. Intrinsically motivated RL attempts…