← Search

Sehoon Kim

16 accepted papers

2025

Multipole Attention for Efficient Long Context Reasoning

NeurIPS 2025poster

Large Reasoning Models (LRMs) have shown promising accuracy improvements on complex problem-solving tasks. While these models have attained high accuracy by leveraging additional computation at test time, they need to generate long chain-of-thought reasoning in order to think before answering, which…

Cited by 0SourceScholar
2025

Plan-and-Act: Improving Planning of Agents for Long-Horizon Tasks

ICML 2025poster

Large language models (LLMs) have shown remarkable advancements in enabling language agents to tackle simple tasks. However, applying them for complex, multi-step, long-horizon tasks remains a challenge. Recent work have found success by separating high-level planning from low-level execution, which…

Cited by 0SourcePDFScholar
2025

QuantSpec: Self-Speculative Decoding with Hierarchical Quantized KV Cache

ICML 2025poster

Large Language Models (LLMs) are increasingly being deployed on edge devices for long-context settings, creating a growing need for fast and efficient long-context inference. In these scenarios, the Key-Value (KV) cache is the primary bottleneck in terms of both GPU memory and latency, as the full K…

Cited by 0SourcePDFScholar
2025

Squeezed Attention: Accelerating Long Context Length LLM Inference

ACL 2025long

Emerging Large Language Model (LLM) applications require long input context in order to perform complex tasks like document analysis and code generation.For these long context length applications, the length of the input prompt poses a significant challenge in terms of inference efficiency since the…

2024

An LLM Compiler for Parallel Function Calling

ICML 2024poster

The reasoning capabilities of the recent LLMs enable them to execute external function calls to overcome their inherent limitations, such as knowledge cutoffs, poor arithmetic skills, or lack of access to private data. This development has allowed LLMs to select and coordinate multiple functions bas…

2024

KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantization

NeurIPS 2024poster

LLMs are seeing growing use for applications which require large context windows, and with these large context windows KV cache activations surface as the dominant contributor to memory consumption during inference. Quantization is a promising approach for compressing KV cache activations; however,…

2024

LLM2LLM: Boosting LLMs with Novel Iterative Data Enhancement

ACL 2024findings

Pretrained large language models (LLMs) are currently state-of-the-art for solving the vast majority of natural language processing tasks. While many real-world applications still require fine-tuning to reach satisfactory levels of performance, many of them are in the low-data regime, making fine-tu…

2024

SqueezeLLM: Dense-and-Sparse Quantization

ICML 2024poster

Generative Large Language Models (LLMs) have demonstrated remarkable results for a wide range of tasks. However, deploying these models for inference has been a significant challenge due to their unprecedented resource requirements. This has forced existing deployment frameworks to use multi-GPU inf…

2024

TinyAgent: Function Calling at the Edge

EMNLP 2024system demonstrations

Recent large language models (LLMs) have enabled the development of advanced agentic systems that can integrate various tools and APIs to fulfill user queries through function calling. However, the deployment of these LLMs on the edge has not been explored since they typically require cloud-based in…

2023

Speculative Decoding with Big Little Decoder

NeurIPS 2023poster

The recent emergence of Large Language Models based on the Transformer architecture has enabled dramatic advancements in the field of Natural Language Processing. However, these models have long inference latency, which limits their deployment and makes them prohibitively expensive for various real-…

2022

A Fast Post-Training Pruning Framework for Transformers

NeurIPS 2022accept

Pruning is an effective way to reduce the huge inference cost of Transformer models. However, prior work on pruning Transformers requires retraining the models. This can add high training cost and high complexity to model deployment, making it difficult to use in many practical situations. To addres…

2022

BigColor: Colorization Using a Generative Color Prior for Natural Images

ECCV 2022poster

"For realistic and vivid colorization, generative priors have recently been exploited. However, such generative priors often fail for in-the-wild complex images due to their limited representation space. In this paper, we propose BigColor, a novel colorization approach that provides vivid colorizati…

2022

Integer-Only Zero-Shot Quantization for Efficient Speech Recognition

ICASSP 2022accepted

End-to-end neural network models achieve improved performance on various automatic speech recognition (ASR) tasks. However, these models perform poorly on edge hardware due to large memory and computation requirements. While quantizing model weights and/or activations to low-precision can be a promi…

Cited by 0SourceScholar
2022

Squeezeformer: An Efficient Transformer for Automatic Speech Recognition

NeurIPS 2022accept

The recently proposed Conformer model has become the de facto backbone model for various downstream speech tasks based on its hybrid attention-convolution architecture that captures both local and global features. However, through a series of systematic studies, we find that the Conformer architectu…

2021

I-BERT: Integer-only BERT Quantization

ICML 2021oral

Transformer based models, like BERT and RoBERTa, have achieved state-of-the-art results in many Natural Language Processing tasks. However, their memory footprint, inference latency, and power consumption are prohibitive efficient inference at the edge, and even at the data center. While quantizatio…

2021

Terra: Imperative-Symbolic Co-Execution of Imperative Deep Learning Programs

NeurIPS 2021poster

Imperative programming allows users to implement their deep neural networks (DNNs) easily and has become an essential part of recent deep learning (DL) frameworks. Recently, several systems have been proposed to combine the usability of imperative programming with the optimized performance of symbol…

Cited by 7SourcePDFScholar