TASK-SIGNIFICANCE-AWARE META LEARNING FOR FEW-IMAGE-BASED STRUCTURAL DAMAGE RECOGNITION

YANG XU,YI LI,YUNFEI FAN,XIAODONG ZHENG,YUEQUAN BAO
DOI: https://doi.org/10.12783/shm2023/36869
2023-01-01
Abstract:Recently, structural damage recognition has gained significant progress using deep learning and computer vision techniques. However, massive training images, the interclass balance and completeness of damage categories are essential to ensure recognition accuracy. In addition, the generalization ability for new damage categories and robustness under real-world scenarios are limited. This study proposes a task-aware meta-learning paradigm using limited images for universal structural damage segmentation. First, a novel task generation strategy instead of random sampling is designed based on feature density clustering. A synthetical metric of Jaccard distance and Euclidean distance is established to measure the feature similarity among multitype damage images. The class separability discovered in the high-level feature space of multi-type structural damage enhances the interpretability of randomly-generated tasks for conventional meta-learning. Second, a dual-stage optimization framework is built based on Model-Agnostic Meta-Learning (MAML), comprising an internal optimization stage of the semantic segmentation model (U-Net) and an external optimization stage of the meta-learning machine. Third, a set of core samples around the cluster center is selected to form an additional query pool and evaluate the tasksignificance scores of different tasks within a meta-batch by the same criteria. The task-significance scores are utilized in the external optimization to control the orientation of gradient updates towards more significant tasks. To verify the effectiveness and necessity of the proposed method, ablation experiments are performed using a multi-type structural damage dataset, including concrete crack, steel fatigue crack, and concrete spalling. The proposed method outperforms directly training the original U-Net and the conventional MAML algorithm using only a handful of training samples with improvements in segmentation accuracy. In addition, the improvement in recognition accuracy increases when using fewer training images, further indicating the efficacy of the proposed method. The generalization ability for new structural damage of steel corrosion is also demonstrated.
What problem does this paper attempt to address?