← Search

Jeongin Bae

5 accepted papers

2026

Affine-Scaled Attention: Towards Flexible and Stable Transformer Attention

ICML 2026poster

Transformer attention is typically implemented using softmax normalization, which enforces attention weights with unit sum normalization. While effective in many settings, this constraint can limit flexibility in controlling attention magnitudes and may contribute to overly concentrated or unstable …

Cited by 0SourceScholar
2026

AnyBCQ: Hardware Efficient Flexible Binary-Coded Quantization for Multi-Precision LLMs

ICLR 2026poster

The deployment of large language models (LLMs) is increasingly constrained by memory and latency bottlenecks, motivating the need for quantization techniques that flexibly balance accuracy and efficiency. Recent work has introduced multi-precision models, which enable inference at multiple precisio…

Cited by 0SourcecodeScholar
2025

CodeGEMM: A Codebook-Centric Approach to Efficient GEMM in Quantized LLMs

NeurIPS 2025poster

Weight-only quantization is widely used to mitigate the memory-bound nature of LLM inference. Codebook-based methods extend this trend by achieving strong accuracy in the extremely low-bit regime (e.g., 2-bit). However, current kernels rely on dequantization, which repeatedly fetches centroids and r…

Cited by 0SourceScholar
2025

Unifying Uniform and Binary-coding Quantization for Accurate Compression of Large Language Models

ACL 2025long

How can we quantize large language models while preserving accuracy? Quantization is essential for deploying large language models (LLMs) efficiently. Binary-coding quantization (BCQ) and uniform quantization (UQ) are promising quantization schemes that have strong expressiveness and optimizability,…

2022

AlphaTuning: Quantization-Aware Parameter-Efficient Adaptation of Large-Scale Pre-Trained Language Models

EMNLP 2022finding

There are growing interests in adapting large-scale language models using parameter-efficient fine-tuning methods. However, accelerating the model itself and achieving better inference efficiency through model compression has not been thoroughly explored yet.Model compression could provide the benef…

Cited by 42SourcePDFScholar