Finding the Missing Data: A BERT-inspired Approach Against Package Loss
in Wireless Sensing
Finding the Missing Data: A BERT-inspired Approach Against Package Loss in Wireless Sensing
Zijian Zhao Tingwei Chen Fanyi Meng Hang Li Xiaoyang Li Guangxu Zhu

Abstract
Despite the development of various deep learning methods for Wi-Fi sensing,package loss often results in noncontinuous estimation of the Channel StateInformation (CSI), which negatively impacts the performance of the learningmodels. To overcome this challenge, we propose a deep learning model based onBidirectional Encoder Representations from Transformers (BERT) for CSIrecovery, named CSI-BERT. CSI-BERT can be trained in an self-supervised manneron the target dataset without the need for additional data. Furthermore, unliketraditional interpolation methods that focus on one subcarrier at a time,CSI-BERT captures the sequential relationships across different subcarriers.Experimental results demonstrate that CSI-BERT achieves lower error rates andfaster speed compared to traditional interpolation methods, even when facingwith high loss rates. Moreover, by harnessing the recovered CSI obtained fromCSI-BERT, other deep learning models like Residual Network and Recurrent NeuralNetwork can achieve an average increase in accuracy of approximately 15% inWi-Fi sensing tasks. The collected dataset WiGesture and code for our model arepublicly available at https://github.com/RS2002/CSI-BERT.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| action-classification-on-wigesture | CSI-BERT | Accuracy (% ): 76.91 |
| person-identification-on-wigesture | CSI-BERT | Accuracy (% ): 93.94 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.