Adaptive Uncertainty-Weighted ADMM for Distributed Optimization

Jianping Ye,Caleb Wan,Samy Wu Fung
DOI: https://doi.org/10.48550/arXiv.2109.01089
2022-04-20
Abstract:We present AUQ-ADMM, an adaptive uncertainty-weighted consensus ADMM method for solving large-scale convex optimization problems in a distributed manner. Our key contribution is a novel adaptive weighting scheme that empirically increases the progress made by consensus ADMM scheme and is attractive when using a large number of subproblems. The weights are related to the uncertainty associated with the solutions of each subproblem, and are efficiently computed using low-rank approximations. We show AUQ-ADMM provably converges and demonstrate its effectiveness on a series of machine learning applications, including elastic net regression, multinomial logistic regression, and support vector machines. We provide an implementation based on the PyTorch package.
Optimization and Control,Numerical Analysis
What problem does this paper attempt to address?