← Search

Dongsoo Lee

21 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
2026

ICaRus: Identical Cache Reuse for Efficient Multi-Model Inference

ICLR 2026poster

Multi model inference, where multiple task-specialized models collaborate to solve complex real-world problems, has recently emerged as a prominent paradigm, particularly in the development of agentic AI systems. However, in such scenarios, each model must maintain its own Key-Value (KV) cache for t…

Cited by 0SourceScholar
2026

SelfJudge: Faster Speculative Decoding via Self-Supervised Judge Verification

ICML 2026poster

Speculative decoding accelerates LLM inference by verifying candidate tokens from a draft model against a larger target model. Recent "judge'' decoding boosts this process by relaxing verification criteria by accepting draft tokens that may exhibit minor discrepancies from target model output, but e…

Cited by 0SourceScholar
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

LRQ: Optimizing Post-Training Quantization for Large Language Models by Learning Low-Rank Weight-Scaling Matrices

NAACL 2025long

With the commercialization of large language models (LLMs), weight-activation quantization has emerged to compress and accelerate LLMs, achieving high throughput while reducing inference costs. However, existing post-training quantization (PTQ) techniques for quantizing weights and activations of LL…

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

2024

DropBP: Accelerating Fine-Tuning of Large Language Models by Dropping Backward Propagation

NeurIPS 2024poster

Large language models (LLMs) have achieved significant success across various domains. However, training these LLMs typically involves substantial memory and computational costs during both forward and backward propagation. While parameter-efficient fine-tuning (PEFT) considerably reduces the traini…

2024

LUT-GEMM: Quantized Matrix Multiplication based on LUTs for Efficient Inference in Large-Scale Generative Language Models

ICLR 2024poster

Recent advances in self-supervised learning and the Transformer architecture have significantly improved natural language processing (NLP), achieving remarkably low perplexity. However, the growing size of NLP models introduces a memory wall problem during the generation phase. To mitigate this issu…

2024

Rethinking Channel Dimensions to Isolate Outliers for Low-bit Weight Quantization of Large Language Models

ICLR 2024poster

Large Language Models (LLMs) have recently demonstrated a remarkable success across various tasks. However, efficiently serving LLMs has been a challenge due to its large memory bottleneck, specifically in small batch inference settings (e.g. mobile devices). Weight-only quantization can be a promis…

2023

FlexRound: Learnable Rounding based on Element-wise Division for Post-Training Quantization

ICML 2023poster

Post-training quantization (PTQ) has been gaining popularity for the deployment of deep neural networks on resource-limited devices since unlike quantization-aware training, neither a full training dataset nor end-to-end training is required at all. As PTQ schemes based on reconstructing each layer…

Cited by 35SourcePDFScholar
2023

Information Geometry of the Retinal Representation Manifold

NeurIPS 2023poster

The ability for the brain to discriminate among visual stimuli is constrained by their retinal representations. Previous studies of visual discriminability have been limited to either low-dimensional artificial stimuli or pure theoretical considerations without a realistic encoding model. Here we pr…

Cited by 6SourcePDFScholar
2023

Memory-Efficient Fine-Tuning of Compressed Large Language Models via sub-4-bit Integer Quantization

NeurIPS 2023poster

Large language models (LLMs) face the challenges in fine-tuning and deployment due to their high memory demands and computational costs. While parameter-efficient fine-tuning (PEFT) methods aim to reduce the memory usage of the optimizer state during fine-tuning, the inherent size of pre-trained LLM…

Cited by 110SourcePDFScholar
2023

Winning Both the Accuracy of Floating Point Activation and the Simplicity of Integer Arithmetic

ICLR 2023poster

Even though floating point (FP) numbers have been adopted as a de facto standard data format for deep learning computing, the complexity of FP arithmetic impedes a broader deployment of Deep Neural Networks (DNNs). Recent works such as quantization have attempted to replace the FP matrix multiplicat…

Cited by 6SourcePDFScholar
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
2022

Encoding Weights of Irregular Sparsity for Fixed-to-Fixed Model Compression

ICLR 2022poster

Even though fine-grained pruning techniques achieve a high compression ratio, conventional sparsity representations (such as CSR) associated with irregular sparsity degrade parallelism significantly. Practical pruning methods, thus, usually lower pruning rates (by structured pruning) to improve para…

Cited by 5SourcePDFScholar
2022

Maximum Likelihood Training of Implicit Nonlinear Diffusion Model

NeurIPS 2022accept

Whereas diverse variations of diffusion models exist, extending the linear diffusion into a nonlinear diffusion process is investigated by very few works. The nonlinearity effect has been hardly understood, but intuitively, there would be promising diffusion patterns to efficiently train the generat…

2020

FleXOR: Trainable Fractional Quantization

NeurIPS 2020poster

Quantization based on the binary codes is gaining attention because each quantized bit can be directly utilized for computations without dequantization using look-up tables. Previous attempts, however, only allow for integer numbers of quantization bits, which ends up restricting the search space fo…

Cited by 15SourcePDFScholar
2020

Structured Compression by Weight Encryption for Unstructured Pruning and Quantization

CVPR 2020poster

Model compression techniques, such as pruning and quantization, are becoming increasingly important to reduce the memory footprints and the amount of computations. Despite model size reduction, achieving performance enhancement on devices is, however, still challenging mainly due to the irregular re…

Cited by 59PDFScholar
2019

Double Viterbi: Weight Encoding for High Compression Ratio and Fast On-Chip Reconstruction for Deep Neural Network

ICLR 2019poster

Weight pruning has been introduced as an efficient model compression technique. Even though pruning removes significant amount of weights in a network, memory requirement reduction was limited since conventional sparse matrix formats require significant amount of memory to store index-related inform…

Cited by 14SourcePDFScholar
2018

Viterbi-based Pruning for Sparse Matrix with Fixed and High Index Compression Ratio

ICLR 2018poster

Weight pruning has proven to be an effective method in reducing the model size and computation cost while not sacrificing the model accuracy. Conventional sparse matrix formats, however, involve irregular index structures with large storage requirement and sequential reconstruction process, resultin…

Cited by 24SourcePDFScholar