Vision transformer-based autonomous crack detection on asphalt and concrete surfaces

Previous research has shown the high accuracy of convolutional neural networks (CNNs) in asphalt and concrete crack detection in controlled conditions. Yet, human-like generalisation remains a significant challenge for industrial applications where the range of conditions varies significantly. Given the intrinsic biases of CNNs, this paper proposes a vision transformer (ViT)-based framework for crack detection on asphalt and concrete surfaces. With transfer learning and the differentiable intersection over union (IoU) loss function, the encoder-decoder network equipped with ViT could achieve an enhanced real-world crack segmentation performance. Compared to the CNN-based models (DeepLabv3+ and U-Net), TransUNet with a CNN-ViT backbone achieved up to -61% and -3.8% better mean IoU on the original images of the respective datasets with very small and multiscale crack semantics. Moreover, ViT assisted the encoder-decoder network to show a robust performance against various noisy signals where the mean Dice score attained by the CNN-based models significantly dropped (<10%).

Automated summary

This research proposes a ViT-based framework for crack detection on asphalt and concrete surfaces, achieving enhanced real-world crack segmentation performance through transfer learning and IoU loss function. Compared to CNN-based models, TransUNet with a CNN-ViT backbone shows better average IoU on small and multiscale crack semantics and ViT helps the encoder-decoder network exhibit robust performance against various noisy signals.