collaborators

5 papers

stat.ML2026

Retrieval-Based In-Context Learning: A Domain Adaptation Framework

Yilun Zhu, Naihao Deng, Yingcong Li +2

In-context retrieval (ICR) is a retrieval-based form of in-context learning (ICL) in which demonstrations are retrieved from a source database based on similarity to the query, rat…

stat.ML2026

Estimation of the Label-Noise Transition Matrix with Performance Guarantees via Selective Classification

Xabier de Juan, Santiago Mazuelas, Yilun Zhu +1

Modern machine learning depends heavily on massive datasets, but obtaining high-quality annotations at scale is often expensive. As a result, learning from noisily-labeled data has…

cs.CL2026

It Takes One to Bias Them All: Breaking Bad with One-Shot GRPO

Naihao Deng, Yilun Zhu, Naichen Shi +2

Warning: This paper contains several toxic and offensive statements. Modern large language models (LLMs) are typically aligned through large-scale post-training to ensure fair and…

cs.CL2026

Wait, am I Being Fair? Characterizing Deductive Stereotyping and Mitigating It with Fair-GCG

Naihao Deng, Yilun Zhu, Joan Nwatu +2

Warning: This paper contains several toxic and offensive statements. While reasoning generally improves fairness in recent large language models (LLMs), failures persist. In this w…

cs.LG2025

Domain Generalization Under Posterior Drift

Yilun Zhu, Naihao Deng, Naichen Shi +2

Domain generalization (DG) is the problem of generalizing from several distributions (or domains), for which labeled training data are available, to a new test domain for which no…