Multi-task Learning-based Joint CSI Prediction and Predictive Transmitter Selection for Security

Shashi Bhushan Kotwal,Chinmoy Kundu,Sudhakar Modem,Holger Claussen,Lester Ho
2024-05-01
Abstract:In mobile communication scenarios, the acquired channel state information (CSI) rapidly becomes outdated due to fast-changing channels. Opportunistic transmitter selection based on current CSI for secrecy improvement may be outdated during actual transmission, negating the diversity benefit of transmitter selection. Motivated by this problem, we propose a joint CSI prediction and predictive selection of the optimal transmitter strategy based on historical CSI by exploiting the temporal correlation among CSIs. The proposed solution utilizes the multi-task learning (MTL) framework by employing a single Long Short-Term Memory (LSTM) network architecture that simultaneously learns two tasks of predicting the CSI and selecting the optimal transmitter in parallel instead of learning these tasks sequentially. The proposed LSTM architecture outperforms convolutional neural network (CNN) based architecture due to its superior ability to capture temporal features in the data. Compared to the sequential task learning models, the MTL architecture provides superior predicted secrecy performance for a large variation in the number of transmitters and the speed of mobile nodes. It also offers significant computational and memory efficiency, leading to a substantial saving in computational time by around 40 percent.
Signal Processing
What problem does this paper attempt to address?
The problem that this paper attempts to solve is: in mobile communication scenarios, due to the rapid obsolescence of channel state information (CSI), the transmitter selection strategy based on the current CSI may become ineffective during the actual transmission process, and thus the diversity gain brought by transmitter selection cannot be utilized. In addition, obsolete CSI may lead to incorrect transmitter selection, thereby reducing the secrecy performance of the system. To solve this problem, the author proposes a joint CSI prediction and predictive transmitter selection method based on the multi - task learning (MTL) framework. This method simultaneously learns two tasks, namely predicting CSI and selecting the optimal transmitter, by using a single long - short - term memory network (LSTM) architecture through exploiting the temporal correlation in historical CSI. Compared with the traditional sequential learning model, this joint learning method not only improves the predicted secrecy performance but also has significant advantages in terms of computational and memory efficiency. Specifically, the main contributions of the paper include: 1. Proposing an MTL - based ML architecture that uses the LSTM model to capture the temporal relationships in CSI while performing CSI prediction and predictive transmitter selection simultaneously. 2. Demonstrating that the LSTM architecture outperforms the convolutional neural network (CNN) architecture in predicting CSI and selecting the optimal transmitter because the LSTM is better at capturing the temporal features of data. 3. Showing that, compared with the sequential task - learning model, the MTL model has better predicted secrecy performance when the node speed and the number of transmitters change significantly. 4. Compared with the sequential learning model, the MTL architecture has more advantages in computational and memory efficiency and can save approximately 40% of the computational time. Through these improvements, the paper provides an effective solution to address the challenges brought by the rapid obsolescence of CSI in the mobile communication environment and improves the secrecy and resource utilization efficiency of the system.