← Search

Arun Babu

6 accepted papers

2025

Transfusion: Predict the Next Token and Diffuse Images with One Multi-Modal Model

ICLR 2025oral

We introduce Transfusion, a recipe for training a multi-modal model over discrete and continuous data. Transfusion combines the language modeling loss function (next token prediction) with diffusion to train a single transformer over mixed-modality sequences. We pretrain multiple Transfusion models…

Cited by 150SourcePDFScholar
2023

Efficient Self-supervised Learning with Contextualized Target Representations for Vision, Speech and Language

ICML 2023oral

Current self-supervised learning algorithms are often modality-specific and require large amounts of computational resources. To address these issues, we increase the training efficiency of data2vec, a learning objective that generalizes across several modalities. We do not encode masked tokens, use…

2023

Toward Joint Language Modeling for Speech Units and Text

EMNLP 2023long findings

Speech and text are two major forms of human language. The research community has been focusing on mapping speech to text or vice versa for many years. However, in the field of language modeling, very little effort has been made to model them jointly. In light of this, we explore joint language mode…

Cited by 0SourceScholar
2022

data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Language

ICML 2022oral

While the general idea of self-supervised learning is identical across modalities, the actual algorithms and objectives differ widely because they were developed with a single modality in mind. To get us closer to general self-supervised learning, we present data2vec, a framework that uses the same…

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

Span Pointer Networks for Non-Autoregressive Task-Oriented Semantic Parsing

EMNLP 2021finding

An effective recipe for building seq2seq, non-autoregressive, task-oriented parsers to map utterances to semantic frames proceeds in three steps: encoding an utterance x, predicting a frame’s length |y|, and decoding a |y|-sized frame with utterance and ontology tokens. Though empirically strong, th…

Cited by 25SourcePDFScholar