← Search

Miguel Sarabia

7 accepted papers

2026

FlashRNN: Unlocking Parallel Training of Nonlinear RNNs for Large Language Models

ICLR 2026oral

Recurrent Neural Networks (RNNs) laid the foundation for sequence modeling, but their intrinsic sequential nature restricts parallel computation, creating a fundamental barrier to scaling. This has led to the dominance of parallelizable architectures like Transformers and, more recently, State Space…

Cited by 0SourcecodeScholar
2025

Understanding Input Selectivity in Mamba: Impact on Approximation Power, Memorization, and Associative Recall Capacity

ICML 2025poster

State-Space Models (SSMs), and particularly Mamba, have recently emerged as a promising alternative to Transformers. Mamba introduces input selectivity to its SSM layer (S6) and incorporates convolution and gating into its block definition. While these modifications do improve Mamba's performance ov…

Cited by 0SourcePDFScholar
2024

Can You Rely on Synthetic Labellers in Preference-Based Reinforcement Learning? It’s Complicated

AAAI 2024technical

Preference-based Reinforcement Learning (PbRL) enables non-experts to train Reinforcement Learning models using preference feedback. However, the effort required to collect preference labels from real humans means that PbRL research primarily relies on synthetic labellers. We validate the most commo…

Cited by 2SourcePDFScholar
2024

Learning Spatially-Aware Language and Audio Embeddings

NeurIPS 2024poster

Humans can picture a sound scene given an imprecise natural language description. For example, it is easy to imagine an acoustic environment given a phrase like "the lion roar came from right behind me!". For a machine to have the same degree of comprehension, the machine must know what a lion is (…

Cited by 1SourcePDFScholar
2023

DeepPCR: Parallelizing Sequential Operations in Neural Networks

NeurIPS 2023poster

Parallelization techniques have become ubiquitous for accelerating inference and training of deep neural networks. Despite this, several operations are still performed in a sequential manner. For instance, the forward and backward passes are executed layer-by-layer, and the output of diffusion model…

Cited by 1SourcePDFScholar
2023

On the Role of LIP Articulation in Visual Speech Perception

ICASSP 2023accepted

Generating realistic lip motion from audio to simulate speech production is critical for driving natural character animation. Previous research has shown that traditional metrics used to optimize and assess models for generating lip motion from speech are not a good indicator of subjective opinion o…

Cited by 0SourceScholar
2023

Sample-Efficient Preference-based Reinforcement Learning with Dynamics Aware Rewards

CoRL 2023poster

Preference-based reinforcement learning (PbRL) aligns a robot behavior with human preferences via a reward function learned from binary feedback over agent behaviors. We show that encoding environment dynamics in the reward function improves the sample efficiency of PbRL by an order of magnitude. In…

Cited by 7SourcecodeScholar