Mercury: Efficient On-Device Distributed DNN Training via Stochastic Importance Sampling
Abstract
As intelligence is moving from data centers to the edges, intelligent edge devices such as smartphones, drones, robots, and smart IoT devices are equipped with the capability to altogether train a deep learning model on the devices from the data collected by themselves. Despite its considerable value, the key bottleneck of making on-device distributed training practically useful in real-world deployments is that they consume a significant amount of training time under wireless networks with constrained bandwidth. To tackle this critical bottleneck, we present Mercury, an importance sampling-based framework that enhances the training efficiency of on-device distributed training without compromising the accuracies of the trained models. The key idea behind the design of Mercury is to focus on samples that provide more important information in each training iteration. In doing this, the training efficiency of each iteration is improved. As such, the total number of iterations can be considerably reduced so as to speed up the overall training process. We implemented Mercury and deployed it on a self-developed testbed. We demonstrate its effectiveness and show that Mercury consistently outperforms two status quo frameworks on six commonly used datasets across tasks in image classification, speech recognition, and natural language processing.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
Mercury proposed a sampling-based framework that enables efficient on-device distributed training without compromising the training quality.
“Mercury [7] proposed a sampling-based framework that enables efficient on-device distributed training without compromising the training quality as a new inspiration for solving this challenge.”
From Federated learning for the Internet of Things: Applications, challenges, and opportunities (2022) · cites this paper as [7]
Cites
Deep Residual Learning for Image Recognition
He, K., Zhang, X., Ren, S., & Sun, J. (2016) · 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr.2016.90 · Cited by 235,474 · see citing works
Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A. N., Kaiser, L., & Polosukhin, I. (2017) · arXiv · doi:10.48550/arxiv.1706.03762 · Cited by 120,404 · see citing works
Hochreiter, S., & Schmidhuber, J. (1997) · Neural Computation · doi:10.1162/neco.1997.9.8.1735 · Cited by 102,318 · see citing works
ImageNet: A large-scale hierarchical image database
Deng, J., Dong, W., Socher, R., Li, L.-J., Li, K., & Fei-Fei, L. (2009) · 2009 IEEE Conference on Computer Vision and Pattern Recognition · doi:10.1109/cvpr.2009.5206848 · Cited by 65,333 · see citing works
MobileNetV2: Inverted Residuals and Linear Bottlenecks
Sandler, M., Howard, A., Zhu, M., Zhmoginov, A., & Chen, L.-C. (2018) · 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition · doi:10.1109/cvpr.2018.00474 · Cited by 28,755 · see citing works
Data Mining: Practical Machine Learning Tools and Techniques
Witten, I. H., Frank, E., & Hall, M. A. (2011) · Elsevier eBooks · doi:10.1016/c2009-0-19715-5 · Cited by 25,493 · see citing works
Learning Multiple Layers of Features from Tiny Images
Krizhevsky, A. (2024) · doi:10.57702/zp44cu3g · Cited by 23,890 · see citing works
Communication-Efficient Learning of Deep Networks from Decentralized Data
McMahan, H. B., Moore, E., Ramage, D., Hampson, S., & Arcas, B. A. Y. (2016) · arXiv · doi:10.48550/arxiv.1602.05629 · Cited by 14,215 · see citing works
Hinton, G., Deng, L., Yu, D., Dahl, G., Mohamed, A.-R., Jaitly, N., Senior, A., Vanhoucke, V., Nguyen, P., Sainath, T., et al. (2012) · IEEE Signal Processing Magazine · doi:10.1109/msp.2012.2205597 · Cited by 10,260 · see citing works
Bengio, Y., Louradour, J., Collobert, R., & Weston, J. (2009) · Proceedings of the 26th Annual International Conference on Machine Learning · doi:10.1145/1553374.1553380 · Cited by 5,366 · cited 1× · see citing works
“Importance sampling shares similarities with curriculum learning [4] in terms of selecting important data to generate mini-batch for training.”
BackgroundAID: A Benchmark Data Set for Performance Evaluation of Aerial Scene Classification
Xia, G.-S., Hu, J., Hu, F., Shi, B., Bai, X., Zhong, Y., Zhang, L., & Lu, X. (2017) · IEEE Transactions on Geoscience and Remote Sensing · doi:10.1109/tgrs.2017.2685945 · Cited by 2,423 · see citing works
Deep Neural Networks for Acoustic Modeling in Speech Recognition
Hinton, G. E., Deng, L., Yu, D., Dahl, G. E., Mohamed, A., Jaitly, N., Senior, A., Vanhoucke, V., Nguyen, P., Sainath, T. N., et al. (2012) · Cited by 1,839 · see citing works
QSGD: Communication-Efficient SGD via Gradient Quantization and Encoding
Alistarh, D., Grubic, D., Li, J. Z., Tomioka, R., & Vojnović, M. (2016) · London School of Economics and Political Science Research Online (London School of Economics and Political Science) · doi:10.48550/arxiv.1610.02132 · Cited by 1,189 · see citing works
Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Training
Lin, Y., Han, S., Mao, H., Wang, Y., & Dally, W. J. (2017) · arXiv · doi:10.48550/arxiv.1712.01887 · Cited by 1,134 · see citing works
Scaling Distributed Machine Learning with the Parameter Server
Li, M. (2014) · International Conference on Big Data Science and Computing (BigDataScience) · doi:10.1145/2640087.2644155 · Cited by 1,000 · see citing works
TernGrad: Ternary Gradients to Reduce Communication in Distributed Deep LearningPreprint
Wen, W., Xu, C., Yan, F., Wu, C., Wang, Y., Chen, Y., & Li, H. H. (2017) · arXiv (Cornell University) · doi:10.48550/arxiv.1705.07878 · Cited by 718 · see citing works
Poseidon: An Efficient Communication Architecture for Distributed Deep Learning on GPU ClustersPreprint
Zhang, H., Zheng, Z., Xu, S.-Z., Dai, W., Ho, Q.-R., Liang, X., Hu, Z., Wei, J.-L., Xie, P.-T., & Xing, E. (2017) · USENIX Annual Technical Conference · doi:10.48550/arxiv.1706.03292 · Cited by 390 · see citing works
Petuum: A New Platform for Distributed Machine Learning on Big Data
Xing, E. P., Ho, Q., Dai, W., Kim, J. K., Wei, J., Lee, S., Zheng, X., Xie, P., Kumar, A., & Yu, Y. (2015) · IEEE Transactions on Big Data · doi:10.1109/tbdata.2015.2472014 · Cited by 388 · see citing works
GeePS: scalable deep learning on distributed GPUs with a GPU-specialized parameter server
Cui, H.-G., Zhang, H., Ganger, G., Gibbons, P. B., & Xing, E. (2016) · European Conference on Computer Systems · doi:10.1145/2901318.2901323 · Cited by 339 · see citing works
NestDNN: Resource-Aware Multi-Tenant On-Device Deep Learning for Continuous Mobile Vision
Fang, B., Zeng, X., & Zhang, M. (2018) · Proceedings of the 24th Annual International Conference on Mobile Computing and Networking · doi:10.1145/3241539.3241559 · Cited by 291 · see citing works
Self-citationStochastic Optimization with Importance Sampling for Regularized Loss Minimization
Zhao, P., & Zhang, T. (2015) · Rare & Special e-Zone (The Hong Kong University of Science and Technology) · Cited by 197 · see citing works
Gradient Sparsification for Communication-Efficient Distributed Optimization
Wangni, J., Wang, J., Liu, J., & Zhang, T. (2018) · Rare & Special e-Zone (The Hong Kong University of Science and Technology) · Cited by 166 · see citing works
Pipe-SGD: A Decentralized Pipelined SGD Framework for Distributed Deep Net TrainingPreprint
Li, Y., Yu, M.-C., Li, S.-Z., Avestimehr, A., Kim, N., & Schwing, A. (2018) · Neural Information Processing Systems · doi:10.48550/arxiv.1811.03619 · Cited by 112 · see citing works
Zeng, X., Fang, B., Shen, H., & Zhang, M. (2020) · ACM Conference on Embedded Networked Sensor Systems (SenSys) · doi:10.1145/3384419.3430721 · Cited by 102 · see citing works
Self-citationConvergence analysis of distributed stochastic gradient descent with shuffling
Meng, Q., Chen, W., Wang, Y., Ma, Z.-M., & Liu, T. (2019) · Neurocomputing · doi:10.1016/j.neucom.2019.01.037 · Cited by 99 · see citing works
Watcharapichat, P., Morales, V. L., Fernandez, R. C., & Pietzuch, P. R. (2016) · ACM Symposium on Cloud Computing (SoCC) · doi:10.1145/2987550.2987586 · Cited by 76 · see citing works
Deep Learning in the Era of Edge Computing: Challenges and Opportunities
Zhang, M., Zhang, F., Lane, N. D., Shu, Y., Zeng, X., Fang, B., Shen, Y., & Xu, H. F. (2020) · Wiley eBooks · doi:10.1002/9781119551713.ch3 · Cited by 57 · see citing works
Self-citationBiased Importance Sampling for Deep Neural Network TrainingPreprint
Katharopoulos, A., & Fleuret, F. (2017) · arXiv (Cornell University) · doi:10.48550/arxiv.1706.00043 · Cited by 46 · see citing works
Jiang, S., Ma, Z., Zeng, X., Xu, C., Zhang, M., Zhang, C., & Liu, Y. (2020) · IEEE Conference on Computer Communications (INFOCOM) · doi:10.1109/infocom41043.2020.9155435 · Cited by 24 · see citing works
Self-citation
Flagged first, then by citations
3 of the 32 references couldn’t be matched to a known paper.
Cited by
Sorted by citations
Federated learning for the Internet of Things: Applications, challenges, and opportunities
Zhang, T., Gao, L., He, C., Zhang, M., Krishnamachari, B., & Avestimehr, A. S. (2022) · IEEE Internet of Things Magazine · doi:10.1109/iotm.004.2100182 · Cited by 412 · see citing works
Self-citationClient Selection in Federated Learning: Principles, Challenges, and Opportunities
Fu, L., Zhang, H., Gao, G., Zhang, M., & Liu, X. (2023) · IEEE Internet of Things Journal · doi:10.1109/jiot.2023.3299573 · Cited by 310 · see citing works
Self-citationLi, C., Zeng, X., Zhang, M., & Cao, Z. (2022) · ACM Annual International Conference on Mobile Computing and Networking (MobiCom) · doi:10.1145/3495243.3517017 · Cited by 191 · see citing works
Self-citationEdge Video Analytics: A Survey on Applications, Systems and Enabling Techniques
Xu, R., Razavi, S., & Zheng, R. (2023) · IEEE Communications Surveys & Tutorials · doi:10.1109/comst.2023.3323091 · Cited by 109 · see citing works
A Survey of Trustworthy Federated Learning: Issues, Solutions, and Challenges
Zhang, Y., Zeng, D., Luo, J., Fu, X., Chen, G., Xu, Z., & King, I. (2024) · ACM Transactions on Intelligent Systems and Technology · doi:10.1145/3678181 · Cited by 90 · see citing works
Artificial Intelligence of Things: A Survey
Siam, M. S. I., Ahn, H., Liu, L., Alam, S., Shen, H., Cao, Z., Shroff, N. B., Krishnamachari, B., Srivastava, M. B., & Zhang, M. (2024) · ACM Transactions on Sensor Networks · doi:10.1145/3690639 · Cited by 90 · see citing works
Self-citationTo Store or Not? Online Data Selection for Federated Learning with Limited Storage
Gong, C., Zheng, Z.-Z., Wu, F., Shao, Y.-F., Li, B.-S., & Chen, G.-H. (2022) · The Web Conference · doi:10.1145/3543507.3583426 · Cited by 33 · see citing works
Addressing Heterogeneity in Federated Learning with Client Selection via Submodular Optimization
Zhang, J., Wang, J., Li, Y., Xin, F., Dong, F., Luo, J., & Wu, Z. (2023) · ACM Transactions on Sensor Networks · doi:10.1145/3638052 · Cited by 27 · see citing works
Participant and Sample Selection for Efficient Online Federated Learning in UAV Swarms
Wu, F., Qu, Y., Wu, T., Dong, C., Guo, K., Wu, Q., & Guo, S. (2024) · IEEE Internet of Things Journal · doi:10.1109/jiot.2024.3363181 · Cited by 24 · see citing works
Liu, W., Zhang, X., Duan, J., Joe‐Wong, C., Zhou, Z., & Chen, X. (2023) · IEEE Transactions on Mobile Computing · doi:10.1109/tmc.2023.3337016 · Cited by 24 · see citing works
Su, Y. R., Li, J., An, S., Xu, H., & Peng, W. (2025) · IEEE Transactions on Industrial Informatics · doi:10.1109/tii.2025.3574406 · Cited by 21 · see citing works
Client Selection for Wireless Federated Learning With Data and Latency Heterogeneity
Chen, X., Zhou, X., Zhang, H., Sun, M., & Poor, H. V. (2024) · IEEE Internet of Things Journal · doi:10.1109/jiot.2024.3425757 · Cited by 20 · see citing works
InArt: In-Network Aggregation with Route Selection for Accelerating Distributed Training
Liu, J., Zhai, Y., Zhao, G., Xu, H., Fang, J., Zeng, Z., & Zhu, Y. (2024) · International World Wide Web Conference (WWW) · doi:10.1145/3589334.3645394 · Cited by 18 · see citing works
Federated Learning: Challenges, SoTA, Performance Improvements and Application Domains
Schoinas, Ι., Triantafyllou, A., Ioannidis, D., Tzovaras, D. K., Drosou, A., Votis, K., Λάγκας, Θ., Argyriou, V., & Sarigiannidis, P. G. (2024) · IEEE Open Journal of the Communications Society · doi:10.1109/ojcoms.2024.3458088 · Cited by 10 · see citing works
ODE: An Online Data Selection Framework for Federated Learning With Limited Storage
Gong, C., Zheng, Z., Shao, Y., Li, B., Wu, F., & Chen, G. (2024) · IEEE/ACM Transactions on Networking · doi:10.1109/tnet.2024.3365534 · Cited by 9 · see citing works
Efficient one-shot Neural Architecture Search with progressive choice freezing evolutionary search
Zhang, C., Wan, Q., Wang, L., Wen, Y., Chen, M., Tan, J., Yan, K., & Fu, X. (2024) · Neurocomputing · doi:10.1016/j.neucom.2024.127702 · Cited by 9 · see citing works
Online Stream Sampling for Low-Memory On-Device Edge Training for WiFi Sensing
Hernandez, S. M., & Bulut, E. (2022) · ACM Workshop on Wireless Security and Machine Learning (WiseML) · doi:10.1145/3522783.3529521 · Cited by 8 · see citing works
Wang, Z., Liu, S., Guo, B., Yu, Z., & Zhang, D. (2024) · IEEE Transactions on Mobile Computing · doi:10.1109/tmc.2024.3427636 · Cited by 6 · see citing works
Dong, C., Wu, F., Qu, Y., Akerkar, R. A., Wu, T., Zhou, F., Wu, Q., & Guo, S. (2024) · IEEE Transactions on Cognitive Communications and Networking · doi:10.1109/tccn.2024.3520557 · Cited by 6 · see citing works
Patching in Order: Efficient On-Device Model Fine-Tuning for Multi-DNN Vision Applications
Cao, Z., Cheng, Y., Zhou, Z., Lu, A., Hu, Y., Liu, J., Zhang, M., & Li, Z. (2024) · IEEE Transactions on Mobile Computing · doi:10.1109/tmc.2024.3443057 · Cited by 5 · see citing works
Self-citationFLightNER: A Federated Learning Approach to Lightweight Named-Entity Recognition
Abadeer, M., Shi, W., & Corriveau, J. (2022) · IEEE International Conference on Trust, Security and Privacy in Computing and Communications (TrustCom) · doi:10.1109/trustcom56396.2022.00098 · Cited by 4 · see citing works
Joint Device and Training Scheduling for Wireless Federated Learning
Chen, X., Zhou, X., Zhang, H., Sun, M., & Zhao, T. (2025) · IEEE Internet of Things Journal · doi:10.1109/jiot.2025.3564461 · Cited by 3 · see citing works
Improving local training in federated learning via temperature scaling
Lee, K., Zhang, P., Kim, S., & Ko, J. G. (2025) · Advanced Engineering Informatics · doi:10.1016/j.aei.2025.104283 · Cited by 3 · see citing works
GreedW: A Flexible and Efficient Decentralized Framework for Distributed Machine Learning
Wang, T., Jiang, X., Li, Q., & Cai, H. (2023) · IEEE Transactions on Computers · doi:10.1109/tc.2023.3343110 · Cited by 3 · see citing works
Accelerating On-Device DNN Training Workloads via Runtime Convergence Monitor
Choi, S., Shin, J., & Kim, L. (2022) · IEEE Transactions on Computer-Aided Design of Integrated Circuits and Systems · doi:10.1109/tcad.2022.3206394 · Cited by 3 · see citing works
Trustworthy Distributed Deep Neural Network Training in an Edge Device Network
Shubha, S. S., & Shen, H. (2022) · IEEE International Conference on Big Data (BigData) · doi:10.1109/bigdata55660.2022.10021055 · Cited by 3 · see citing works
PieBridge: Fast and Parameter-Efficient On-Device Training via Proxy Networks
Yin, W., Xu, D., Huang, G., Zhang, Y., Wei, S., Xu, M., & Liu, X. (2024) · ACM Conference on Embedded Networked Sensor Systems (SenSys) · doi:10.1145/3666025.3699327 · Cited by 1 · see citing works
HeaPS: Heterogeneity-aware participant selection for efficient federated learning
Yang, D., Hu, B., Gao, Y., Jin, A.-L., Liu, A., Yeung, K. L., & You, Y. (2025) · Journal of Parallel and Distributed Computing · doi:10.1016/j.jpdc.2025.105168 · Cited by 1 · see citing works
GeoFL: A Framework for Efficient Geo-Distributed Cross-Device Federated Learning
Gan, M., 李澜鹏, Alam, S., Liu, L., Liu, L., Zhang, M., Zeng, H., & Cao, Z. (2026) · IEEE Transactions on Networking · doi:10.1109/ton.2026.3668183 · Cited by 1 · see citing works
Self-citationEEOKD: Energy-Efficient Online Knowledge Distillation for Mobile Video Inference
Guo, G., & Zhang, J. (2023) · International Conference on Communications · doi:10.1109/icc45041.2023.10278833 · Cited by 1 · see citing works
FedCaSe: Enhancing Federated Learning with Heterogeneity-aware Caching and Scheduling
Khan, R. I. S., Paul, A. K., Cheng, Y., Jian, X., & Butt, A. R. (2024) · ACM Symposium on Cloud Computing (SoCC) · doi:10.1145/3698038.3698559 · Cited by 1 · see citing works
Online Federated Learning on Distributed Unknown Data Using UAVs
Zhang, X., Xu, H., Yin, X., Xiao, M., Wu, J., & Zhou, J. (2025) · IEEE International Conference on Data Engineering (ICDE) · doi:10.1109/icde65448.2025.00009 · Cited by 1 · see citing works
Data-Quality-based Aggregation Methods in Federated Learning: a Comprehensive Study
Boumediri, T., Djamaa, B., & Boulahia, S. Y. (2025) · International Conference on Pattern Analysis and Intelligent Systems (PAIS) · doi:10.1109/pais66004.2025.11126046 · Cited by 1 · see citing works
ACache: An Importance-Based Cache with Augmentation Cutting for Accelerating DNN Training
Zhao, L., Zheng, X., & Wang, F. (2025) · International Conference on Computer Supported Cooperative Work in Design (CSCWD) · doi:10.1109/cscwd64889.2025.11033388
Data Divergence-Aware Client Selection via Knowledge Graph for Federated LLM Fine-Tuning
Zhang, B., Wang, D., Zhu, Y., & Han, Z. (2025) · IEEE Transactions on Mobile Computing · doi:10.1109/tmc.2025.3608172
Federated Learning in the Healthcare Industry
Mondal, R. (2025) · Studies in systems, decision and control · doi:10.1007/978-3-031-99270-4_2
CrowdMesh: A Dynamic Model Parallel Training System on Mobile Devices
Wang, Z., Guo, B., Liu, S., Yu, Z., & Zhang, D. (2025) · IEEE Internet of Things Journal · doi:10.1109/jiot.2025.3623109
Nu, C., Zhang, Z., Li, Y., & Zhao, Y. (2025) · Proceedings of the International Conference on Parallel Processing · doi:10.1145/3754598.3754659
Energy-efficient online knowledge distillation for mobile video inference
Guo, G., Zhang, J., & Liu, B. (2025) · Computer Networks · doi:10.1016/j.comnet.2025.111962
ENTS: Experiences in Co-Designed Environmental Sensing
Madden, J., Taylor, S., Jaliff, L., Wu, A. X., Levy, A., Lin, J., Falah, A., Potyondy, T., Hester, J., Wells, G., et al. (2026) · doi:10.1145/3774906.3802780
RACER: A Caching Layer based Optimization for Accelerating Reinforcement Learning Workloads
Gogineni, K., Mei, Y., Gogineni, K., Wei, P., Lan, T., & Venkataramani, G. (2026) · IEEE International Symposium on Performance Analysis of Systems and Software (ISPASS) · doi:10.1109/ispass69572.2026.00051
PINCH: Predictive Importance-Sampling-Informed Cache for I/O-Bound DNN Training
Liu, Y., & Yin, S. (2026) · Proceedings of the International Conference on Parallel Processing · doi:10.1145/3832810.3832851
- Authors
- Zeng, X., Yan, M., & Zhang, M.
- Affiliations
- Michigan State University (US)
- Venue
- Proceedings of the 19th ACM Conference on Embedded Networked Sensor Systems
- Publisher
- ACM
- Published
- 2021-11-15
- Pages
- 29-41
- OpenAlex
- W3212514695
- Cited by
- 43
- References
- 32
- Citations by year
- 2026: 4 · 2025: 12 · 2024: 13 · 2023: 7 · 2022: 7
- FWCI
- 6.24
- Type
- conference paper
- Language
- en
- Open access
- closed
- Locations
- ACM Conference on Embedded Networked Sensor Systems (SenSys) (publishedVersion)
- Funders
- NSF (National Science Foundation) (CNS-1617627, CNS-1814551, DMS-2012439, PFI:BIC-1632051)
- Topics
- Physical Sciences › Computer Science › Computer Networks and Communications › IoT and Edge/Fog Computing
- Physical Sciences › Computer Science › Artificial Intelligence › Privacy-Preserving Technologies in Data
- Physical Sciences › Computer Science › Artificial Intelligence › Stochastic Gradient Optimization Techniques
- Keywords
- importance sampling, edge computing, deep learning training efficiency, wireless networks
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review