← Search

Jakub Tarnawski

10 accepted papers

2024

DéjàVu: KV-cache Streaming for Fast, Fault-tolerant Generative LLM Serving

ICML 2024poster

Distributed LLM serving is costly and often underutilizes hardware accelerators due to three key challenges: bubbles in pipeline-parallel deployments caused by the bimodal latency of prompt and token processing, GPU memory overprovisioning, and long recovery times in case of failures. DéjàVu address…

Cited by 29Source ↗PDF ↗Scholar ↗
2024

Efficiently Computing Similarities to Private Datasets

ICLR 2024poster

Many methods in differentially private model training rely on computing the similarity between a query point (such as public or synthetic data) and private data. We abstract out this common subroutine and study the following fundamental algorithmic problem: Given a similarity function $f$ and a larg…

Cited by 6Source ↗PDF ↗Scholar ↗
2024

Fairness in Submodular Maximization over a Matroid Constraint

AISTATS 2024poster

Submodular maximization over a matroid constraint is a fundamental problem with various applications in machine learning. Some of these applications involve decision-making over datapoints with sensitive attributes such as gender or race. In such settings, it is crucial to guarantee that the selecte…

Cited by 6Source ↗PDF ↗Scholar ↗
2024

Integrated Hardware Architecture and Device Placement Search

ICML 2024poster

Distributed execution of deep learning training involves a dynamic interplay between hardware accelerator architecture and device placement strategy. This is the first work to explore the co-optimization of determining the optimal architecture and device placement strategy through novel algorithms,…

2023

Fairness in Streaming Submodular Maximization over a Matroid Constraint

ICML 2023poster

Streaming submodular maximization is a natural model for the task of selecting a representative subset from a large-scale dataset. If datapoints have sensitive attributes such as gender or race, it becomes important to enforce fairness to avoid bias and discrimination. This has spurred significant i…

2022

Near-Optimal Correlation Clustering with Privacy

NeurIPS 2022accept

Correlation clustering is a central problem in unsupervised learning, with applications spanning community detection, duplicate detection, automated labeling and many more. In the correlation clustering problem one receives as input a set of nodes and for each node a list of co-clustering preference…

Cited by 18Source ↗PDF ↗Scholar ↗
2021

Correlation Clustering in Constant Many Parallel Rounds

ICML 2021oral

Correlation clustering is a central topic in unsupervised learning, with many applications in ML and data mining. In correlation clustering, one receives as input a signed graph and the goal is to partition it to minimize the number of disagreements. In this work we propose a massively parallel comp…

Cited by 48Source ↗PDF ↗Scholar ↗
2018

Beyond 1/2-Approximation for Submodular Maximization on Massive Data Streams

ICML 2018oral

Many tasks in machine learning and data mining, such as data diversification, non-parametric learning, kernel machines, clustering etc., require extracting a small but representative summary from a massive dataset. Often, such problems can be posed as maximizing a submodular set function subject to…

Cited by 112Source ↗PDF ↗Scholar ↗