Energy-Efficient Power Control for Multiple-Task Split Inference in UAVs: A Tiny Learning-Based Approach (2401.00445v1)
Abstract: The limited energy and computing resources of unmanned aerial vehicles (UAVs) hinder the application of aerial artificial intelligence. The utilization of split inference in UAVs garners significant attention due to its effectiveness in mitigating computing and energy requirements. However, achieving energy-efficient split inference in UAVs remains complex considering of various crucial parameters such as energy level and delay constraints, especially involving multiple tasks. In this paper, we present a two-timescale approach for energy minimization in split inference, where discrete and continuous variables are segregated into two timescales to reduce the size of action space and computational complexity. This segregation enables the utilization of tiny reinforcement learning (TRL) for selecting discrete transmission modes for sequential tasks. Moreover, optimization programming (OP) is embedded between TRL's output and reward function to optimize the continuous transmit power. Specifically, we replace the optimization of transmit power with that of transmission time to decrease the computational complexity of OP since we reveal that energy consumption monotonically decreases with increasing transmission time. The replacement significantly reduces the feasible region and enables a fast solution according to the closed-form expression for optimal transmit power. Simulation results show that the proposed algorithm can achieve a higher probability of successful task completion with lower energy consumption.
- N. H. Motlagh, M. Bagaa, and T. Taleb, “Uav-based iot platform: A crowd surveillance use case,” IEEE Communications Magazine, vol. 55, no. 2, pp. 128–134, 2017.
- Y. Tan, J. Liu, and N. Kato, “Blockchain-based lightweight authentication for resilient UAV communications: Architecture, scheme, and future directions,” IEEE Wireless Communications, vol. 29, no. 3, pp. 24–31, 2022.
- D. Zhou, M. Sheng, B. Li, J. Li, and Z. Han, “Distributionally robust planning for data delivery in distributed satellite cluster network,” IEEE Transactions on Wireless Communications, vol. 18, no. 7, pp. 3642–3657, 2019.
- R. Liu, M. Sheng, K.-S. Lui, X. Wang, Y. Wang, and D. Zhou, “An analytical framework for resource-limited small satellite networks,” IEEE Communications Letters, vol. 20, no. 2, pp. 388–391, 2016.
- J. Shao and J. Zhang, “Communication-computation trade-off in resource-constrained edge inference,” IEEE Communications Magazine, vol. 58, no. 12, pp. 20–26, 2020.
- X. Liu, Y. Deng, and T. Mahmoodi, “Wireless distributed learning: A new hybrid split and federated learning approach,” IEEE Transactions on Wireless Communications, vol. 22, no. 4, pp. 2650–2665, 2023.
- T. Sun, X. Wang, M. Umehira, and Y. Ji, “Split learning assisted multi-UAV system for image classification task,” in 2023 IEEE 97th Vehicular Technology Conference (VTC2023-Spring), 2023, pp. 1–6.
- E. Li, L. Zeng, Z. Zhou, and X. Chen, “Edge AI: On-demand accelerating deep neural network inference via edge computing,” IEEE Transactions on Wireless Communications, vol. 19, no. 1, pp. 447–457, 2020.
- B. Atan, M. Basaran, N. Calik, S. T. Basaran, G. Akkuzu, and L. Durak-Ata, “AI-empowered fast task execution decision for delay-sensitive IoT applications in edge computing networks,” IEEE Access, vol. 11, pp. 1324–1334, 2023.
- S. Naveen, M. R. Kounte, and M. R. Ahmed, “Low latency deep learning inference model for distributed intelligent IoT edge clusters,” IEEE Access, vol. 9, pp. 160 607–160 621, 2021.
- X. Zhou, Y. Gao, C. Li, and Z. Huang, “A multiple gradient descent design for multi-task learning on edge computing: Multi-objective machine learning approach,” IEEE Transactions on Network Science and Engineering, vol. 9, no. 1, pp. 121–133, 2022.
- K. Guo, M. Sheng, J. Tang, T. Q. S. Quek, and Z. Qiu, “Exploiting hybrid clustering and computation provisioning for green c-ran,” IEEE Journal on Selected Areas in Communications, vol. 34, no. 12, pp. 4063–4076, 2016.
- X. Gu, G. Zhang, M. Wang, W. Duan, M. Wen, and P.-H. Ho, “Uav-aided energy-efficient edge computing networks: Security offloading optimization,” IEEE Internet of Things Journal, vol. 9, no. 6, pp. 4245–4258, 2022.
- K. Zeng, X. Li, and T. Shen, “Energy-stabilized computing offloading algorithm for uavs with energy harvesting,” IEEE Internet of Things Journal, pp. 1–1, 2023.
- F. Zhou, Y. Wu, R. Q. Hu, and Y. Qian, “Computation rate maximization in uav-enabled wireless-powered mobile-edge computing systems,” IEEE Journal on Selected Areas in Communications, vol. 36, no. 9, pp. 1927–1941, 2018.
- A. M. Seid, J. Lu, H. N. Abishu, and T. A. Ayall, “Blockchain-enabled task offloading with energy harvesting in multi-uav-assisted iot networks: A multi-agent drl approach,” IEEE Journal on Selected Areas in Communications, vol. 40, no. 12, pp. 3517–3532, 2022.
- D. Zhai, M. Sheng, X. Wang, Y. Li, J. Song, and J. Li, “Rate and energy maximization in scma networks with wireless information and power transfer,” IEEE Communications Letters, vol. 20, no. 2, pp. 360–363, 2016.
- Q. Lan, Q. Zeng, P. Popovski, D. GÃŒndÃŒz, and K. Huang, “Progressive feature transmission for split classification at the wireless edge,” IEEE Transactions on Wireless Communications, vol. 22, no. 6, pp. 3837–3852, 2023.
- X. Huang and S. Zhou, “Dynamic compression ratio selection for edge inference systems with hard deadlines,” IEEE Internet of Things Journal, vol. 7, no. 9, pp. 8800–8810, 2020.
- J. Yan, S. Bi, and Y.-J. A. Zhang, “Optimal model placement and online model splitting for device-edge co-inference,” IEEE Transactions on Wireless Communications, vol. 21, no. 10, pp. 8354–8367, 2022.
- Z. Hao, G. Xu, Y. Luo, H. Hu, J. An, and S. Mao, “Multi-agent collaborative inference via DNN decoupling: Intermediate feature compression and edge learning,” IEEE Transactions on Mobile Computing, vol. 22, no. 10, pp. 6041–6055, 2023.
- Y.-T. Yang and H.-Y. Wei, “Edge-IoT computing and networking resource allocation for decomposable deep learning inference,” IEEE Internet of Things Journal, vol. 10, no. 6, pp. 5178–5193, 2023.
- Y. Sun, D. Xu, D. W. K. Ng, L. Dai, and R. Schober, “Optimal 3D-trajectory design and resource allocation for solar-powered uav communication systems,” IEEE Transactions on Communications, vol. 67, no. 6, pp. 4281–4298, 2019.
- Y. Wang, M. Sheng, X. Wang, L. Wang, and J. Li, “Mobile-edge computing: Partial computation offloading using dynamic voltage scaling,” IEEE Transactions on Communications, vol. 64, no. 10, pp. 4268–4282, 2016.
- H. v. Hasselt, A. Guez, and D. Silver, “Deep reinforcement learning with double Q-learning,” in Proceedings of the Thirtieth AAAI Conference on Artificial Intelligence, ser. AAAI’16, 2016, pp. 2094–2100.
- S. K. Kumar, “On weight initialization in deep neural networks,” arXiv preprint arXiv:1704.08863, 2017.
- A. M.-C. So and Y. J. A. Zhang, “Distributionally robust slow adaptive OFDMA with soft qos via linear programming,” IEEE J. Sel. Areas Commun., vol. 31, no. 5, pp. 947–958, 2013.
- P. Yang, X. Xi, Y. Fu, T. Q. Quek, X. Cao, and D. Wu, “Multicast eMBB and bursty URLLC service multiplexing in a CoMP-enabled RAN,” IEEE Trans. Wireless Commun., vol. 20, no. 5, pp. 3061–3077, 2021.
- S. Boyd, N. Parikh, E. Chu, B. Peleato, J. Eckstein et al., “Distributed optimization and statistical learning via the alternating direction method of multipliers,” Foundations and Trends® in Machine learning, vol. 3, no. 1, pp. 1–122, 2011.
- J. Tang, B. Shim, and T. Q. Quek, “Service multiplexing and revenue maximization in sliced C-RAN incorporated with URLLC and multicast eMBB,” IEEE J. Sel. Areas Commun., vol. 37, no. 4, pp. 881–895, 2019.
- H. Tabrizi, B. Peleato, G. Farhadi, J. M. Cioffi, and G. Aldabbagh, “Spatial reuse in dense wireless areas: A cross-layer optimization approach via ADMM,” IEEE Trans. Wireless Commun., vol. 14, no. 12, pp. 7083–7095, 2015.
Sponsored by Paperpile, the PDF & BibTeX manager trusted by top AI labs.
Get 30 days freePaper Prompts
Sign up for free to create and run prompts on this paper using GPT-5.
Top Community Prompts
Collections
Sign up for free to add this paper to one or more collections.