-
Contrastive Thinking Decoding: Steering Answer Generation in Reasoning Models
Taehyeon Kim, Youngsoo Jang, Hyunsoo Lee, Yu Jin Kim, and Moontae Lee
NeurIPS 2026
First author · Steering answer generation in reasoning models.
-
Efficiently Learning To Reason or Not to Reason: Root-token Policy Optimization for
Adaptive Thinking
Taehyeon Kim, Hyunsoo Lee, Youngsoo Jang†, and
Moontae Lee†
ACL 2026 · Oral Presentation
First author · Adaptive thinking at 2% of the training compute of the prior methods evaluated in the paper.
-
Multi-Drafter Speculative Decoding with Alignment Feedback
Taehyeon Kim*, Hojung Jung*, and Se-Young Yun
ACL 2026 · Findings
Co-first author · Speculative decoding with multiple drafters and alignment feedback.
-
Instructive Decoding: Instruction-Tuned Large Language Models are Self-Refiner from
Noisy Instructions
Taehyeon Kim*, Joonkee Kim*, Gihun Lee*,
and Se-Young Yun
ICLR 2024 · Spotlight
Co-first author · Instruction-guided decoding for noisy instructions.
-
Accelerating Blockwise Parallel Language Models with Draft Refinement
Taehyeon Kim, Ananda Theertha Suresh, Kishore Papineni, Michael Riley,
Sanjiv Kumar, and Adrian Benton
NeurIPS 2024
First author · Google Research collaboration. Draft refinement increased accepted token length by 5–25%, yielding over 3× speedup over autoregressive decoding on the evaluated 7B/13B LLMs.
-
Comparing Kullback-Leibler Divergence and Mean Squared Error Loss in Knowledge
Distillation
Taehyeon Kim*, Jaehoon Oh*, Nak Yil Kim,
Sangwook Cho, and Se-Young Yun
IJCAI 2021
Co-first author · Comparing KL and MSE objectives for knowledge distillation.
-
MERIT Feedback Elicits Better Bargaining in LLM Negotiators
J. Oh, M. Aghazada, Y. Shin, S. Yun, and Taehyeon Kim†
ACL 2026
-
AdaSTaR: Adaptive Data Sampling for Training Self-Taught Reasoners
W. Koh, W. Oh, J. Jang, M. Lee, H. Kim, A. Kim, J. Kim, J. Lee, Taehyeon Kim†, and S. Yun†
NeurIPS 2025
-
Guiding Reasoning in Small Language Models with LLM Assistance
Y. Kim*, E. Yi*, M. Kim, S. Yun†, and Taehyeon Kim†
CoLM 2025 Workshop · Test-time Scaling and Reasoning Models
-
C2: Scalable Auto-Feedback for LLM-based Chart Generation
W. Koh, J. Yoon, M. Lee, Y. Song, J. Cho, J. Kang, Taehyeon Kim, S. Yun, Y. Yu, and B. Lee
NAACL 2025 · Oral Presentation
-
A Unified Framework for Speculative Decoding with Multiple Drafters as a Bandit
Taehyeon Kim*, Hojung Jung*, and S. Yun
NeurIPS 2024 Workshop · ENLSP-IV
-
Block Transformer: Global-to-Local Language Modeling for Fast Inference
N. Ho*, S. Bae*, Taehyeon Kim, H. Jo, Y. Kim, T. Schuster, A. Fisch, J. Thorne, and S. Yun
NeurIPS 2024
-
FLR: Label-Mixture Regularization for Federated Learning with Noisy Labels
Taehyeon Kim, D. Kim, and S. Yun
TMLR 2024
-
Towards Fast Multilingual LLM Inference: Speculative Decoding and Specialized Drafters
E. Yi*, Taehyeon Kim*, H. Jeung, D. Chang, and S. Yun
EMNLP 2024
-
Exploring and Improving Drafts in Blockwise Parallel Decoding
Taehyeon Kim, A. T. Suresh, K. Papineni, M. Riley, S. Kumar, and A. Benton
ICML 2024 Workshop · Es-FoMo
-
Leveraging Normalization Layer in Adapters with Progressive Learning and Adaptive Distillation for Cross-Domain Few-Shot Learning
Y. Yang, Taehyeon Kim, and S. Yun
AAAI 2024
-
Navigating Data Heterogeneity in Federated Learning: A Semi-Supervised Federated
Object Detection
Taehyeon Kim, Eric Lin, Junu Lee, Christian Lau, and Vaikkunth Mugunthan
NeurIPS 2023
-
Weather4cast at NeurIPS 2022: Super-Resolution Rain Movie Prediction under Spatio-temporal Shifts
Gruca et al.
NeurIPS 2022 Competitions Track
-
Region-Conditioned Orthogonal 3D U-Net for Weather4Cast Competition
Taehyeon Kim, S. Kang, H. Shin, D. Yoon, S. Eom, K. Shin, and S. Yun
NeurIPS 2022 Workshop · Weather4Cast Competition
-
Layover Intermediate Layer for Multi-Label Classification in Efficient Transfer Learning
S. Eom, Taehyeon Kim, and S. Yun
NeurIPS 2022 Workshop · Has it Trained Yet?
-
Revisiting the Activation Function for Federated Image Classification
J. Shin, Taehyeon Kim, and S. Yun
NeurIPS 2022 Workshop · Federated Learning
-
Supernet Training for Federated Image Classification under System Heterogeneity
Taehyeon Kim and S. Yun
ICML 2022 Workshop · Dynamic Neural Networks · Oral Presentation
-
Revisiting Architecture-aware Knowledge Distillation: Smaller Models and Faster Search
Taehyeon Kim, H. Myeong, and S. Yun
ICML 2022 Workshop · HAET
-
Revisiting Orthogonality Regularization: A Study for Convolutional Neural Networks in Image Classification
Taehyeon Kim and S. Yun
IEEE Access 2022
-
FINE Samples for Learning with Noisy Labels
Taehyeon Kim*, Jongwoo Ko*, Sangwook Cho,
Jinhwan Choi, and Se-Young Yun
NeurIPS 2021
-
Adaptive Local Bayesian Optimization Over Multiple Discrete Variables
Taehyeon Kim*, J. Ahn*, N. Kim*, and S. Yun
NeurIPS 2020 Workshop · Black-Box Optimization Challenge
-
Efficient Model for Image Classification With Regularization Tricks
Taehyeon Kim, J. Kim, and S. Yun
NeurIPS 2019 Competition and Demonstration Track