← Search

Yi Su

18 accepted papers

2025

Accurate KV Cache Quantization with Outlier Tokens Tracing

ACL 2025long

The impressive capabilities of Large Language Models (LLMs) come at the cost of substantial computational resources during deployment. While KV Cache can significantly reduce recomputation during inference, it also introduces additional memory overhead. KV Cache quantization presents a promising sol…

2025

EVOLvE: Evaluating and Optimizing LLMs For In-Context Exploration

ICML 2025poster

Despite their success in many domains, large language models (LLMs) remain under-studied in scenarios requiring optimal decision-making under uncertainty. This is crucial as many real-world applications, ranging from personalized recommendations to healthcare interventions, demand that LLMs not only…

Cited by 5SourcePDFScholar
2025

Training Language Models to Self-Correct via Reinforcement Learning

ICLR 2025oral

Self-correction is a highly desirable capability of large language models (LLMs), yet it has consistently been found to be largely ineffective in modern LLMs. Current methods for training self-correction typically depend on either multiple models, a more advanced model, or additional forms of super…

Cited by 113SourcePDFScholar
2025

Understanding How Value Neurons Shape the Generation of Specified Values in LLMs

EMNLP 2025

Rapid integration of large language models (LLMs) into societal applications has intensified concerns about their alignment with universal ethical principles, as their internal value representations remain opaque despite behavioral alignment advancements. Current approaches struggle to systematicall

Cited by 0SourcePDFScholar
2024

Demonstration Augmentation for Zero-shot In-context Learning

ACL 2024findings

Large Language Models (LLMs) have demonstrated an impressive capability known as In-context Learning (ICL), which enables them to acquire knowledge from textual demonstrations without the need for parameter updates.However, many studies have highlighted that the model’s performance is sensitive to t…

2024

Leveraging Large Language Models for Exploiting ASR Uncertainty

ICASSP 2024accepted

While large language models excel in a variety of natural language processing (NLP) tasks, to perform well on spoken language understanding (SLU) tasks, they must either rely on off-the-shelf automatic speech recognition (ASR) systems for transcription, or be equipped with an in-built speech modalit…

Cited by 0SourceScholar
2024

Online Feature Updates Improve Online (Generalized) Label Shift Adaptation

NeurIPS 2024poster

This paper addresses the prevalent issue of label shift in an online setting with missing labels, where data distributions change over time and obtaining timely labels is challenging. While existing methods primarily focus on adjusting or updating the final layer of a pre-trained classifier, we expl…

Cited by 2SourcePDFScholar
2023

Beware of Model Collapse! Fast and Stable Test-time Adaptation for Robust Question Answering

EMNLP 2023long main

Although pre-trained language models (PLM) have achieved great success in question answering (QA), their robustness is still insufficient to support their practical applications, especially in the face of distribution shifts. Recently, test-time adaptation (TTA) has shown great potential for solving…

Cited by 0SourceScholar
2023

Offline RL for Natural Language Generation with Implicit Language Q Learning

ICLR 2023poster

Large language models distill broad knowledge from text corpora. However, they can be inconsistent when it comes to completing user specified tasks. This issue can be addressed by finetuning such models via supervised learning on curated datasets, or via reinforcement learning. In this work, we prop…

Cited by 109SourcePDFScholar
2023

Unified Off-Policy Learning to Rank: a Reinforcement Learning Perspective

NeurIPS 2023poster

Off-policy Learning to Rank (LTR) aims to optimize a ranker from data collected by a deployed logging policy. However, existing off-policy learning to rank methods often make strong assumptions about how users generate the click data, i.e., the click model, and hence need to tailor their methods spe…

2022

Context-Aware Language Modeling for Goal-Oriented Dialogue Systems

NAACL 2022findings

Goal-oriented dialogue systems face a trade-off between fluent language generation and task-specific control. While supervised learning with large language models is capable of producing realistic text, how to steer such responses towards completing a specific task without sacrificing language quali…

Cited by 27SourcePDFScholar
2022

Data-Driven Offline Decision-Making via Invariant Representation Learning

NeurIPS 2022accept

The goal in offline data-driven decision-making is synthesize decisions that optimize a black-box utility function, using a previously-collected static dataset, with no active interaction. These problems appear in many forms: offline reinforcement learning (RL), where we must produce actions that op…

Cited by 25SourcePDFScholar
2021

R2D2: Recursive Transformer based on Differentiable Tree for Interpretable Hierarchical Language Modeling

ACL 2021long

Human language understanding operates at multiple levels of granularity (e.g., words, phrases, and sentences) with increasing levels of abstraction that can be hierarchically combined. However, existing deep models with stacked layers do not explicitly model any sort of hierarchical process. In this…

2020

Doubly robust off-policy evaluation with shrinkage

ICML 2020poster

We propose a new framework for designing estimators for off-policy evaluation in contextual bandits. Our approach is based on the asymptotically optimal doubly robust estimator, but we shrink the importance weights to minimize a bound on the mean squared error, which results in a better bias-varianc…

Cited by 119SourcePDFScholar
2019

CAB: Continuous Adaptive Blending for Policy Evaluation and Learning

ICML 2019oral

The ability to perform offline A/B-testing and off-policy learning using logged contextual bandit feedback is highly desirable in a broad range of applications, including recommender systems, search engines, ad placement, and personalized health care. Both offline A/B-testing and off-policy learning…

Cited by 86SourcePDFScholar