← Search

Jeffrey Willette

5 accepted papers

2025

Delta Attention: Fast and Accurate Sparse Attention Inference by Delta Correction

NeurIPS 2025poster

The attention mechanism of a transformer has a quadratic complexity, leading to high inference costs and latency for long sequences. However, attention matrices are mostly sparse, which implies that many entries may be omitted from computation for efficient inference. Sparse attention inference meth…

Cited by 0SourceScholar
2025

Training Free Exponential Context Extension via Cascading KV Cache

ICLR 2025poster

The transformer's context window is vital for tasks such as few-shot learning and conditional generation as it preserves previous tokens for active memory. However, as the context lengths increase, the computational costs grow quadratically, hindering the deployment of large language models (LLMs) i…

2024

SEA: Sparse Linear Attention with Estimated Attention Mask

ICLR 2024poster

The transformer architecture has driven breakthroughs in recent years on tasks which require modeling pairwise relationships between sequential elements, as is the case in natural language understanding. However, long seqeuences pose a problem due to the quadratic complexity of the attention operati…

2023

Scalable Set Encoding with Universal Mini-Batch Consistency and Unbiased Full Set Gradient Approximation

ICML 2023poster

Recent work on mini-batch consistency (MBC) for set functions has brought attention to the need for sequentially processing and aggregating chunks of a partitioned set while guaranteeing the same output for all partitions. However, existing constraints on MBC architectures lead to models with limite…

2022

MPViT: Multi-Path Vision Transformer for Dense Prediction

CVPR 2022poster

Dense computer vision tasks such as object detection and segmentation require effective multi-scale feature representation for detecting or classifying objects or regions with varying sizes. While Convolutional Neural Networks (CNNs) have been the dominant architectures for such tasks, recently intr…

Cited by 360PDFcodeScholar