Benjamin Cohen-Wang

According to our database, Benjamin Cohen-Wang authored at least 5 papers between 2021 and 2025.

Collaborative distances:

• Dijkstra number of four.
• Erdős number of four.

2025

Learning to Attribute with Attention.

Benjamin Cohen-Wang, Yung-Sung Chuang, Aleksander Madry

CoRR, April, 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, Xi Victoria Lin, James R. Glass, Shang-wen Li, Wen-tau Yih

CoRR, February, 2025

Ask Your Distribution Shift if Pre-Training is Right for You.

Benjamin Cohen-Wang, Joshua Vendrow, Aleksander Madry

Trans. Mach. Learn. Res., 2025

2024

ContextCite: Attributing Model Generation to Context.

Benjamin Cohen-Wang, Harshay Shah, Kristian Georgiev, Aleksander Madry

Proceedings of the Advances in Neural Information Processing Systems 38: Annual Conference on Neural Information Processing Systems 2024, 2024

2021

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é

Proceedings of the 24th International Conference on Artificial Intelligence and Statistics, 2021