← Search

Taehyeon Kim

13 accepted papers

2025

AdaSTaR: Adaptive Data Sampling for Training Self-Taught Reasoners

NeurIPS 2025poster

Self-Taught Reasoners (STaR), synonymously known as Rejection sampling Fine-Tuning (RFT), is an integral part of the training pipeline of self-improving reasoning Language Models (LMs). The self-improving mechanism often employs random observation (data) sampling. However, this results in trained…

Cited by 0SourceScholar
2025

Adaptive Task Allocation in Multi-Human Multi-Robot Teams Under Team Heterogeneity and Dynamic Information Uncertainty

ICRA 2025

Task allocation in multi-human multi-robot (MHMR) teams presents significant challenges due to the inherent heterogeneity of team members, the dynamics of task execution, and the information uncertainty of operational states. Existing approaches often fail to address these challenges simultaneously,

Cited by 6SourceScholar
2025

MMR: A Large-scale Benchmark Dataset for Multi-target and Multi-granularity Reasoning Segmentation

ICLR 2025poster

The fusion of Large Language Models (LLMs) with vision models is pioneering new possibilities in user-interactive vision-language tasks. A notable application is reasoning segmentation, where models generate pixel-level segmentation masks by comprehending implicit meanings in human instructions. How…

2025

PrefMMT: Modeling Human Preferences in Preference-based Reinforcement Learning with Multimodal Transformers

IROS 2025

Preference-based reinforcement learning (PbRL) shows promise in aligning robot behaviors with human preferences, but its success depends heavily on the accurate modeling of human preferences through reward models. Most methods adopt Markovian assumptions for preference modeling (PM), which overlook

Cited by 0SourceScholar
2024

Accelerating Blockwise Parallel Language Models with Draft Refinement

NeurIPS 2024poster

Autoregressive language models have achieved remarkable advancements, yet their potential is often limited by the slow inference speeds associated with sequential token generation. Blockwise parallel decoding (BPD) was proposed by Stern et al. [42] as a method to improve inference speed of language…

Cited by 2SourcePDFScholar
2024

Block Transformer: Global-to-Local Language Modeling for Fast Inference

NeurIPS 2024poster

We introduce the Block Transformer which adopts hierarchical global-to-local modeling to autoregressive transformers to mitigate the inference bottlenecks associated with self-attention. Self-attention requires the key-value (KV) cache of all previous sequences to be retrieved from memory at every d…

2024

Instructive Decoding: Instruction-Tuned Large Language Models are Self-Refiner from Noisy Instructions

ICLR 2024spotlight

While instruction-tuned language models have demonstrated impressive zero-shot generalization, these models often struggle to generate accurate responses when faced with instructions that fall outside their training set. This paper presents Instructive Decoding (ID), a simple yet effective approach…

2024

Leveraging Normalization Layer in Adapters with Progressive Learning and Adaptive Distillation for Cross-Domain Few-Shot Learning

AAAI 2024technical

Cross-domain few-shot learning presents a formidable challenge, as models must be trained on base classes and then tested on novel classes from various domains with only a few samples at hand. While prior approaches have primarily focused on parameter-efficient methods of using adapters, they often…

2024

Towards Fast Multilingual LLM Inference: Speculative Decoding and Specialized Drafters

EMNLP 2024main

Large language models (LLMs) have revolutionized natural language processing and broadened their applicability across diverse commercial applications. However, the deployment of these models is constrained by high inference time in multilingual settings. To mitigate this challenge, this paper explor…

2023

Navigating Data Heterogeneity in Federated Learning: A Semi-Supervised Federated Object Detection

NeurIPS 2023poster

Federated Learning (FL) has emerged as a potent framework for training models across distributed data sources while maintaining data privacy. Nevertheless, it faces challenges with limited high-quality labels and non-IID client data, particularly in applications like autonomous driving. To address t…

2021

Comparing Kullback-Leibler Divergence and Mean Squared Error Loss in Knowledge Distillation

IJCAI 2021poster

Knowledge distillation (KD), transferring knowledge from a cumbersome teacher model to a lightweight student model, has been investigated to design efficient neural architectures. Generally, the objective function of KD is the Kullback-Leibler (KL) divergence loss between the softened probability di…

2021

FINE Samples for Learning with Noisy Labels

NeurIPS 2021poster

Modern deep neural networks (DNNs) become frail when the datasets contain noisy (incorrect) class labels. Robust techniques in the presence of noisy labels can be categorized into two folds: developing noise-robust functions or using noise-cleansing methods by detecting the noisy data. Recently, noi…