Comparing the Value of Labeled and Unlabeled Data in Method-of-Moments Latent Variable Estimation. - Mayee F. Chen; Benjamin Cohen-Wang; Stephen Mussmann; Frederic Sala; Christopher Ré (AISTATS, 2021)

ContextCite: Attributing Model Generation to Context. - Benjamin Cohen-Wang; Harshay Shah; Kristian Georgiev; Aleksander Madry (NeurIPS, 2024)

Ask Your Distribution Shift if Pre-Training is Right for You. - Benjamin Cohen-Wang; Joshua Vendrow; Aleksander Madry (Trans. Mach. Learn. Res., 2025)

SelfCite: Self-Supervised Alignment for Context Attribution in Large Language Models. - Yung-Sung Chuang; Benjamin Cohen-Wang; Shannon Zejiang Shen; Zhaofeng Wu; Hu Xu 001; Xi Victoria Lin; James R. Glass; Shang-wen Li 001; Wen-tau Yih (CoRR, 2025)

Learning to Attribute with Attention. - Benjamin Cohen-Wang; Yung-Sung Chuang; Aleksander Madry (CoRR, 2025)