← Search

Kyungwoo Song

30 accepted papers

2026

Bounded Hyperbolic Tangent: A Stable and Efficient Alternative to Pre-Layer Normalization in Large Language Models

ICML 2026poster

Pre-Layer Normalization (Pre-LN) is the de facto choice for large language models (LLMs) and is crucial for stable pretraining and effective transfer learning. However, Pre-LN is inefficient due to repeated statistical calculations and suffers from the curse of depth. As layers grow, the magnitude a…

Cited by 0SourceScholar
2026

Dissecting Causal Mechanism Shifts via FANS: Function And Noise Separation

ICML 2026poster

Identifying the drivers of causal mechanism shifts, distinguishing functional changes from noise alterations, known as dissection, is a critical yet under-explored problem in data science (e.g., biomedical science and manufacturing). This paper introduces a more general and unified framework, the fu…

Cited by 0SourceScholar
2026

Multi-LLM Adaptive Conformal Inference for Reliable LLM Response

ICLR 2026poster

Ensuring factuality is essential for the safe use of Large Language Models (LLMs) in high-stakes domains such as medicine and law. Conformal inference provides distribution-free guarantees, but existing approaches are either overly conservative, discarding many true-claims, or rely on adaptive error…

Cited by 0SourcecodeScholar
2026

P3T: Prototypical Point-Level Prompt Tuning with Enhanced Generalization for 3D Vision-Language Models

ICRA 2026poster

With the rise of pre-trained models in the 3D point cloud domain for a wide range of real-world applications, adapting them to downstream tasks has become increasingly important. However, conventional full fine-tuning methods are computationally expensive and storage-intensive. Although prompt tunin…

2026

Semi-Supervised Preference Optimization with Limited Feedback

ICLR 2026oral

The field of preference optimization has made outstanding contributions to the alignment of language models with human preferences. Despite these advancements, recent methods still rely heavily on substantial paired (labeled) feedback data, leading to substantial resource expenditures. To address th…

Cited by 0SourcecodeScholar
2026

Spurious Correlation-Aware Embedding Regularization for Worst-Group Robustness

ICLR 2026poster

Deep learning models achieve strong performance across various domains but often rely on spurious correlations, making them vulnerable to distribution shifts. This issue is particularly severe in subpopulation shift scenarios, where models struggle in underrepresented groups. While existing methods…

Cited by 0SourceScholar
2025

Brain-inspired $L_p$-Convolution benefits large kernels and aligns better with visual cortex

ICLR 2025poster

Convolutional Neural Networks (CNNs) have profoundly influenced the field of computer vision, drawing significant inspiration from the visual processing mechanisms inherent in the brain. Despite sharing fundamental structural and representational similarities with the biological visual system, diffe…

Cited by 0SourcePDFScholar
2025

CCL: Causal-aware In-context Learning for Out-of-Distribution Generalization

NeurIPS 2025poster

In-context learning (ICL), a nonparametric learning method based on the knowledge of demonstration sets, has become a de facto standard for large language models (LLMs). The primary goal of ICL is to select valuable demonstration sets to enhance the performance of LLMs. Traditional ICL methods choos…

Cited by 0SourcecodeScholar
2025

DaWin: Training-free Dynamic Weight Interpolation for Robust Adaptation

ICLR 2025poster

Adapting a pre-trained foundation model on downstream tasks should ensure robustness against distribution shifts without the need to retrain the whole model. Although existing weight interpolation methods are simple yet effective, we argue their static nature limits downstream performance while achi…

2025

Flat Posterior Does Matter For Bayesian Model Averaging

UAI 2025

Bayesian neural networks (BNNs) estimate the posterior distribution of model parameters and utilize posterior samples for Bayesian Model Averaging (BMA) in prediction. However, despite the crucial role of flatness in the loss landscape in improving the generalization of neural networks, its impact o

2025

LBC: Language-Based-Classifier for Out-Of-Variable Generalization

NAACL 2025long

Large Language Models (LLMs) have great success in natural language processing tasks such as response generation. However, their use in tabular data has been limited due to their inferior performance compared to traditional machine learning models (TMLs) such as XGBoost. We find that the pre-trained…

2025

Sufficient Invariant Learning for Distribution Shift

CVPR 2025poster

Learning robust models under distribution shifts between training and test datasets is a fundamental challenge in machine learning. While learning invariant features across environments is a popular approach, it often assumes that these features are fully observed in both training and test sets--a c…

2025

TIDES: Technical Information Discovery and Extraction System

EMNLP 2025

Addressing the challenges in QA for specific technical domains requires identifying relevant portions of extensive documents and generating answers based on this focused content. Traditional pre-trained LLMs often struggle with domain-specific terminology, while fine-tuned LLMs demand substantial co

