lx10077 / LocalPower

Codes for LocalPower

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Communication-Efficient Distributed SVD via Local Power Iterations

This repository contains the codes for the paper

Communication-Efficient Distributed SVD via Local Power Iterations

If you find this code useful in your research, please consider citing:

@inproceedings{li2021communication,
  title={Communication-efficient distributed SVD via local power iterations},
  author={Li, Xiang and Wang, Shusen and Chen, Kun and Zhang, Zhihua},
  booktitle={International Conference on Machine Learning},
  pages={6504--6514},
  year={2021},
  organization={PMLR}
}

The Algorithm: LocalPower

image

Table of Contents

File name Description
pm.py Main functions for various algorithms
one-shot.py Code for Table 1
eta.py Code for Table 3, 5
plot_p.py Code for Figure 2 and 5-7
plot_decay.py Code for Table 7 and Figure 8-10
plot_m.py Code for Figure 11-12
plot_sin.py Code for Figure 13

About

Codes for LocalPower


Languages

Language:Python 100.0%