← Search

Duy Khuong Nguyen

2 accepted papers

2023

A Probabilistic Framework for Pruning Transformers Via a Finite Admixture of Keys

ICASSP 2023accepted

Pairwise dot product-based self-attention is key to the success of transformers which achieve state-of-the-art performance across a variety of applications in language and vision, but are costly to compute. It has been shown that most attention scores and keys in transformers are redundant and can b…

Cited by 0SourceScholar
2022

Improving Transformers with Probabilistic Attention Keys

ICML 2022spotlight

Multi-head attention is a driving force behind state-of-the-art transformers, which achieve remarkable performance across a variety of natural language processing (NLP) and computer vision tasks. It has been observed that for many applications, those attention heads learn redundant embedding, and mo…