← Search

Wei-Po Wang

2 accepted papers

2025

Fundamental Limits of Prompt Tuning Transformers: Universality, Capacity and Efficiency

ICLR 2025poster

We investigate the statistical and computational limits of prompt tuning for transformer-based foundation models. Our key contributions are that prompt tuning on *single-head* transformers with only a *single* self-attention layer: (i) is universal, and (ii) supports efficient (even almost-linear…

Cited by 14SourcePDFScholar
2024

Outlier-Efficient Hopfield Layers for Large Transformer-Based Models

ICML 2024poster

We introduce an Outlier-Efficient Modern Hopfield Model (termed `OutEffHop`) and use it to address the outlier inefficiency problem of training gigantic transformer-based models. Our main contribution is a novel associative memory model facilitating _outlier-efficient_ associative memory retrievals.…