most citedSparse Sinkhorn Attention

77 citations · 97 across the 7 of their papers we have counts for

collaborators

11 papers

cs.CV20219 cited

OmniNet: Omnidirectional Representations from Transformers

Yi Tay, Mostafa Dehghani, Vamsi Aribandi +6

This paper proposes Omnidirectional Representations from Transformers (OmniNet). In OmniNet, instead of maintaining a strictly horizontal receptive field, each token is allowed to…

cs.LG2021

Label Smoothed Embedding Hypothesis for Out-of-Distribution Detection

Dara Bahri, Heinrich Jiang, Yi Tay +1

Detecting out-of-distribution (OOD) examples is critical in many applications. We propose an unsupervised method to detect OOD samples using a -NN density estimate with respect…

cs.LG2021

Locally Adaptive Label Smoothing for Predictive Churn

Dara Bahri, Heinrich Jiang

Training modern neural networks is an inherently noisy process that can lead to high \emph{prediction churn} -- disagreements between re-trainings of the same model due to factors…

cs.CL2020

StructFormer: Joint Unsupervised Induction of Dependency and Constituency Structure from Masked Language Modeling

Yikang Shen, Yi Tay, Che Zheng +3

There are two major classes of natural language grammar -- the dependency grammar that models one-to-one correspondences between words and the constituency grammar that models the…

cs.IR2020

Surprise: Result List Truncation via Extreme Value Theory

Dara Bahri, Che Zheng, Yi Tay +2

Work in information retrieval has largely been centered around ranking and relevance: given a query, return some number of results ordered by relevance to the user. The problem of…

cs.CL2020

Generative Models are Unsupervised Predictors of Page Quality: A Colossal-Scale Study

Dara Bahri, Yi Tay, Che Zheng +3

Large generative language models such as GPT-2 are well-known for their ability to generate text as well as their utility in supervised downstream tasks via fine-tuning. Our work i…