FLPK-BiSeNet: Federated Learning Based on Priori Knowledge and Bilateral Segmentation Network for Image Edge Extraction

Lin Teng,Yulong Qiao,Muhammad Shafiq,Gautam Srivastava,Abdul Rehman Javed,Thippa Reddy Gadekallu,Shoulin Yin
DOI: https://doi.org/10.1109/tnsm.2023.3273991
2023-01-01
IEEE Transactions on Network and Service Management
Abstract:Federated learning can effectively ensure data security and improve the problem of data islanding. However, the performance of federated learning-based schemes could be better due to the imbalance of image data. Therefore, this paper proposes a federated learning approach based on priori knowledge and a bilateral segmentation network for image edge extraction. First, federated learning can distribute training images for some special complex images due to the small sample and unshared data. Then, the image with similar edge information to the original image is learned to obtain prior knowledge, and the local uniform sparsity method is used to strengthen the detail features and weaken the background features. Based on the bilateral segmentation network, we introduce a dilated pyramid pooling layer and multi-scale feature fusion module to fuse the shallow detailed features in the context path with the deep abstract features obtained through the dilated pyramid pooling. The final result is obtained by fusing the result with prior knowledge and the result with the context path. Finally, we conduct experiments on some public datasets, and the results show that the proposed method greatly improves extraction accuracy compared with the traditional and the most advanced methods.
computer science, information systems
What problem does this paper attempt to address?