← Search

Divyansh Jhunjhunwala

7 accepted papers

2026

Navigating the Accuracy-Size Trade-Off with Flexible Model Merging

ICLR 2026poster

Model merging has emerged as an efficient method to combine multiple single-task fine-tuned models. The merged model can enjoy multi-task capabilities without expensive training. While promising, merging into a single model often suffers from an accuracy gap with respect to individual fine-tuned mod…

Cited by 0SourceScholar
2025

Ravan: Multi-Head Low-Rank Adaptation for Federated Fine-Tuning

NeurIPS 2025poster

Large Language Models (LLMs) have yet to effectively leverage the vast amounts of edge-device data, and Federated Learning (FL) offers a promising paradigm to collaboratively fine-tune LLMs without transferring private edge data to the cloud. To operate within the computational and communication con…

Cited by 0SourceScholar
2024

FedFisher: Leveraging Fisher Information for One-Shot Federated Learning

AISTATS 2024poster

Standard federated learning (FL) algorithms typically require multiple rounds of communication between the server and the clients, which has several drawbacks, including requiring constant network connectivity, repeated investment of computational resources, and susceptibility to privacy attacks. On…

2023

FedExP: Speeding Up Federated Averaging via Extrapolation

ICLR 2023top-25%

Federated Averaging (FedAvg) remains the most popular algorithm for Federated Learning (FL) optimization due to its simple implementation, stateless nature, and privacy guarantees combined with secure aggregation. Recent work has sought to generalize the vanilla averaging in FedAvg to a generalized…

2022

Fedvarp: Tackling the variance due to partial client participation in federated learning

UAI 2022poster

Data-heterogeneous federated learning (FL) systems suffer from two significant sources of convergence error: 1) client drift error caused by performing multiple local optimization steps at clients, and 2) partial client participation error caused by the fact that only a small subset of the edge clie…

Cited by 72SourcePDFScholar
2021

Adaptive Quantization of Model Updates for Communication-Efficient Federated Learning

ICASSP 2021accepted

Communication of model updates between client nodes and the central aggregating server is a major bottleneck in federated learning, especially in bandwidth-limited settings and high-dimensional models. Gradient quantization is an effective way of reducing the number of bits required to communicate e…

Cited by 0SourceScholar
2021

Leveraging Spatial and Temporal Correlations in Sparsified Mean Estimation

NeurIPS 2021poster

We study the problem of estimating at a central server the mean of a set of vectors distributed across several nodes (one vector per node). When the vectors are high-dimensional, the communication cost of sending entire vectors may be prohibitive, and it may be imperative for them to use sparsificat…

Cited by 18SourcePDFScholar