Publications

2026

RecaLLM: Addressing the Lost-in-Thought Phenomenon with Explicit In-Context Retrieval
Kyle Whitecross and Negin Rahimi
Conference on Language Modeling (CoLM), 2026.
[Paper] / [Code]

2024

Investigating the Impact of Model Width and Density on Generalization in Presence of Label Noise
Yihao Xue, Kyle Whitecross and Baharan Mirzasoleiman
Conference on Uncertainty in Artificial Intelligence (UAI), 2024.
[Paper]

2022

Investigating Why Contrastive Learning Benefits Robustness Against Label Noise
Yihao Xue, Kyle Whitecross and Baharan Mirzasoleiman
International Conference on Machine Learning (ICML), 2022.
[Paper] / [Slides]