← Search

Shawn Im

8 accepted papers

2026

How Do Transformers Learn to Associate Tokens: Gradient Leading Terms Bring Mechanistic Interpretability

ICLR 2026oral

Semantic associations such as the link between "bird" and "flew" are foundational for language modeling as they enable models to go beyond memorization and instead generalize and generate coherent text. Understanding how these associations are learned and represented in language models is essential…

Cited by 0SourceScholar
2026

Normalized Rewards for Preference Optimization

ICML 2026poster

Direct Alignment Algorithms (DAAs) such as DPO have become a common way to post-train and align LLMs with human preferences. However, DAAs have been observed to over-optimize their implicit reward model and decrease the likelihood of preferred responses. This results in a decrease in the total likel…

Cited by 0SourceScholar
2025

Position: Challenges and Future Directions of Data-Centric AI Alignment

ICML 2025poster

As AI systems become increasingly capable and influential, ensuring their alignment with human values, preferences, and goals has become a critical research focus. Current alignment methods primarily focus on designing algorithms and loss functions but often underestimate the crucial role of data. T…

Cited by 0SourcePDFScholar
2025

Towards Interpretability Without Sacrifice: Faithful Dense Layer Decomposition with Mixture of Decoders

NeurIPS 2025poster

Multilayer perceptrons (MLPs) are an integral part of large language models, yet their dense representations render them difficult to understand, edit, and steer. Recent methods learn interpretable approximations via neuron-level sparsity, yet fail to faithfully reconstruct the original mapping--sig…

Cited by 0SourcecodeScholar
2025

Understanding Multimodal LLMs Under Distribution Shifts: An Information-Theoretic Approach

ICML 2025poster

Multimodal large language models (MLLMs) have shown promising capabilities but struggle under distribution shifts, where evaluation data differ from instruction tuning distributions. Although previous works have provided empirical evaluations, we argue that establishing a formal framework that can c…

Cited by 2SourcePDFScholar