Federated learning-based Lightweight Selective Feature Fusion and Irregular-Aware Network for Crack Detection
Article
Figures
Metrics
Preview PDF
Reference
Related
Cited by
Materials
Abstract:
Employing deep learning for pavement crack detection can significantly enhance accuracy, and federated learning can help to overcome the challenges of data silos and data security. This paper proposes a lightweight method to detect cracks. We use lightweight encoder modules to extract multi-scale features, and further feature fusion and modelling by Selective Fusion Blocks and Irregular-aware Blocks. Moreover, this method is the first to combine federated learning with crack detection, effectively resolving the tension between data privacy and data sharing in decentralised devices. Experiments were conducted to compare our proposed method with other crack detection methods on two publicly available datasets, including the original method without federated learning and five state-of-the-art methods combined with the same feder-ated learning framework. The experiment proves that our model obtains comparable F1 score with minimal parameters and computational effort. For example, compared to the original method, the number of parameters and computation are reduced by 67.1% and 73.6%, respectively.
Keywords:
Project Supported:
The National Natural Science Foundation of China (General Program, Key Program, Major Research Plan)