← Search

Stephen Roller

8 accepted papers

2023

Scaling Laws for Generative Mixed-Modal Language Models

ICML 2023poster

Generative language models define distributions over sequences of tokens that can represent essentially any combination of data modalities (e.g., any permutation of image tokens from VQ-VAEs, speech tokens from HuBERT, BPE tokens for language or code, and so on). To better understand the scaling pro…

Cited by 104SourcePDFScholar
2022

Language Models that Seek for Knowledge: Modular Search & Generation for Dialogue and Prompt Completion

EMNLP 2022finding

Language models (LMs) have recently been shown to generate more factual responses by employing modularity (Zhou et al., 2022) in combination with retrieval (Adolphs et al., 2021). We extend the recent approach of Adolphs et al. (2021) to include internet search as a module. Our SeeKeR (Search engine…

Cited by 124SourcePDFScholar
2022

Staircase Attention for Recurrent Processing of Sequences

NeurIPS 2022accept

Attention mechanisms have become a standard tool for sequence modeling tasks, in particular by stacking self-attention layers over the entire input sequence as in the Transformer architecture. In this work we introduce a novel attention procedure called staircase attention that, unlike self-attentio…

Cited by 16SourcePDFScholar
2021

Adding Chit-Chat to Enhance Task-Oriented Dialogues

NAACL 2021long

Existing dialogue corpora and models are typically designed under two disjoint motives: while task-oriented systems focus on achieving functional goals (e.g., booking hotels), open-domain chatbots aim at making socially engaging conversations. In this work, we propose to integrate both types of syst…

Cited by 89SourcePDFScholar
2021

Not All Memories are Created Equal: Learning to Forget by Expiring

ICML 2021oral

Attention mechanisms have shown promising results in sequence modeling tasks that require long-term memory. Recent work investigated mechanisms to reduce the computational cost of preserving and storing memories. However, not all content in the past is equally important to remember. We propose Expir…

2020

Neural Text Generation With Unlikelihood Training

ICLR 2020poster

Neural text generation is a key tool in natural language applications, but it is well known there are major problems at its core. In particular, standard likelihood training and decoding leads to dull and repetitive outputs. While some post-hoc fixes have been proposed, in particular top-k and nucle…

Cited by 622SourcecodeScholar
2019

Wizard of Wikipedia: Knowledge-Powered Conversational Agents

ICLR 2019poster

In open-domain dialogue intelligent agents should exhibit the use of knowledge, however there are few convincing demonstrations of this to date. The most popular sequence to sequence models typically “generate and hope” generic utterances that can be memorized in the weights of the model when mappin…

Cited by 1074SourcePDFScholar