← Search

Aditya Balu

7 accepted papers

2024

DIMAT: Decentralized Iterative Merging-And-Training for Deep Learning Models

CVPR 2024poster

Recent advances in decentralized deep learning algorithms have demonstrated cutting-edge performance on various tasks with large pre-trained models. However a pivotal prerequisite for achieving this level of competitiveness is the significant communication and computation overheads when updating the…

2024

Slice-100K: A Multimodal Dataset for Extrusion-based 3D Printing

NeurIPS 2024poster

G-code (Geometric code) or RS-274 is the most widely used computer numerical control (CNC) and 3D printing programming language. G-code provides machine instructions for the movement of the 3D printer, especially for the nozzle, stage, and extrusion of material for extrusion-based additive manufactu…

Cited by 0SourcePDFScholar
2022

MDPGT: Momentum-Based Decentralized Policy Gradient Tracking

AAAI 2022technical

We propose a novel policy gradient method for multi-agent reinforcement learning, which leverages two different variance-reduction techniques and does not require large batches over iterations. Specifically, we propose a momentum-based decentralized policy gradient tracking (MDPGT) where a new momen…

2021

Cross-Gradient Aggregation for Decentralized Learning from Non-IID Data

ICML 2021spotlight

Decentralized learning enables a group of collaborative agents to learn models using a distributed dataset without the need for a central parameter server. Recently, decentralized learning algorithms have demonstrated state-of-the-art results on benchmark data sets, comparable with centralized algor…

2021

Decentralized Deep Learning Using Momentum-Accelerated Consensus

ICASSP 2021accepted

We consider the problem of decentralized deep learning where multiple agents collaborate to learn from a distributed dataset. While several decentralized deep learning approaches exist, the majority consider a central parameter-server topology for aggregating the model parameters from the agents. Ho…

Cited by 0SourceScholar
2021

Differentiable Spline Approximations

NeurIPS 2021poster

The paradigm of differentiable programming has significantly enhanced the scope of machine learning via the judicious use of gradient-based optimization. However, standard differentiable programming methods (such as autodiff) typically require that the machine learning models be differentiable, limi…

2017

Collaborative Deep Learning in Fixed Topology Networks

NeurIPS 2017poster

There is significant recent interest to parallelize deep learning algorithms in order to handle the enormous growth in data and model sizes. While most advances focus on model parallelization and engaging multiple computing agents via using a central parameter server, aspect of data parallelization…