← Search

Weiwei Kong

6 accepted papers

2025

Differentially private optimization for non-decomposable objective functions

ICLR 2025poster

Unsupervised pre-training is a common step in developing computer vision models and large language models. In this setting, the absence of labels requires the use of similarity-based loss functions, such as the contrastive loss, that favor minimizing the distance between similar inputs and maximizin…

Cited by 2SourcePDFScholar
2025

Escaping Collapse: The Strength of Weak Data for Large Language Model Training

NeurIPS 2025poster

Synthetically-generated data plays an increasingly larger role in training large language models. However, while synthetic data has been found to be useful, studies have also shown that without proper curation it can cause LLM performance to plateau, or even "collapse", after many training iteration…

Cited by 0SourceScholar
2024

Private prediction for large-scale synthetic text generation

EMNLP 2024finding

We present an approach for generating differentially private synthetic text using large language models (LLMs), via private prediction. In the private prediction framework, we only require the output synthetic data to satisfy differential privacy guarantees. This is in contrast to approaches that tr…

2020

Rankmax: An Adaptive Projection Alternative to the Softmax Function

NeurIPS 2020poster

Several machine learning models involve mapping a score vector to a probability vector. Usually, this is done by projecting the score vector onto a probability simplex, and such projections are often characterized as Lipschitz continuous approximations of the argmax function, whose Lipschitz constan…

Cited by 20SourcePDFScholar
2019

A new dog learns old tricks: RL finds classic optimization algorithms

ICLR 2019poster

This paper introduces a novel framework for learning algorithms to solve online combinatorial optimization problems. Towards this goal, we introduce a number of key ideas from traditional algorithms and complexity theory. First, we draw a new connection between primal-dual methods and reinforcement…

Cited by 57SourcePDFScholar