← Search

Heejun Lee

6 accepted papers

2025

A Training-Free Sub-quadratic Cost Transformer Model Serving Framework with Hierarchically Pruned Attention

ICLR 2025poster

In modern large language models (LLMs), increasing the context length is crucial for improving comprehension and coherence in long-context, multi-modal, and retrieval-augmented language generation. While many recent transformer models attempt to extend their context length over a million tokens, th…

Cited by 9SourcePDFScholar
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…

2022

K-MHaS: A Multi-label Hate Speech Detection Dataset in Korean Online News Comment

COLING 2022main

Online hate speech detection has become an important issue due to the growth of online content, but resources in languages other than English are extremely limited. We introduce K-MHaS, a new multi-label dataset for hate speech detection that effectively handles Korean language patterns. The dataset…