Sunghyun Baek
Author directory2026
IWP: Token Pruning as Implicit Weight Pruning in Large Vision Language Models
Dong-Jae Lee | Sunghyun Baek | Junmo Kim
Findings of the Association for Computational Linguistics: EMNLP 2026
Dong-Jae Lee | Sunghyun Baek | Junmo Kim
Findings of the Association for Computational Linguistics: EMNLP 2026
Large Vision–Language Models (LVLMs) show impressive performance across image and video understanding tasks, yet their computational cost grows rapidly with the number of visual tokens. Existing token pruning methods address this by reducing the number of visual tokens, but overlook the internal mechanism of attention. In this paper, we propose a novel training-free token pruning framework grounded in the dual-form perspective of attention. We reformulate attention as an implicit linear layer whose weight matrix is the sum of rank-1 outer products, each generated by a single token’s key–value pair. Token pruning thus reduces to identifying an optimal subset of these rank-1 updates that retains the essential information of the original dual weight matrix. Extending this perspective to standard softmax attention in LVLMs, we derive a novel importance metric quantifying both a token’s information magnitude and information duplication. To efficiently select the subset with the proposed metric, we introduce Progressive Chunked Maximal Marginal Relevance. Extensive experiments demonstrate that our method achieves a better trade-off between performance and efficiency, while providing another perspective on existing pruning approaches.
\textlangle /think \textrangle Doesn’t Stop Reasoning: Analysis of Spurious CoT Termination
Seunghee Koh | Sungjae Choi | Minchan Kwon | Sunghyun Baek | Junmo Kim
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Seunghee Koh | Sungjae Choi | Minchan Kwon | Sunghyun Baek | Junmo Kim
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Chain-of-thought (CoT) reasoning improves large reasoning models (LRMs) on complex tasks but often produces long, redundant traces. Recent training-free early-exit methods shorten these traces by choosing an intermediate point to stop reasoning. We study one such strategy that injects an end-of-think token (EoT, </think>) at this point to trigger the reasoning-to-answering transition, and find that the injected EoT does not always induce a clean answering phase. Answering-phase generation can continue before the model regenerates another EoT, with the span preceding this regenerated EoT scaling with the reasoning tokens saved by early exit and exhibiting continued reasoning behavior. We call this spurious CoT termination, where reasoning-like generation continues into the answering phase. We hypothesize that insufficient attention to the injected EoT contributes to spurious CoT termination and probe this hypothesis with Exit-token Attention Biasing (EAB). Across four LRMs, five benchmarks, and two early-exit methods, increasing attention to the injected EoT reduces spurious CoT termination and answering-phase length. These results reveal a limitation of controlling LRMs by externally matching their explicit think-block format. Inserting the EoT token conforms to this format but does not by itself guarantee the intended reasoning-to-answering transition. Our code is available at https://github.com/Seunghee-Koh/Spurious-CoT-Termination.
Direct Diversity Optimization for Diverse Successful Trajectories in Preference Post-Training
Junwon Ko | Dong-Jae Lee | Minchan Kwon | Sunghyun Baek | Junmo Kim
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Junwon Ko | Dong-Jae Lee | Minchan Kwon | Sunghyun Baek | Junmo Kim
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
LLM agents for sequential decision tasks are often post-trained with trajectory-level outcome labels, but such labels provide little supervision for preserving multiple successful branches from the same decision state. We study this problem as successful strategy coverage: how broadly a model realizes distinct successful strategies under a fixed rollout budget. We present Direct Diversity Optimization (DDO), an offline post-training method that combines Divergence-Tree Collection (DTC) with the Reference-Relative Target-Odds Objective (RTO). DTC constructs state-aligned branch sets rooted at shared decision states, and RTO trains the model to match reference-relative targets over successful alternatives. DDO achieves the strongest task success and successful strategy coverage among the compared post-training methods across BabyAI, BabaIsAI, and WebShop. It also achieves the highest recovery rate after local action replacement and higher task success and coverage than successful-only imitation and decoding-time diversification controls.
Forget What Matters, Keep the Rest: Selective Unlearning of Informative Tokens
Seunghee Koh | Sunghyun Baek | Youngdong Kim | Junmo Kim
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Seunghee Koh | Sunghyun Baek | Youngdong Kim | Junmo Kim
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Unlearning in large language models (LLMs) has emerged as a promising safeguard against adversarial behaviors. When the forgetting loss is applied uniformly without considering token-level semantic importance, model utility can be unnecessarily degraded. Recent studies have explored token-wise loss regularizers that prioritize informative tokens, but largely rely on ground-truth confidence or external linguistic parsers, which limits their ability to capture contextual information or the model’s overall predictive state. Intuitively, function words like “the” primarily serve syntactic roles and are highly predictable with little ambiguity, but informative words admit multiple plausible alternatives with greater uncertainty. Based on this intuition, we propose Entropy-guided Token Weighting (ETW), a token-level unlearning regularizer that uses entropy of the predictive distribution as a proxy for token informativeness. We demonstrate that informative tokens tend to have higher entropy, whereas structural tokens tend to have lower entropy. This behavior enables ETW to achieve more effective unlearning while better preserving model utility than existing token-level approaches.