Weiting Zhang
Current Organizations / Institutes
The organizations/institutes from the scholar's latest article
Research Subjects
Weiting Zhang has a total of 35 co-authors. Here are the top co-authors by number of shared publications:
- Dong Yang
School of Electronics and Information Engineering, Beijing Jiaotong University, Beijing, China
7 shared publications - Hongchao Wang
State Key Laboratory of Advanced Rail Autonomous Operation, Beijing Jiaotong University, Beijing, China
4 shared publications - Wen Wu
Peng Cheng Laboratory, Shenzhen, China
4 shared publications - Mikael Gidlund
Mid Sweden University
3 shared publications - Peng Yang
Huazhong University of Science and Technology
3 shared publications - Xuemin Shen
University of Waterloo
3 shared publications
Latest Publications
- Open Access
Accuracy-Guaranteed Collaborative DNN Inference in Industrial IoT via Deep Reinforcement Learning
Collaboration among industrial Internet of Things (IoT) devices and edge networks is essential to support computation-intensive deep neural network (DNN) inference services which require low delay and high accuracy. Sampling rate adaption which dynamically configures the sampling rates of industrial IoT devices according to network conditions, is the key in minimizing the service delay. In this paper, we investigate the collaborative DNN inference problem in industrial IoT networks. To capture the channel variation and task arrival randomness, we formulate the problem as a constrained Markov decision process (CMDP). Specifically, sampling rate adaption, inference task offloading and edge computing resource allocation are jointly considered to minimize the average service delay while guaranteeing the long-term accuracy requirements of different inference services. Since CMDP cannot be directly solved by general reinforcement learning (RL) algorithms due to the intractable long-term constraints, we first transform the CMDP into an MDP by leveraging the Lyapunov optimization technique. Then, a deep RL-based algorithm is proposed to solve the MDP. To expedite the training process, an optimization subroutine is embedded in the proposed algorithm to directly obtain the optimal edge computing resource allocation. Extensive simulation results are provided to demonstrate that the proposed RL-based algorithm can significantly reduce the average service delay while preserving long-term inference accuracy with a high probability.
Open AccessAccuracy-Guaranteed Collaborative DNN Inference in Industrial IoT via Deep Reinforcement Learning
Collaboration among industrial Internet of Things (IoT) devices and edge networks is essential to support computation-intensive deep neural network (DNN) inference services which require low delay and high accuracy. Sampling rate adaption which dynamically configures the sampling rates of industrial IoT devices according to network conditions, is the key in minimizing the service delay. In this paper, we investigate the collaborative DNN inference problem in industrial IoT networks. To capture the channel variation and task arrival randomness, we formulate the problem as a constrained Markov decision process (CMDP). Specifically, sampling rate adaption, inference task offloading and edge computing resource allocation are jointly considered to minimize the average service delay while guaranteeing the long-term accuracy requirements of different inference services. Since CMDP cannot be directly solved by general reinforcement learning (RL) algorithms due to the intractable long-term constraints, we first transform the CMDP into an MDP by leveraging the Lyapunov optimization technique. Then, a deep RL-based algorithm is proposed to solve the MDP. To expedite the training process, an optimization subroutine is embedded in the proposed algorithm to directly obtain the optimal edge computing resource allocation. Extensive simulation results are provided to demonstrate that the proposed RL-based algorithm can significantly reduce the average service delay while preserving long-term inference accuracy with a high probability.
- 10 March 2022
- Published by Springer Nature
- p. 263-290
Collaborative Deep Neural Network Inference via Mobile Edge Computing
- 10 March 2022
- Published by Springer Nature
- p. 263-290
Collaborative Deep Neural Network Inference via Mobile Edge Computing
- 7 December 2021
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Internet of Things Journal
- Vol. 9 (14) , 12056-12070
Learning-Based Computation Offloading for IoRT Through Ka/Q-Band Satellite–Terrestrial Integrated Networks
In this article, we propose a multilayer Ka/Q-band satellite–terrestrial integrated network for the Internet of Remote Things (IoRT) to achieve a high transmission rate with communication robustness in dynamic network environments. Under this architecture, we investigate how to jointly manage the offloading path selection and resource allocation to offload computation-intensive and delay-sensitive tasks in the IoRT. Considering continuous low earth orbit (LEO) satellite movements and Markovian rainfall changes, the computation offloading problem is described as a Markov decision process (MDP) formulation with the objective of maximizing the number of offloaded tasks with satisfied delay requirements and minimizing the power consumption of the LEO satellites. A deep reinforcement learning (DRL) approach is leveraged to make optimal decisions by taking account of dynamic queues of IoRT devices, channel conditions that vary with rainfall intensities and satellite positions, and computing capabilities of ground stations. Extensive simulations are conducted to validate the effectiveness and superiority of our proposed scheme.
- 7 December 2021
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Internet of Things Journal
- Vol. 9 (14) , 12056-12070
Learning-Based Computation Offloading for IoRT Through Ka/Q-Band Satellite–Terrestrial Integrated Networks
In this article, we propose a multilayer Ka/Q-band satellite–terrestrial integrated network for the Internet of Remote Things (IoRT) to achieve a high transmission rate with communication robustness in dynamic network environments. Under this architecture, we investigate how to jointly manage the offloading path selection and resource allocation to offload computation-intensive and delay-sensitive tasks in the IoRT. Considering continuous low earth orbit (LEO) satellite movements and Markovian rainfall changes, the computation offloading problem is described as a Markov decision process (MDP) formulation with the objective of maximizing the number of offloaded tasks with satisfied delay requirements and minimizing the power consumption of the LEO satellites. A deep reinforcement learning (DRL) approach is leveraged to make optimal decisions by taking account of dynamic queues of IoRT devices, channel conditions that vary with rainfall intensities and satellite positions, and computing capabilities of ground stations. Extensive simulations are conducted to validate the effectiveness and superiority of our proposed scheme.
- Top Cited Papers
- 7 October 2021
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Journal on Selected Areas in Communications
- Vol. 39 (12) , 3688-3703
Optimizing Federated Learning in Distributed Industrial IoT: A Multi-Agent Approach
In this paper, we aim to make the best joint decision of device selection and computing and spectrum resource allocation for optimizing federated learning (FL) performance in distributed industrial Internet of Things (IIoT) networks. To implement efficient FL over geographically dispersed data, we introduce a three-layer collaborative FL architecture to support deep neural network (DNN) training. Specifically, using the data dispersed in IIoT devices, the industrial gateways locally train the DNN model and the local models can be aggregated by their associated edge servers every FL epoch or by a cloud server every a few FL epochs for obtaining the global model. To optimally select participating devices and allocate computing and spectrum resources for training and transmitting the model parameters, we formulate a stochastic optimization problem with the objective of minimizing FL evaluating loss while satisfying delay and long-term energy consumption requirements. Since the objective function of the FL evaluating loss is implicit and the energy consumption is temporally correlated, it is difficult to solve the problem via traditional optimization methods. Thus, we propose a “ Reinforcement on Federated ” (RoF) scheme, based on deep multi-agent reinforcement learning, to solve the problem. Specifically, the RoF scheme is executed decentralizedly at edge servers, which can cooperatively make the optimal device selection and resource allocation decisions. Moreover, a device refinement subroutine is embedded into the RoF scheme to accelerate convergence while effectively saving the on-device energy. Simulation results demonstrate that the RoF scheme can facilitate efficient FL and achieve better performance compared with state-of-the-art benchmarks.
- 7 October 2021
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Journal on Selected Areas in Communications
- Vol. 39 (12) , 3688-3703
Top Cited PapersOptimizing Federated Learning in Distributed Industrial IoT: A Multi-Agent Approach
In this paper, we aim to make the best joint decision of device selection and computing and spectrum resource allocation for optimizing federated learning (FL) performance in distributed industrial Internet of Things (IIoT) networks. To implement efficient FL over geographically dispersed data, we introduce a three-layer collaborative FL architecture to support deep neural network (DNN) training. Specifically, using the data dispersed in IIoT devices, the industrial gateways locally train the DNN model and the local models can be aggregated by their associated edge servers every FL epoch or by a cloud server every a few FL epochs for obtaining the global model. To optimally select participating devices and allocate computing and spectrum resources for training and transmitting the model parameters, we formulate a stochastic optimization problem with the objective of minimizing FL evaluating loss while satisfying delay and long-term energy consumption requirements. Since the objective function of the FL evaluating loss is implicit and the energy consumption is temporally correlated, it is difficult to solve the problem via traditional optimization methods. Thus, we propose a “ Reinforcement on Federated ” (RoF) scheme, based on deep multi-agent reinforcement learning, to solve the problem. Specifically, the RoF scheme is executed decentralizedly at edge servers, which can cooperatively make the optimal device selection and resource allocation decisions. Moreover, a device refinement subroutine is embedded into the RoF scheme to accelerate convergence while effectively saving the on-device energy. Simulation results demonstrate that the RoF scheme can facilitate efficient FL and achieve better performance compared with state-of-the-art benchmarks.
- 10 May 2021
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Internet of Things Journal
- Vol. 9 (2) , 1300-1312
Deep-Reinforcement-Learning-Based Latency Minimization in Edge Intelligence Over Vehicular Networks
A novel paradigm that combines federated learning with blockchain to empower edge intelligence over vehicular networks (FBVN) can enable latency-sensitive deep neural network-based applications to be executed in a distributed pattern. However, the complex environments in FBVN make the system latency much harder to minimize by traditional methods. In this article, we model the training and transmission latency of each autonomous vehicle (AV) and consensus latency of the blockchain in-edge side in FBVN. Considering the dynamic and time-varying wireless channel conditions, unpredictable packet error rate, and unstable data sets quality, we adopt duel deep $Q$ -learning (DDQL) as the solving approach. We propose a federated DDQL algorithm, in which the learning agent is deployed on each AV side, and the sensing states on each AV do not need to be shared so that it increases scalability and flexibility for practical implementation. Simulation results show that the proposed algorithm has better performance in reducing system latency compared with the other schemes.
- 10 May 2021
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Internet of Things Journal
- Vol. 9 (2) , 1300-1312
Deep-Reinforcement-Learning-Based Latency Minimization in Edge Intelligence Over Vehicular Networks
A novel paradigm that combines federated learning with blockchain to empower edge intelligence over vehicular networks (FBVN) can enable latency-sensitive deep neural network-based applications to be executed in a distributed pattern. However, the complex environments in FBVN make the system latency much harder to minimize by traditional methods. In this article, we model the training and transmission latency of each autonomous vehicle (AV) and consensus latency of the blockchain in-edge side in FBVN. Considering the dynamic and time-varying wireless channel conditions, unpredictable packet error rate, and unstable data sets quality, we adopt duel deep $Q$ -learning (DDQL) as the solving approach. We propose a federated DDQL algorithm, in which the learning agent is deployed on each AV side, and the sensing states on each AV do not need to be shared so that it increases scalability and flexibility for practical implementation. Simulation results show that the proposed algorithm has better performance in reducing system latency compared with the other schemes.
- 23 March 2021
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Transactions on Vehicular Technology
- Vol. 70 (8) , 7605-7618
Deep Reinforcement Learning Based Resource Management for DNN Inference in Industrial IoT
Performing deep neural network (DNN) inference in real time requires excessive network resources, which poses a big challenge to the resource-limited industrial Internet of things (IIoT) networks. To address the challenge, in this paper, we introduce an end-edge-cloud orchestration architecture, in which the inference task assignment and DNN model placement are flexibly coordinated. Specifically, the DNN models, trained and pre-stored in the cloud, are properly placed at the end and edge to perform DNN inference. To achieve efficient DNN inference, a multi-dimensional resource management problem is formulated to maximize the average inference accuracy while satisfying the strict delay requirements of inference tasks. Due to the mix-integer decision variables, it is difficult to solve the formulated problem directly. Thus, we transform the formulated problem into a Markov decision process which can be solved efficiently. Furthermore, a deep reinforcement learning based resource management scheme is proposed to make real-time optimal resource allocation decisions. Simulation results are provided to demonstrate that the proposed scheme can efficiently allocate the available spectrum, caching, and computing resources, and improve average inference accuracy by 31.4$\%$ compared with the deep deterministic policy gradient benchmark.
- 23 March 2021
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Transactions on Vehicular Technology
- Vol. 70 (8) , 7605-7618
Deep Reinforcement Learning Based Resource Management for DNN Inference in Industrial IoT
Performing deep neural network (DNN) inference in real time requires excessive network resources, which poses a big challenge to the resource-limited industrial Internet of things (IIoT) networks. To address the challenge, in this paper, we introduce an end-edge-cloud orchestration architecture, in which the inference task assignment and DNN model placement are flexibly coordinated. Specifically, the DNN models, trained and pre-stored in the cloud, are properly placed at the end and edge to perform DNN inference. To achieve efficient DNN inference, a multi-dimensional resource management problem is formulated to maximize the average inference accuracy while satisfying the strict delay requirements of inference tasks. Due to the mix-integer decision variables, it is difficult to solve the formulated problem directly. Thus, we transform the formulated problem into a Markov decision process which can be solved efficiently. Furthermore, a deep reinforcement learning based resource management scheme is proposed to make real-time optimal resource allocation decisions. Simulation results are provided to demonstrate that the proposed scheme can efficiently allocate the available spectrum, caching, and computing resources, and improve average inference accuracy by 31.4$\%$ compared with the deep deterministic policy gradient benchmark.
- 7 October 2020
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Transactions on Industrial Informatics
- Vol. 17 (8) , 5461-5473
DeepHealth: A Self-Attention Based Method for Instant Intelligent Predictive Maintenance in Industrial Internet of Things
With the rapid development of artificial intelligence and industrial Internet of things (IIoT) technologies, intelligent predictive maintenance (IPdM) has received considerable attention. To efficiently predict impending failures and mitigate unexpected downtime, while satisfying the instant maintenance demands of industrial facilities is very important for improving the production efficiency. In this paper, a self-attention based framework, called DeepHealth, is proposed for the instant IPdM. Specifically, the framework is composed of two submodels (i.e., DH-1 and DH-2), which are respectively utilized to perform the health perception and sequence prediction. Considering the potential temporal correlation in time series, we deploy an enhanced attention mechanism to capture global dependencies from the vibration signals, and leverage the long- and short-term sequence prediction of sensor signals to carry out instant decision-making. On this basis, we conduct destructive experiments based on the IIoT-enabled rotating machinery and construct a balanced industrial dataset for model evaluations.
- 7 October 2020
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Transactions on Industrial Informatics
- Vol. 17 (8) , 5461-5473
DeepHealth: A Self-Attention Based Method for Instant Intelligent Predictive Maintenance in Industrial Internet of Things
With the rapid development of artificial intelligence and industrial Internet of things (IIoT) technologies, intelligent predictive maintenance (IPdM) has received considerable attention. To efficiently predict impending failures and mitigate unexpected downtime, while satisfying the instant maintenance demands of industrial facilities is very important for improving the production efficiency. In this paper, a self-attention based framework, called DeepHealth, is proposed for the instant IPdM. Specifically, the framework is composed of two submodels (i.e., DH-1 and DH-2), which are respectively utilized to perform the health perception and sequence prediction. Considering the potential temporal correlation in time series, we deploy an enhanced attention mechanism to capture global dependencies from the vibration signals, and leverage the long- and short-term sequence prediction of sensor signals to carry out instant decision-making. On this basis, we conduct destructive experiments based on the IIoT-enabled rotating machinery and construct a balanced industrial dataset for model evaluations.
- 18 August 2020
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Transactions on Industrial Informatics
- Vol. 17 (7) , 4988-4998
Accuracy-Guaranteed Collaborative DNN Inference in Industrial IoT via Deep Reinforcement Learning
Collaboration among industrial Internet of Things (IoT) devices and edge networks is critical to support computation-intensive deep neural network (DNN) inference services which require low delay and high accuracy. Sampling rate adaption which dynamically configures the sampling rates of industrial IoT devices according to network conditions, is the key in minimizing the service delay. In this paper, we investigate the collaborative DNN inference problem in industrial IoT networks. To capture the channel variation and task arrival randomness, we formulate the problem as a constrained Markov decision process (CMDP). Specifically, sampling rate adaption, inference task offloading and edge computing resource allocation are jointly considered to minimize the average service delay while guaranteeing the long-term accuracy requirements of different inference services. Since CMDP cannot be directly solved by traditional reinforcement learning (RL) algorithms due to the intractable long-term constraints, we first transform the CMDP into an MDP by leveraging the Lyapunov optimization technique. Then, a deep RL-based algorithm is proposed to solve the MDP. To expedite the training process, an optimization subroutine is embedded in the proposed algorithm to directly obtain the optimal edge computing resource allocation. Extensive simulation results are provided to demonstrate that the proposed algorithm can significantly reduce the average service delay while preserving long-term inference accuracy with a high probability, as compared to benchmark schemes.
- 18 August 2020
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Transactions on Industrial Informatics
- Vol. 17 (7) , 4988-4998
Accuracy-Guaranteed Collaborative DNN Inference in Industrial IoT via Deep Reinforcement Learning
Collaboration among industrial Internet of Things (IoT) devices and edge networks is critical to support computation-intensive deep neural network (DNN) inference services which require low delay and high accuracy. Sampling rate adaption which dynamically configures the sampling rates of industrial IoT devices according to network conditions, is the key in minimizing the service delay. In this paper, we investigate the collaborative DNN inference problem in industrial IoT networks. To capture the channel variation and task arrival randomness, we formulate the problem as a constrained Markov decision process (CMDP). Specifically, sampling rate adaption, inference task offloading and edge computing resource allocation are jointly considered to minimize the average service delay while guaranteeing the long-term accuracy requirements of different inference services. Since CMDP cannot be directly solved by traditional reinforcement learning (RL) algorithms due to the intractable long-term constraints, we first transform the CMDP into an MDP by leveraging the Lyapunov optimization technique. Then, a deep RL-based algorithm is proposed to solve the MDP. To expedite the training process, an optimization subroutine is embedded in the proposed algorithm to directly obtain the optimal edge computing resource allocation. Extensive simulation results are provided to demonstrate that the proposed algorithm can significantly reduce the average service delay while preserving long-term inference accuracy with a high probability, as compared to benchmark schemes.
- Open Access
RMLNet—A Reliable Wireless Network for a Multiarea TDOA-Based Localization System
Ultrawideband (UWB) wireless communication is a promising spread-spectrum technology for accurate localization among devices characterized by a low transmission power, a high rate and immunity to multipath propagation. The accurately of the clock synchronization algorithm and the time-difference-of-arrival (TDOA) localization algorithm provide precise position information of mobile nodes with centimeter-level accuracy for the UWB localization system. However, the reliability of target node localization for multi-area localization remains a subject of research. Especially for dynamic and harsh indoor environments, an effective scheme among competing target nodes for localization due to the scarcity of radio resources remains a challenge. In this paper, we present RMLNet, an approach focus on the medium access control (MAC) layer, which guarantees general localization application reliability on multi-area localization. Specifically, the design requires specific and optimized solutions for managing and coordinating multiple anchor nodes. In addition, an approach for target area determination is proposed, which can approximately determine the region of the target node by the received signal strength indication (RSSI), to support RMLNet. Furthermore, we implement the system to estimate the localization of the target node and evaluate its performance in practice. Experiments and simulations show that RMLNet can achieve localization application reliability multi-area localization with a better localization performance of competing target nodes.
Open AccessRMLNet—A Reliable Wireless Network for a Multiarea TDOA-Based Localization System
Ultrawideband (UWB) wireless communication is a promising spread-spectrum technology for accurate localization among devices characterized by a low transmission power, a high rate and immunity to multipath propagation. The accurately of the clock synchronization algorithm and the time-difference-of-arrival (TDOA) localization algorithm provide precise position information of mobile nodes with centimeter-level accuracy for the UWB localization system. However, the reliability of target node localization for multi-area localization remains a subject of research. Especially for dynamic and harsh indoor environments, an effective scheme among competing target nodes for localization due to the scarcity of radio resources remains a challenge. In this paper, we present RMLNet, an approach focus on the medium access control (MAC) layer, which guarantees general localization application reliability on multi-area localization. Specifically, the design requires specific and optimized solutions for managing and coordinating multiple anchor nodes. In addition, an approach for target area determination is proposed, which can approximately determine the region of the target node by the received signal strength indication (RSSI), to support RMLNet. Furthermore, we implement the system to estimate the localization of the target node and evaluate its performance in practice. Experiments and simulations show that RMLNet can achieve localization application reliability multi-area localization with a better localization performance of competing target nodes.
- Open Access
- 24 September 2019
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Access
- Vol. 7, 141487-141497
AESGRU: An Attention-Based Temporal Correlation Approach for End-to-End Machine Health Perception
Accurate and real-time perception of the operating status of rolling bearings, which constitute a key component of rotating machinery, is of vital significance. However, most existing solutions not only require substantial expertise to conduct feature engineering, but also seldom consider the temporal correlation of sensor sequences, ultimately leading to complex modeling processes. Therefore, we present a novel model, named Attention-based Equitable Segmentation Gated Recurrent Unit Networks (AESGRU), to improve diagnostic accuracy and model-building efficiency. Specifically, our proposed AESGRU consists of two modules, an equitable segmentation approach and an improved deep model. We first transform the original dataset into time-series segments with temporal correlation, so that the model enables end-to-end learning from the strongly correlated data. Then, we deploy a single-layer bidirectional GRU network, which is enhanced by attention mechanism, to capture the long-term dependency of sensor segments and focus limited attention resources on those informative sampling points. Finally, our experimental results show that the proposed approach outperforms previous approaches in terms of the accuracy.
- 24 September 2019
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Access
- Vol. 7, 141487-141497
Open AccessAESGRU: An Attention-Based Temporal Correlation Approach for End-to-End Machine Health Perception
Accurate and real-time perception of the operating status of rolling bearings, which constitute a key component of rotating machinery, is of vital significance. However, most existing solutions not only require substantial expertise to conduct feature engineering, but also seldom consider the temporal correlation of sensor sequences, ultimately leading to complex modeling processes. Therefore, we present a novel model, named Attention-based Equitable Segmentation Gated Recurrent Unit Networks (AESGRU), to improve diagnostic accuracy and model-building efficiency. Specifically, our proposed AESGRU consists of two modules, an equitable segmentation approach and an improved deep model. We first transform the original dataset into time-series segments with temporal correlation, so that the model enables end-to-end learning from the strongly correlated data. Then, we deploy a single-layer bidirectional GRU network, which is enhanced by attention mechanism, to capture the long-term dependency of sensor segments and focus limited attention resources on those informative sampling points. Finally, our experimental results show that the proposed approach outperforms previous approaches in terms of the accuracy.