3 papers
cs.LG2023
ElasticTrainer: Speeding Up On-Device Training with Runtime Elastic Tensor Selection
Kai Huang, Boyuan Yang, Wei Gao
On-device training is essential for neural networks (NNs) to continuously adapt to new online data, but can be time-consuming due to the device's limited computing power. To speed…
cs.LG2023
Tackling Intertwined Data and Device Heterogeneities in Federated Learning with Unlimited Staleness
Haoming Wang, Wei Gao
Federated Learning (FL) can be affected by data and device heterogeneities, caused by clients' different local data distributions and latencies in uploading model updates (i.e., st…
cs.AI2022
Context-Hierarchy Inverse Reinforcement Learning
Wei Gao, David Hsu, Wee Sun Lee
An inverse reinforcement learning (IRL) agent learns to act intelligently by observing expert demonstrations and learning the expert's underlying reward function. Although learning…