Application of Transformer Models to Landslide Susceptibility Mapping

Sensors (Basel). 2022 Nov 23;22(23):9104. doi: 10.3390/s22239104.

Abstract

Landslide susceptibility mapping (LSM) is of great significance for the identification and prevention of geological hazards. LSM is based on convolutional neural networks (CNNs); CNNs use fixed convolutional kernels, focus more on local information and do not retain spatial information. This is a property of the CNN itself, resulting in low accuracy of LSM. Based on the above problems, we use Vision Transformer (ViT) and its derivative model Swin Transformer (Swin) to conduct LSM for the selected study area. Machine learning and a CNN model are used for comparison. Fourier transform amplitude, feature similarity and other indicators were used to compare and analyze the difference in the results. The results show that the Swin model has the best accuracy, F1-score and AUC. The results of LSM are combined with landslide points, faults and other data analysis; the ViT model results are the most consistent with the actual situation, showing the strongest generalization ability. In this paper, we believe that the advantages of ViT and its derived models in global feature extraction ensure that ViT is more accurate than CNN and machine learning in predicting landslide probability in the study area.

Keywords: Vision Transformer; data-driven; feature extraction; landslide susceptibility mapping.

MeSH terms

  • Algorithms
  • Data Analysis
  • Electric Power Supplies*
  • Machine Learning
  • Neural Networks, Computer*

Grants and funding