Defect Screening on Nuclear Power Plant Concrete Structures: A Two-staged Method Based on Contrastive Representation Learning

Abstract

Intelligent defect detection methods are important for the surface of the containment of nuclear power plants and face many challenges in the field of computer vision. Due to the irregular shapes and large variation of defects, as well as the similarity of the features between some defects and the background. Most existing deep learning-based defect detection networks suffer from insufficient feature extraction capabilities, making it difficult to detect defects from the background. Inspired by discrete representation learning methods, we propose a two-stage defect detection model called ConVQVAE, which uses a semi-supervised method for representation learning to achieve binary classification for defect detection. In addition, contrastive learning is performed in two stages to enhance the model's diversity representation ability and inter-class disentanglement representation ability. Finally, we experimentally verify that the proposed method can obtain good classification results.