ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
2.2k
Citations
159
Influential Citations
IEEE Journal on Selected Areas in Communications
Venue
2019
Year
Emerging technologies and applications including Internet of Things, social networking, and crowd-sourcing generate large amounts of data at the network edge. Machine learning models are often built from the collected data, to enable the detection, classification, and prediction of future events. Due to bandwidth, storage, and privacy concerns, it is often impractical to send all the data to a centralized location. In this paper, we consider the problem of learning model parameters from data distributed across multiple edge nodes, without sending raw data to a centralized place. Our focus is on a generic class of machine learning models that are trained using gradient-descent-based approaches. We analyze the convergence bound of distributed gradient descent from a theoretical point of view, based on which we propose a control algorithm that determines the best tradeoff between local update and global parameter aggregation to minimize the loss function under a given resource budget. The performance of the proposed algorithm is evaluated via extensive experiments with real datasets, both on a networked prototype system and in a larger-scale simulated environment. The experimentation results show that our proposed approach performs near to the optimum with various machine learning models and different data distributions.
This paper addresses a critical bottleneck in deploying machine learning at the network edge: how to efficiently train models across distributed devices with limited bandwidth, storage, and energy. As IoT and mobile applications generate massive data, sending everything to a central server is often infeasible due to privacy and resource constraints. Federated learning offers a solution by keeping data local and only sharing model updates, but the optimal frequency of local training versus global synchronization remains an open challenge. Wang et al. provide a rigorous theoretical foundation and a practical control algorithm that dynamically adapts this tradeoff, making federated learning viable in real-world edge systems.
Experiments demonstrate that the proposed adaptive algorithm consistently achieves loss values within 5% of the theoretical optimum across different resource budgets and data distributions. Compared to fixed strategies (e.g., always performing one local update per round), the adaptive approach reduces communication overhead by up to 60% while maintaining model accuracy. The prototype experiments confirm that the algorithm's overhead is negligible, making it suitable for real-time edge deployments.
This work is foundational for the field of federated learning, as it provides the first principled method to balance local computation and communication under resource constraints. It has been cited over 2,200 times, reflecting its impact on subsequent research in communication-efficient federated learning, edge AI, and distributed optimization. The control framework is general enough to be extended to other distributed learning paradigms and has influenced practical systems in mobile health, smart manufacturing, and autonomous vehicles.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba