← Search

Payal Bajaj

9 accepted papers

2023

Magneto: A Foundation Transformer

ICML 2023poster

A big convergence of model architectures across language, vision, speech, and multimodal is emerging. However, under the same name ''Transformers'', the above areas use different implementations for better performance, e.g., Post-LayerNorm for BERT, and Pre-LayerNorm for GPT and vision Transformers.…

Cited by 12SourcePDFScholar
2023

Model-Generated Pretraining Signals Improves Zero-Shot Generalization of Text-to-Text Transformers

ACL 2023long

This paper explores the effectiveness of model-generated signals in improving zero-shot generalization of text-to-text Transformers such as T5. We study various designs to pretrain T5 using an auxiliary model to construct more challenging token replacements for the main model to denoise. Key aspects…

2022

On the Representation Collapse of Sparse Mixture of Experts

NeurIPS 2022accept

Sparse mixture of experts provides larger model capacity while requiring a constant computational overhead. It employs the routing mechanism to distribute input tokens to the best-matched experts according to their hidden representations. However, learning such a routing mechanism encourages token c…

2022

Pretraining Text Encoders with Adversarial Mixture of Training Signal Generators

ICLR 2022poster

We present a new framework AMOS that pretrains text encoders with an Adversarial learning curriculum via a Mixture Of Signals from multiple auxiliary generators. Following ELECTRA-style pretraining, the main encoder is trained as a discriminator to detect replaced tokens generated by auxiliary maske…

2022

XLM-E: Cross-lingual Language Model Pre-training via ELECTRA

ACL 2022long

In this paper, we introduce ELECTRA-style tasks to cross-lingual language model pre-training. Specifically, we present two pre-training tasks, namely multilingual replaced token detection, and translation replaced token detection. Besides, we pretrain the model, named as XLM-E, on both multilingual…

2021

COCO-LM: Correcting and Contrasting Text Sequences for Language Model Pretraining

NeurIPS 2021poster

We present a self-supervised learning framework, COCO-LM, that pretrains Language Models by COrrecting and COntrasting corrupted text sequences. Following ELECTRA-style pretraining, COCO-LM employs an auxiliary language model to corrupt text sequences, upon which it constructs two new tasks for pret…

2021

Language Scaling for Universal Suggested Replies Model

NAACL 2021industry

We consider the problem of scaling automated suggested replies for a commercial email application to multiple languages. Faced with increased compute requirements and low language resources for language expansion, we build a single universal model for improving the quality and reducing run-time cost…

Cited by 2SourcePDFScholar
2018

Embedding Logical Queries on Knowledge Graphs

NeurIPS 2018poster

Learning low-dimensional embeddings of knowledge graphs is a powerful approach used to predict unobserved or missing edges between entities. However, an open challenge in this area is developing techniques that can go beyond simple edge prediction and handle more complex logical queries, which might…

2017

Inferring Generative Model Structure with Static Analysis

NeurIPS 2017poster

Obtaining enough labeled data to robustly train complex discriminative models is a major bottleneck in the machine learning pipeline. A popular solution is combining multiple sources of weak supervision using generative models. The structure of these models affects the quality of the training labels…

Cited by 69SourcePDFScholar