ML Anthology

Niu, Yue

6 publications

CVPR 2024 All Rivers Run to the Sea: Private Learning with Asymmetric Flows Yue Niu, Ramy E. Ali, Saurav Prakash, Salman Avestimehr
TMLR 2023 Overcoming Resource Constraints in Federated Learning: Large Models Can Be Trained with Only Weak Clients Yue Niu, Saurav Prakash, Souvik Kundu, Sunwoo Lee, Salman Avestimehr
TMLR 2023 Revisiting Sparsity Hunting in Federated Learning: Why Does Sparsity Consensus Matter? Sara Babakniya, Souvik Kundu, Saurav Prakash, Yue Niu, Salman Avestimehr
TMLR 2023 mL-BFGS: A Momentum-Based L-BFGS for Distributed Large-Scale Neural Network Optimization Yue Niu, Zalan Fabian, Sunwoo Lee, Mahdi Soltanolkotabi, Salman Avestimehr
NeurIPSW 2022 Federated Learning of Large Models at the Edge via Principal Sub-Model Training Yue Niu, Saurav Prakash, Souvik Kundu, Sunwoo Lee, Salman Avestimehr
NeurIPSW 2022 Federated Sparse Training: Lottery Aware Model Compression for Resource Constrained Edge Sara Babakniya, Souvik Kundu, Saurav Prakash, Yue Niu, Salman Avestimehr