LiteTransNet: An interpretable approach for landslide displacement prediction using transformer model with attention mechanism

• Published in: Engineering geology 2024-03, Vol.331, p.107446, Article 107446
• Main Authors: Ge, Qi, Li, Jin, Wang, Xiaohong, Deng, Yiyan, Zhang, Keying, Sun, Hongyue
• Format: Article
• Language: English
• Subjects: Attention mechanism; Interpretable machine learning; Landslide displacement prediction; Three gorges reservoir area; Transformer
• ISSN: 0013-7952; 1872-6917
• DOI: 10.1016/j.enggeo.2024.107446

Accurate landslide displacement prediction is crucial for effective early warning systems to mitigate hazards. The importance of historical information varies with time during prediction due to the underlying landslide deformation mechanism. Despite advances in dynamic machine learning models like Long Short-Term Memory (LSTM) and Gated Recurrent Unit (GRU), they struggle to fully capture and interpret the varying importance of historical information during prediction, resulting in decreased accuracy and limited understanding of landslide physical mechanisms. Additionally, these approaches rely on manual feature selection, which is independent of the learning process and therefore is prone to estimation errors. To address these challenges, we propose LiteTransNet, a Lightweighted Transformer Network tailored for landslide displacement prediction. Built on the revolutionary Transformer model for sequential data, LiteTransNet leverages localized self-attention to selectively focus on relevant timestamps and provides interpretable results through its attention heatmap. Furthermore, LiteTransNet performs end-to-end time series modeling without extensive feature engineering. We validate LiteTransNet on two landslides in China's Three Gorges Reservoir Area and demonstrate its improved accuracy over recurrent baselines utilizing feature selection methods. Notably, the attention heatmap generated by LiteTransNet provides interpretable insights into the model's temporal dependency learning. It uncovers the varying importance of historical information at different timestamps, showcasing how LiteTransNet's attention aligns with specific periods of the external environment that cause significant disruptions in the landslide system. Our findings reveal that these disruptions continuously impact displacement prediction until their effects fade or are replaced by subsequent intense disturbances, which traditional recurrent methods are unable to capture. Further experiment shows that LiteTransNet enhances efficiency through its streamlined architecture and its parallelization capability. Overall, LiteTransNet innovates landslide prediction by providing accuracy, interpretability, and efficiency, enhancing understanding of deformation mechanisms to facilitate effective early warning systems. [Display omitted] •A bespoke Transformer model is applied to landslide displacement prediction.•Prediction reliability is enhanced via end-to-end time series modeling.•Prediction inter