← Search

Yiqiao Zhong

4 accepted papers

2025

Model Editing as a Robust and Denoised variant of DPO: A Case Study on Toxicity

ICLR 2025poster

Recent alignment algorithms such as direct preference optimization (DPO) have been developed to improve the safety of large language models (LLMs) by training these models to match human behaviors exemplified by preference data. However, these methods are both computationally intensive and lacking…

2025

Unifying Attention Heads and Task Vectors via Hidden State Geometry in In-Context Learning

NeurIPS 2025poster

The unusual properties of in-context learning (ICL) have prompted investigations into the internal mechanisms of large language models. Prior work typically focuses on either special attention heads or task vectors at specific layers, but lacks a unified framework linking these components to the evo…

Cited by 0SourcecodeScholar
2024

How does Multi-Task Training Affect Transformer In-Context Capabilities? Investigations with Function Classes

NAACL 2024short

Large language models (LLM) have recently shown the extraordinary ability to perform unseen tasks based on few-shot examples provided as text, also known as in-context learning (ICL). While recent works have attempted to understand the mechanisms driving ICL, few have explored training strategies th…