← Search

Mohammadreza Ebrahimi

7 accepted papers

2025

Multi-Draft Speculative Sampling: Canonical Decomposition and Theoretical Limits

ICLR 2025spotlight

We consider multi-draft speculative sampling, where the proposal sequences are sampled independently from different draft models. At each step, a token-level draft selection scheme takes a list of valid tokens as input and produces an output token whose distribution matches that of the target mode…

Cited by 0SourcePDFScholar
2024

Differentially Private Stochastic Gradient Descent with Fixed-Size Minibatches: Tighter RDP Guarantees with or without Replacement

NeurIPS 2024poster

Differentially private stochastic gradient descent (DP-SGD) has been instrumental in privately training deep learning models by providing a framework to control and track the privacy loss incurred during training. At the core of this computation lies a subsampling method that uses a privacy amplifi…

2023

Sequential Gradient Coding For Straggler Mitigation

ICLR 2023poster

In distributed computing, slower nodes (stragglers) usually become a bottleneck. Gradient Coding (GC), introduced by Tandon et al., is an efficient technique that uses principles of error-correcting codes to distribute gradient computation in the presence of stragglers. In this paper, we consider th…

Cited by 1SourcePDFScholar
2023

Time-Resolved FMRI Shared Response Model Using Gaussian Process Factor Analysis

ICASSP 2023accepted

Multi-subject fMRI studies are challenging due to the high variability of both brain anatomy and functional brain topographies across participants. An effective way of aggregating multi-subject fMRI data is to extract a shared representation that filters out unwanted variability among subjects. Some…

Cited by 0SourceScholar