← 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 29SourcePDFScholar
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 6SourcePDFScholar
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 6SourcePDFScholar
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 18SourcePDFScholar
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 48SourcePDFScholar
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 112SourcePDFScholar