← Search

Wonjun Kang

9 accepted papers

2026

Draft-based Approximate Inference for LLMs

ICLR 2026poster

Optimizing inference for long-context large language models (LLMs) is increasingly important due to the quadratic compute and linear memory cost of Transformers. Existing approximate inference methods, including key-value (KV) cache dropping, sparse attention, and prompt compression, typically rely…

Cited by 0SourcecodeScholar
2026

LoSA: Locality Aware Sparse Attention in Diffusion Language Models

ICML 2026poster

Block-wise diffusion language models (DLMs) generate multiple tokens in parallel, offering a promising alternative to autoregressive decoding. However, their inference efficiency remains bottlenecked by memory-bound attention in long-context scenarios. Naïve sparse attention is ineffective for DLMs …

Cited by 0SourceScholar
2026

ParallelBench: Understanding the Trade-offs of Parallel Decoding in Diffusion LLMs

ICLR 2026poster

While most autoregressive LLMs are constrained to one-by-one decoding, diffusion LLMs (dLLMs) have attracted growing interest for their potential to dramatically accelerate inference through parallel decoding. Despite this promise, the conditional independence assumption in dLLMs causes parallel dec…

Cited by 0SourcecodeScholar
2026

ReJump: A Tree-Jump Representation for Analyzing and Improving LLM Reasoning

ICML 2026poster

Large Reasoning Models (LRMs) are Large Language Models (LLMs) explicitly trained to generate long-form Chain-of-Thoughts (CoTs), achieving impressive success on challenging tasks like math and programming. However, their underlying reasoning "algorithms" remain poorly understood. To investigate thi…

Cited by 0SourceScholar
2025

Parameter-Efficient Fine-Tuning of State Space Models

ICML 2025poster

Deep State Space Models (SSMs), such as Mamba (Gu & Dao, 2024), have become powerful tools for language modeling, offering high performance and linear scalability with sequence length. However, the application of parameter-efficient fine-tuning (PEFT) methods to SSM-based models remains largely unde…

2025

State-offset Tuning: State-based Parameter-Efficient Fine-Tuning for State Space Models

ACL 2025short

State Space Models (SSMs) have emerged as efficient alternatives to Transformers, mitigating their quadratic computational cost. However, the application of Parameter-Efficient Fine-Tuning (PEFT) methods to SSMs remains largely unexplored. In particular, prompt-based methods like Prompt Tuning and P…

2025

TabFlex: Scaling Tabular Learning to Millions with Linear Attention

ICML 2025spotlight

Leveraging the in-context learning (ICL) capability of Large Language Models (LLMs) for tabular classification has gained significant attention for its training-free adaptability across diverse datasets. Recent advancements, like TabPFN, excel in small-scale tabular datasets but struggle to scale fo…

Cited by 2SourcePDFScholar
2025

VersaPRM: Multi-Domain Process Reward Model via Synthetic Reasoning Data

ICML 2025oral

Process Reward Models (PRMs) have proven effective at enhancing mathematical reasoning for Large Language Models (LLMs) by leveraging increased inference-time computation. However, they are predominantly trained on mathematical data and their generalizability to non-mathematical domains has not been…

Cited by 4SourcePDFScholar
2024

Eta Inversion: Designing an Optimal Eta Function for Diffusion-based Real Image Editing

ECCV 2024poster

"Diffusion models have achieved remarkable success in the domain of text-guided image generation and, more recently, in text-guided image editing. A commonly adopted strategy for editing real images involves inverting the diffusion process to obtain a noisy representation of the original image, whic…