← Search

Benjamin Cohen-Wang

4 accepted papers

2026

Hot PATE: Private Aggregation of Distributions for Diverse Tasks

ICLR 2026poster

The Private Aggregation of Teacher Ensembles (PATE) framework enables privacy-preserving machine learning by aggregating responses from disjoint subsets of sensitive data. Adaptations of PATE to tasks with inherent output diversity such as text generation, where the desired output is a sample from a…

Cited by 0SourceScholar
2025

SelfCite: Self-Supervised Alignment for Context Attribution in Large Language Models

ICML 2025poster

We introduce SelfCite, a novel self-supervised approach that aligns LLMs to generate high-quality, fine-grained, sentence-level citations for the statements in their generated responses. Instead of only relying on costly and labor-intensive annotations, SelfCite leverages a reward signal provided b…

2024

ContextCite: Attributing Model Generation to Context

NeurIPS 2024poster

How do language models use information provided as context when generating a response? Can we infer whether a particular generated statement is actually grounded in the context, a misinterpretation, or fabricated? To help answer these questions, we introduce the problem of *context attribution*: pin…

2021

Comparing the Value of Labeled and Unlabeled Data in Method-of-Moments Latent Variable Estimation

AISTATS 2021poster

Labeling data for modern machine learning is expensive and time-consuming. Latent variable models can be used to infer labels from weaker, easier-to-acquire sources operating on unlabeled data. Such models can also be trained using labeled data, presenting a key question: should a user invest in few…