← Search

Xavier Suau

9 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
2026

GenCtrl -- A Formal Controllability Toolkit for Generative Models

ICLR 2026poster

As generative models become ubiquitous, there is a critical need for fine-grained control over the generation process. Yet, while controlled generation methods from prompting to fine-tuning proliferate, a fundamental question remains unanswered: are these models truly controllable in the first place…

Cited by 0SourceScholar
2025

Controlling Language and Diffusion Models by Transporting Activations

ICLR 2025spotlight

The increasing capabilities of large generative models and their ever more widespread deployment have raised concerns about their reliability, safety, and potential misuse. To address these issues, recent works have proposed to control model generation by steering model activations in order to effec…

2025

LinEAS: End-to-end Learning of Activation Steering with a Distributional Loss

NeurIPS 2025poster

The growing use of generative models in daily life calls for efficient mechanisms to control their generation, to e.g. produce safe content or provide users with tools to explore style changes. Ideally, such mechanisms should require low volume of unpaired data (\ie without explicit preference), and…

Cited by 0SourceScholar
2024

Whispering Experts: Neural Interventions for Toxicity Mitigation in Language Models

ICML 2024poster

An important issue with Large Language Models (LLMs) is their undesired ability to generate toxic language. In this work, we show that the neurons responsible for toxicity can be determined by their power to discriminate toxic sentences, and that toxic language can be mitigated by reducing their act…

Cited by 7SourcePDFScholar
2023

DUET: 2D Structured and Approximately Equivariant Representations

ICML 2023poster

Multiview Self-Supervised Learning (MSSL) is based on learning invariances with respect to a set of input transformations. However, invariance partially or totally removes transformation-related information from the representations, which might harm performance for specific downstream tasks that req…

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

How to Scale Your EMA

NeurIPS 2023spotlight

Preserving training dynamics across batch sizes is an important tool for practical machine learning as it enables the trade-off between batch size and wall-clock time. This trade-off is typically enabled by a scaling rule, for example, in stochastic gradient descent, one should scale the learning ra…

Cited by 30SourcePDFScholar
2023

The Role of Entropy and Reconstruction in Multi-View Self-Supervised Learning

ICML 2023poster

The mechanisms behind the success of multi-view self-supervised learning (MVSSL) are not yet fully understood. Contrastive MVSSL methods have been studied through the lens of InfoNCE, a lower bound of the Mutual Information (MI). However, the relation between other MVSSL methods and MI remains uncle…