Cohen-Wang, Benjamin

8 publications

ICLR 2026 Hot PATE: Private Aggregation of Distributions for Diverse Tasks Edith Cohen, Benjamin Cohen-Wang, Xin Lyu, Jelani Nelson, Tamas Sarlos, Uri Stemmer

TMLR 2025 Ask Your Distribution Shift if Pre-Training Is Right for You Benjamin Cohen-Wang, Joshua Vendrow, Aleksander Madry

ICML 2025 SelfCite: Self-Supervised Alignment for Context Attribution in Large Language Models Yung-Sung Chuang, Benjamin Cohen-Wang, Zejiang Shen, Zhaofeng Wu, Hu Xu, Xi Victoria Lin, James R. Glass, Shang-Wen Li, Wen-Tau Yih

NeurIPS 2024 ContextCite: Attributing Model Generation to Context Benjamin Cohen-Wang, Harshay Shah, Kristian Georgiev, Aleksander Mądry

ICMLW 2024 ContextCite: Attributing Model Generation to Context Benjamin Cohen-Wang, Harshay Shah, Kristian Georgiev, Aleksander Madry

ICMLW 2024 ContextCite: Attributing Model Generation to Context Benjamin Cohen-Wang, Harshay Shah, Kristian Georgiev, Aleksander Madry

NeurIPSW 2023 Ask Your Distribution Shift if Pre-Training Is Right for You Benjamin Cohen-Wang, Joshua Vendrow, Aleksander Madry

AISTATS 2021 Comparing the Value of Labeled and Unlabeled Data in Method-of-Moments Latent Variable Estimation Mayee Chen, Benjamin Cohen-Wang, Stephen Mussmann, Frederic Sala, Christopher Re