← Search

Santiago Ontanon

10 accepted papers

2024

Cleanba: A Reproducible and Efficient Distributed Reinforcement Learning Platform

ICLR 2024poster

Distributed Deep Reinforcement Learning (DRL) aims to leverage more computational resources to train autonomous agents with less training time. Despite recent progress in the field, reproducibility issues have not been sufficiently explored. This paper first shows that the typical actor-learner fram…

2024

Functional Interpolation for Relative Positions improves Long Context Transformers

ICLR 2024poster

Preventing the performance decay of Transformers on inputs longer than those used for training has been an important challenge in extending the context length of these models. Though the Transformer architecture has fundamentally no limits on the input sequence lengths it can process, the choice of…

Cited by 47SourcePDFScholar
2024

MEMORY-VQ: Compression for Tractable Internet-Scale Memory

NAACL 2024short

Retrieval augmentation is a powerful but expensive method to make language models more knowledgeable about the world. Memory-based methods like LUMEN (de Jong et al., 2023a) pre-compute token representations for retrieved passages to drastically speed up inference. However, memory also leads to much…

Cited by 0SourcePDFScholar
2023

CoLT5: Faster Long-Range Transformers with Conditional Computation

EMNLP 2023long main

Many natural language processing tasks benefit from long inputs, but processing long documents with Transformers is expensive -- not only due to quadratic attention complexity but also from applying feedforward and projection layers to every token. However, not all tokens are equally important, espe…

Cited by 0SourceScholar
2023

mLongT5: A Multilingual and Efficient Text-To-Text Transformer for Longer Sequences

EMNLP 2023short findings

We present our work on developing a multilingual, efficient text-to-text transformer that is suitable for handling long inputs. This model, called mLongT5, builds upon the architecture of LongT5, while leveraging the multilingual datasets used for pretraining mT5 and the pretraining tasks of UL2. We…

Cited by 0SourcecodeScholar
2022

FNet: Mixing Tokens with Fourier Transforms

NAACL 2022long

We show that Transformer encoder architectures can be sped up, with limited accuracy costs, by replacing the self-attention sublayers with simple linear transformations that “mix” input tokens. Most surprisingly, we find that replacing the self-attention sublayer in a Transformer encoder with a stan…

2022

LongT5: Efficient Text-To-Text Transformer for Long Sequences

NAACL 2022findings

Recent work has shown that either (1) increasing the input length or (2) increasing model size can improve the performance of Transformer-based neural models. In this paper, we present LongT5, a new model that explores the effects of scaling both the input length and model size at the same time. Spe…

2022

Making Transformers Solve Compositional Tasks

ACL 2022long

Several studies have reported the inability of Transformer models to generalize compositionally, a key type of generalization in many NLP tasks such as semantic parsing. In this paper we explore the design space of Transformer models showing that the inductive biases given to the model by several de…

2021

Improving Compositional Generalization in Classification Tasks via Structure Annotations

ACL 2021short

Compositional generalization is the ability to generalize systematically to a new data distribution by combining known components. Although humans seem to have a great ability to generalize compositionally, state-of-the-art neural models struggle to do so. In this work, we study compositional genera…

Cited by 17SourcePDFScholar
2020

Big Bird: Transformers for Longer Sequences

NeurIPS 2020poster

Transformers-based models, such as BERT, have been one of the most successful deep learning models for NLP. Unfortunately, one of their core limitations is the quadratic dependency (mainly in terms of memory) on the sequence length due to their full attention mechanism. To remedy this, we propose,…