2024

CED: Comparing Embedding Differences for Detecting Out-of-Distribution and Hallucinated Text

EMNLP 2024finding

Detecting out-of-distribution (OOD) samples is crucial for ensuring the safety and robustness of models deployed in real-world scenarios. While most studies on OOD detection focus on fine-tuned models trained on in-distribution (ID) data, detecting OOD in pre-trained models is also important due to…

Cited by 0SourcePDFScholar
2024

Pre-trained Vision and Language Transformers Are Few-Shot Incremental Learners

CVPR 2024poster

Few-Shot Class Incremental Learning (FSCIL) is a task that requires a model to learn new classes incrementally without forgetting when only a few samples for each class are given. FSCIL encounters two significant challenges: catastrophic forgetting and overfitting and these challenges have driven pr…

2024

Towards Calibrated Robust Fine-Tuning of Vision-Language Models

NeurIPS 2024poster

Improving out-of-distribution (OOD) generalization during in-distribution (ID) adaptation is a primary goal of robust fine-tuning of zero-shot models beyond naive fine-tuning. However, despite decent OOD generalization performance from recent robust fine-tuning methods, confidence calibration for re…

2023

BlackVIP: Black-Box Visual Prompting for Robust Transfer Learning

CVPR 2023poster

With the surge of large-scale pre-trained models (PTMs), fine-tuning these models to numerous downstream tasks becomes a crucial problem. Consequently, parameter efficient transfer learning (PETL) of large models has grasped huge attention. While recent PETL methods showcase impressive performance,…

2023

Geodesic Multi-Modal Mixup for Robust Fine-Tuning

NeurIPS 2023poster

Pre-trained multi-modal models, such as CLIP, provide transferable embeddings and show promising results in diverse applications. However, the analysis of learned multi-modal embeddings is relatively unexplored, and the embedding transferability can be improved. In this work, we observe that CLIP ho…

2023

SAAL: Sharpness-Aware Active Learning

ICML 2023poster

While deep neural networks play significant roles in many research areas, they are also prone to overfitting problems under limited data instances. To overcome overfitting, this paper introduces the first active learning method to incorporate the sharpness of loss space into the acquisition function…

2022

Efficient Approximate Inference for Stationary Kernel on Frequency Domain

ICML 2022spotlight

Based on the Fourier duality between a stationary kernel and its spectral density, modeling the spectral density using a Gaussian mixture density enables one to construct a flexible kernel, known as a Spectral Mixture kernel, that can model any stationary kernel. However, despite its expressive powe…

2022

From Noisy Prediction to True Label: Noisy Prediction Calibration via Generative Model

ICML 2022spotlight

Noisy labels are inevitable yet problematic in machine learning society. It ruins the generalization of a classifier by making the classifier over-fitted to noisy labels. Existing methods on noisy label have focused on modifying the classifier during the training procedure. It has two potential prob…

2022

Soft Truncation: A Universal Training Technique of Score-based Diffusion Model for High Precision Score Estimation

ICML 2022spotlight

Recent advances in diffusion models bring state-of-the-art performance on image generation tasks. However, empirical results from previous research in diffusion models imply an inverse correlation between density estimation and sample generation performances. This paper investigates with sufficient…

2022

Unknown-Aware Domain Adversarial Learning for Open-Set Domain Adaptation

NeurIPS 2022accept

Open-Set Domain Adaptation (OSDA) assumes that a target domain contains unknown classes, which are not discovered in a source domain. Existing domain adversarial learning methods are not suitable for OSDA because distribution matching with $\textit{unknown}$ classes leads to negative transfer. Previ…

2021

Counterfactual Fairness with Disentangled Causal Effect Variational Autoencoder

AAAI 2021technical

The problem of fair classification can be mollified if we develop a method to remove the embedded sensitive information from the classification features. This line of separating the sensitive information is developed through the causal inference, and the causal inference enables the counterfactual g…

Cited by 65SourcePDFScholar
2021

LADA: Look-Ahead Data Acquisition via Augmentation for Deep Active Learning

NeurIPS 2021poster

Active learning effectively collects data instances for training deep learning models when the labeled dataset is limited and the annotation cost is high. Data augmentation is another effective technique to enlarge the limited amount of labeled instances. The scarcity of labeled dataset leads us to…

2020

Context Aware Sequence Modeling

IJCAI 2020poster

Context modeling helps understand the data, such as sentence or user behavior. Contextual information captures the important underlying feature, and it enhances the relationship between data instances or hidden representations. As the importance of the sequential model grows, so does the importance…

Cited by 0SourcePDFScholar