← Search

Federico Danieli

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

Normalized Rewards for Preference Optimization

ICML 2026poster

Direct Alignment Algorithms (DAAs) such as DPO have become a common way to post-train and align LLMs with human preferences. However, DAAs have been observed to over-optimize their implicit reward model and decrease the likelihood of preferred responses. This results in a decrease in the total likel…

Cited by 0SourceScholar
2025

Revisiting Uncertainty Quantification Evaluation in Language Models: Spurious Interactions with Response Length Bias Results

ACL 2025short

Uncertainty Quantification (UQ) in Language Models (LMs) is key to improving their safety and reliability. Evaluations often use metrics like AUROC to assess how well UQ methods (e.g., negative sequence probabilities) correlate with task correctness functions (e.g., ROUGE-L). We show that mutual bia…

2025

Theory, Analysis, and Best Practices for Sigmoid Self-Attention

ICLR 2025poster

Attention is a key part of the transformer architecture. It is a sequence-to-sequence mapping that transforms each sequence element into a weighted sum of values. The weights are typically obtained as the softmax of dot products between keys and queries. Recent work has explored alternatives to soft…

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