Transfer Learning and Double U-Net Empowered Wave Propagation Model in Complex Indoor Environment

Ziheng Fu,Swagato Mukherjee,Michael T. Lanagan,Prasenjit Mitra,Tarun Chawla,Ram M. Narayanan
2024-09-21
Abstract:A Machine Learning (ML) network based on transfer learning and transformer networks is applied to wave propagation models for complex indoor settings. This network is designed to predict signal propagation in environments with a variety of objects, effectively simulating the diverse range of furniture typically found in indoor spaces. We propose Attention U-Net with Efficient Networks as the backbone, to process images encoded with the essential information of the indoor environment. The indoor environment is defined by its fundamental structure, such as the arrangement of walls, windows, and doorways, alongside varying configurations of furniture placement. An innovative algorithm is introduced to generate a 3D environment from a 2D floorplan, which is crucial for efficient collection of data for training. The model is evaluated by comparing the predicted signal coverage map with ray tracing (RT) simulations. The prediction results show a root mean square error of less than 6 dB across all tested scenarios, with significant improvements observed when using a Double U-Net structure compared to a single U-Net model.
Signal Processing
What problem does this paper attempt to address?
This paper attempts to solve the problem of accurately predicting millimeter - wave signal propagation in complex indoor environments. Specifically, the paper focuses on how to effectively simulate and predict signal propagation patterns in indoor environments containing various furniture and other obstacles. Traditional methods such as Ray Tracing (RT) are accurate but computationally costly and time - consuming, while empirical models have difficulty dealing with complex and changeable indoor layouts. ### The core challenges of the problem include: 1. **Complex electromagnetic wave - obstacle interactions**: When electromagnetic waves encounter obstacles such as walls and furniture, reflection, refraction and diffraction occur, which make the signal propagation pattern complex. 2. **High computational cost**: Traditional ray - tracing methods require a large amount of computational resources and time, especially when considering diffraction and multiple reflections. 3. **The influence of furniture arrangement**: The presence of furniture has a significant impact on the signal coverage pattern, especially in the high - frequency band (such as 28 GHz). Ignoring the furniture arrangement will lead to large prediction errors. 4. **The need for dynamic environments**: Modern indoor environments (such as smart healthcare, smart homes, etc.) require real - time feedback and optimization, which places higher demands on computational efficiency. ### The method proposed in the paper: To solve the above problems, the paper proposes a machine - learning model based on transfer learning and Transformer networks, especially using the dual - U - Net structure and EfficientNet as the backbone network. This model aims to improve signal propagation prediction in the following ways: - **Data - driven ML model**: Encode the indoor floor plan with furniture arrangement into a matrix and use it as input to output the corresponding radio map. - **Transfer learning**: Use the pre - trained EfficientNet model for fine - tuning to reduce the amount of data and time required for training. - **Dual - U - Net architecture**: The first stage generates a preliminary signal coverage prediction, and the second stage performs refined processing on this basis to improve the prediction accuracy. - **Attention mechanism**: Introduce an Attention Gate to emphasize the areas in the environment that have an important impact on signal propagation. ### Main contributions: 1. Propose a signal propagation model suitable for indoor environments with furniture arrangements, and construct a database containing furniture arrangements and their corresponding RT - simulated coverage. 2. Develop a MATLAB tool for constructing indoor environments that can be used for RT simulation, and share the processed environment and simulation result database. 3. Propose a dual - U - Net architecture with EfficientNet as the backbone. This model performs well in complex indoor environments, especially in signal propagation modeling. Through this method, the paper not only improves the accuracy of signal propagation prediction, but also significantly reduces the computational time and resource requirements, thus providing a more feasible solution for practical applications.