Yang, Haolin
10 publications
ICLR
2026
Localizing Task Recognition and Task Learning in In-Context Learning via Attention Head Analysis
ICML
2025
Reducing Confounding Bias Without Data Splitting for Causal Inference via Optimal Transport
NeurIPS
2025
Unifying Attention Heads and Task Vectors via Hidden State Geometry in In-Context Learning