← Search

Armen Aghajanyan

17 accepted papers

2025

When Worse is Better: Navigating the Compression Generation Trade-off In Visual Tokenization

NeurIPS 2025spotlight

Current image generation methods are based on a two-stage training approach. In stage 1, an auto-encoder is trained to compress an image into a latent space; in stage 2, a generative model is trained to learn a distribution over that latent space. This reveals a fundamental trade-off, do we compress…

Cited by 0SourceScholar
2024

Jointly Training Large Autoregressive Multimodal Models

ICLR 2024poster

In recent years, advances in the large-scale pretraining of language and text-to-image models have revolutionized the field of machine learning. Yet, integrating these two modalities into a single, robust model capable of generating seamless multimodal outputs remains a significant challenge. To add…

Cited by 33SourcePDFScholar
2023

InCoder: A Generative Model for Code Infilling and Synthesis

ICLR 2023top-25%

Code is seldom written in a single left-to-right pass and is instead repeatedly edited and refined. We introduce InCoder, a unified generative model that can perform program synthesis (via left-to-right generation) as well as editing (via masking and infilling). InCoder is trained to generate code f…

2023

MEGABYTE: Predicting Million-byte Sequences with Multiscale Transformers

NeurIPS 2023poster

Autoregressive transformers are spectacular models for short sequences but scale poorly to long sequences such as high-resolution images, podcasts, code, or books. We proposed Megabyte, a multi-scale decoder architecture that enables end-to-end differentiable modeling of sequences of over one millio…

Cited by 91SourcePDFScholar
2023

Retrieval-Augmented Multimodal Language Modeling

ICML 2023poster

Recent multimodal models such as DALL-E and CM3 have achieved remarkable progress in text-to-image and image-to-text generation. However, these models store all their knowledge (e.g., the appearance of the Eiffel Tower) in the model parameters, requiring increasingly larger models and training data…

Cited by 159SourcePDFScholar
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

CCQA: A New Web-Scale Question Answering Dataset for Model Pre-Training

NAACL 2022findings

We propose a novel open-domain question-answering dataset based on the Common Crawl project. With a previously unseen number of around 130 million multilingual question-answer pairs (including about 60 million English data-points), we use our large-scale, natural, diverse and high-quality corpus to…

2022

HTLM: Hyper-Text Pre-Training and Prompting of Language Models

ICLR 2022poster

We introduce HTLM, a hyper-text language model trained on a large-scale web crawl. Modeling hyper-text has a number of advantages: (1) it is easily gathered at scale, (2) it provides rich document-level and end-task-adjacent supervision (e.g. 'class' and 'id' attributes often encode document categor…

Cited by 84SourcePDFScholar
2022

Improving Passage Retrieval with Zero-Shot Question Generation

EMNLP 2022main

We propose a simple and effective re-ranking method for improving passage retrieval in open question answering. The re-ranker re-scores retrieved passages with a zero-shot question generation model, which uses a pre-trained language model to compute the probability of the input question conditioned…

2022

Memorization Without Overfitting: Analyzing the Training Dynamics of Large Language Models

NeurIPS 2022accept

Despite their wide adoption, the underlying training and memorization dynamics of very large language models is not well understood. We empirically study exact memorization in causal and masked language modeling, across model sizes and throughout the training process. We measure the effects of datas…

Cited by 278SourcePDFScholar
2021

Better Fine-Tuning by Reducing Representational Collapse

ICLR 2021poster

Although widely adopted, existing approaches for fine-tuning pre-trained language models have been shown to be unstable across hyper-parameter settings, motivating recent work on trust region methods. In this paper, we present a simplified and efficient method rooted in trust region theory that repl…

2021

Intrinsic Dimensionality Explains the Effectiveness of Language Model Fine-Tuning

ACL 2021long

Although pretrained language models can be fine-tuned to produce state-of-the-art results for a very wide range of language understanding tasks, the dynamics of this process are not well understood, especially in the low data regime. Why can we use relatively vanilla gradient descent algorithms (e.g…

2021

Muppet: Massive Multi-task Representations with Pre-Finetuning

EMNLP 2021main

We propose pre-finetuning, an additional large-scale learning stage between language model pre-training and fine-tuning. Pre-finetuning is massively multi-task learning (around 50 datasets, over 4.8 million total labeled examples), and is designed to encourage learning of representations that genera…

2021

Non-Autoregressive Semantic Parsing for Compositional Task-Oriented Dialog

NAACL 2021long

Semantic parsing using sequence-to-sequence models allows parsing of deeper representations compared to traditional word tagging based models. In spite of these advantages, widespread adoption of these models for real-time conversational use cases has been stymied by higher compute requirements and…

2021

VideoCLIP: Contrastive Pre-training for Zero-shot Video-Text Understanding

EMNLP 2021main

We present VideoCLIP, a contrastive approach to pre-train a unified model for zero-shot video and text understanding, without using any labels on downstream tasks. VideoCLIP trains a transformer for video and text by contrasting temporally overlapping positive video-text pairs with hard negatives fr…

2020

Pre-training via Paraphrasing

NeurIPS 2020poster

We introduce MARGE, a pre-trained sequence-to-sequence model learned with an unsupervised multi-lingual multi-document paraphrasing objective. MARGE provides an alternative to the dominant masked language modeling paradigm, where we self-supervise the \emph{reconstruction} of target text by \emph{re…

Cited by 171SourcePDFScholar