← Search

Itai Gat

23 accepted papers

2025

Accelerated Sampling from Masked Diffusion Models via Entropy Bounded Unmasking

NeurIPS 2025poster

Recent masked diffusion models (MDMs) have shown competitive performance compared to autoregressive models (ARMs) for language modeling. While most literature has focused on performance enhancing sampling procedures, efficient sampling from MDMs has been scarcely explored. We make the observation th…

Cited by 0SourceScholar
2025

Edit Flows: Variable Length Discrete Flow Matching with Sequence-Level Edit Operations

NeurIPS 2025poster

Autoregressive generative models naturally generate variable-length sequences, while non-autoregressive models struggle, often imposing rigid, token-wise structures. We propose Edit Flows, a non-autoregressive model that overcomes these limitations by defining a discrete flow over sequences through…

Cited by 0SourceScholar
2025

Exact Byte-Level Probabilities from Tokenized Language Models for FIM-Tasks and Model Ensembles

ICLR 2025poster

Tokenization is associated with many poorly understood shortcomings in language models (LMs), yet remains an important component for long sequence scaling purposes. This work studies how tokenization impacts model performance by analyzing and comparing the stochastic behavior of tokenized models w…

2025

Flow Matching with General Discrete Paths: A Kinetic-Optimal Perspective

ICLR 2025oral

The design space of discrete-space diffusion or flow generative models are significantly less well-understood than their continuous-space counterparts, with many works focusing only on a simple masked construction. In this work, we aim to take a holistic approach to the construction of discrete gene…

Cited by 4SourcePDFScholar
2025

Generator Matching: Generative modeling with arbitrary Markov processes

ICLR 2025oral

We introduce Generator Matching, a modality-agnostic framework for generative modeling using arbitrary Markov processes. Generators characterize the infinitesimal evolution of a Markov process, which we leverage for generative modeling in a similar vein to flow matching: we construct conditional gen…

Cited by 0SourcePDFScholar
2024

D-Flow: Differentiating through Flows for Controlled Generation

ICML 2024poster

Taming the generation outcome of state of the art Diffusion and Flow-Matching (FM) models without having to re-train a task-specific model unlocks a powerful tool for solving inverse problems, conditional generation, and controlled generation in general. In this work we introduce *D-Flow*, a simple…

Cited by 28SourcePDFScholar
2024

Diverse and Aligned Audio-to-Video Generation via Text-to-Video Model Adaptation

AAAI 2024technical

We consider the task of generating diverse and realistic videos guided by natural audio samples from a wide variety of semantic classes. For this task, the videos are required to be aligned both globally and temporally with the input audio: globally, the input audio is semantically associated with t…

2024

Layer Collaboration in the Forward-Forward Algorithm

AAAI 2024technical

Backpropagation, which uses the chain rule, is the de-facto standard algorithm for optimizing neural networks nowadays. Recently, Hinton (2022) proposed the forward-forward algorithm, a promising alternative that optimizes neural nets layer-by-layer, without propagating gradients throughout the netw…

Cited by 9SourcePDFScholar
2024

Masked Audio Generation using a Single Non-Autoregressive Transformer

ICLR 2024poster

We introduce MAGNeT, a masked generative sequence modeling method that operates directly over several streams of audio tokens. Unlike prior work, MAGNeT is comprised of a single-stage, non-autoregressive transformer. During training, we predict spans of masked tokens obtained from a masking schedule…

Cited by 50SourcePDFScholar
2023

Simple and Controllable Music Generation

NeurIPS 2023poster

We tackle the task of conditional music generation. We introduce MusicGen, a single Language Model (LM) that operates over several streams of compressed discrete music representation, i.e., tokens. Unlike prior work, MusicGen is comprised of a single-stage transformer LM together with efficient toke…

2023

Textually Pretrained Speech Language Models

NeurIPS 2023poster

Speech language models (SpeechLMs) process and generate acoustic data only, without textual supervision. In this work, we propose TWIST, a method for training SpeechLMs using a warm-start from a pretrained textual language models. We show using both automatic and human evaluations that TWIST outperf…

2022

A Functional Information Perspective on Model Interpretation

ICML 2022spotlight

Contemporary predictive models are hard to interpret as their deep nets exploit numerous complex relations between input elements. This work suggests a theoretical framework for model interpretability by measuring the contribution of relevant features to the functional entropy of the network with re…

2022

Speaker Normalization for Self-Supervised Speech Emotion Recognition

ICASSP 2022accepted

Large speech emotion recognition datasets are hard to obtain, and small datasets may contain biases. Deep-net-based classifiers, in turn, are prone to exploit those biases and find shortcuts such as speaker characteristics. These shortcuts usually harm a model’s ability to generalize. To address thi…

Cited by 0SourceScholar
2022

Speech Emotion Recognition Using Self-Supervised Features

ICASSP 2022accepted

Self-supervised pre-trained features have consistently delivered state-of-art results in the field of natural language processing (NLP); however, their merits in the field of speech emotion recognition (SER) still need further investigation. In this paper we introduce a modular End-to-End (E2E) SER…

Cited by 0SourceScholar
2022

Towards A Common Speech Analysis Engine

ICASSP 2022accepted

Recent innovations in self-supervised representation learning have led to remarkable advances in natural language processing. That said, in the speech processing domain, self-supervised representation learning-based systems are not yet considered state-of-the-art.We propose leveraging recent advance…

Cited by 0SourceScholar
2021

Are VQA Systems RAD? Measuring Robustness to Augmented Data with Focused Interventions

ACL 2021short

Deep learning algorithms have shown promising results in visual question answering (VQA) tasks, but a more careful look reveals that they often do not understand the rich signal they are being fed with. To understand and better measure the generalization capabilities of VQA systems, we look at their…

2021

Perceptual Score: What Data Modalities Does Your Model Perceive?

NeurIPS 2021poster

Machine learning advances in the last decade have relied significantly on large-scale datasets that continue to grow in size. Increasingly, those datasets also contain different data modalities. However, large multi-modal datasets are hard to annotate, and annotations may contain biases that we are…

2020

Removing Bias in Multi-modal Classifiers: Regularization by Maximizing Functional Entropies

NeurIPS 2020poster

Many recent datasets contain a variety of different data modalities, for instance, image, question, and answer data in visual question answering (VQA). When training deep net classifiers on those multi-modal datasets, the modalities get exploited at different scales, i.e., some modalities can more e…