Slope Collapse Detection Method Based on Deep Learning Technology

Xindai An,Di Wu,2,★,Xiangwen Xie and Kefeng Song

1Yellow River Engineering Consulting Co.,Ltd.,Zhengzhou,450003,China

2School of Artificial Intelligence and Automation,Huazhong University of Science and Technology,Wuhan,430074,China

ABSTRACT So far,slope collapse detection mainly depends on manpower,which has the following drawbacks:(1)low reliability,(2)high risk of human safe,(3)high labor cost.To improve the efficiency and reduce the human investment of slope collapse detection,this paper proposes an intelligent detection method based on deep learning technology for the task.In this method,we first use the deep learning-based image segmentation technology to find the slope area from the captured scene image.Then the foreground motion detection method is used for detecting the motion of the slope area.Finally,we design a lightweight convolutional neural network with an attention mechanism to recognize the detected motion object,thus eliminating the interference motion and increasing the detection accuracy rate.Experimental results on the artificial data and relevant scene data show that the proposed detection method can effectively identify the slope collapse,which has its applicative value and brilliant prospect.

KEYWORDS Deep learning;slope collapse;image segmentation;image recognition

1 Introduction

The Yellow River is the Mother River of the Chinese nation,which gave birth to the ancient and great Chinese civilization.However,the Yellow River is easy to silt,burst,and move.It burst twice in three years and changed its diversion once within one hundred years in history,which has brought severe disasters to the people along the coast[1].Since the founding of the new China,the party and the state have attached great importance to the governance of the Yellow River.The construction of large-scale embankment engineering and river regulation engineering have provided an important guarantee for the flood control safety of the Yellow River.

It is the important work of the water administration to manage and maintain the river defense engineering and to ensure that the engineering can play the role of flood control and security for a long time[2,3].Piping leakage[4]and foundation stone collapse[5]are the main forms of deformation and failure of river defense engineering.One ant hole may cause the collapse of a thousand-li dyke.Early detection and prevention is not only the practical needs of safety management of the river defense engineering but also the most important task of the water administration department.At present,this task mainly relies on manned patrol[6].However,the scale of the Yellow River defense engineering is huge[7].Relying on manpower for patrol inspection not only requires a lot of material resources,but also it is difficult to avoid problems such as missed inspection and late inspection.

Since entering the new century,with the development of information technology,water conservancy big data has been more and more applied in the governance of the Yellow River and other drainage basins[8].In theNational Informatization Development Strategy Outlineissued by the CPC Central Committee and the State Council,it is proposed to improve the application level of informatization and guide the deep integration of the new generation of information technology and the economic and social fields.The Guidelines on Promoting the Development of Water Conservancy Big Data,the Top-level Design for the Integration and Sharing of Water Conservancy Informatization Resources,the Overall Plan for Smart Water Conservancy,and other documents issued by the Ministry of Water Resources have repeatedly proposed major measures such as using the dynamic monitoring data of the space-earth integration,focusing on the new service of flood and drought disaster prediction and early warning.The research on early warning of dynamic monitoring and safety risk of river defense engineering can effectively make up for the deficiencies in the engineering supervision information collection and perception ability of the current digital Yellow River.Aiming at the problems existing in the inspection task of river defense engineering,this paper plans to use deep learning and image processing technology to design the corresponding detection and recognition model to assist or replace the manual inspection of slope stability of river defense engineering.

2 Current Situation and Problems

So far,the stability monitoring of the slope of river defense engineering mainly depends on human resources,which has the following major problems:

The cost of manual inspection is high.First,the characteristic of Yellow River defense engineering is multiple types and long tunnels,and it is requiring comprehensive and regular manual inspection,which needs a large number of personnel investments.Second,manual-style inspection requires equipping with professional and technical personnel and establishing relevant management institutions,thus resulting in large expenditure on management and maintenance.Only taking the management and protection of the river defense engineering in the lower reaches of the Yellow River as an example,the annual management and protection cost of embankment per kilometer is about 170,000 yuan and that of channel dangerous project is about 200,000 yuan.The total annual management and protection cost of the embankment and dangerous works has reached 200 to 300 million yuan a year.

The reliability of manual inspection is low.First,manual inspection is limited by natural and traffic conditions.During the flood,it is often accompanied by rainfall and strong convective weather,which is difficult to achieve full coverage of river defense engineering and timely capture of dangerous situations only with the help of simple equipment and limited manual inspection range.Second,the quality of the early river defense engineering of the Yellow River is uneven,the new projects lack large flood inspection,and there are many hidden danger points in the embankment body and dam foundation,which leads to dangerous situations occurring frequently and gustily.

The personal safety risk of manual inspection is high.First,during the flood period,the river level rises and the flow velocity is large,which may lead to dangerous situations such as river regime change and bank collapse.These dangerous situations are often sudden,and result in the personal safety of patrol inspectors is often at great risk.Second,the flood of the Yellow River,especially the flood in the upper reaches,usually lasts a long time.Continuous work makes the inspection team face fatigue,and the fatigue state further increases the personal safety risk.

In addition,some researchers try to use computer vision algorithms to automatically monitor debris flow and landslides.Patent [9] proposed a landslide monitoring device based on computer vision,which includes monitoring equipment,control equipment,electric control system,and characteristic targets.The landslide can be monitored by solving and analyzing the target position information through the monitoring equipment.The device can monitor the mountain deformation to a certain extent.However,the picture flutter caused by external interference has a negative impact on the analysis process.In addition,the device is still based on the contact type,and the installation process is complex and dangerous.Literature[10–13]used object detection networks to learn the characteristics of earthquake landslides and mud-rock flow landslides.This kind of method can effectively identify the disasters with obvious landslide characteristics to a certain extent.However,when the landslide characteristics are not obvious,the recognition performance will be greatly degraded,which cannot meet the requirements of real-time landslide detection.

In addition,there is no work to study the stability monitoring of river defense engineering,which is still a major problem to be solved in the fields of scientific research and industry.

Aiming to the above current situation and existing problems,this paper proposes a slope collapse monitoring method of river defense engineering based on deep learning and image processing technology.The method first uses a designed attention U-net model to accurately segment the slope area.Then the background modeling algorithm is adopted to detect the moving targets in the slope area based on the segmentation results.Finally,a lightweight convolution neural network is proposed to identify the detected moving targets to eliminate the interference of incoherent moving targets.The experimental results on simulated and real scene data show that the proposed method can effectively detect the slope collapse and eliminate the interfering movement,which has a good practical application effect.

3 Proposed Method

The overall framework of the proposed model is shown in Fig.1,which is divided into four steps:

Figure 1:The overall framework of the proposed method

Firstly,the slope data set used for slope reg.Ion segmentation is collected and labeled.Secondly,a lightweight U-net network model based on an attention mechanism is designed and trained with the collected data set.Thirdly,the background modeling and moving object detection algorithm are used to detect the moving objects in the segmented slope area.Finally,we design a lightweight convolutional neural network to identify the detected moving target,to judge whether the movement is caused by the slope collapse.Each of the steps is described in detail below.

3.1 Slope Region Segmentation Based on Deep Learning

Accurate segmentation of the slope area of river defense engineering is the first step of the proposed method.The segmentation effect has a direct impact on the later movement detection results.Considering the accuracy and timeliness of the algorithm,a lightweight U-net segmentation network based on an attention mechanism is designed to segment the engineering slope region.

The U-net network was proposed by Ronneberger et al.[14].The model realizes the fusion between high-resolution information and low-level semantic information through a “U”network structure,and the skip connections mechanism between symmetrical layers is used to better restore the detail information of the original image,which can be well applied to image segmentation tasks.The network mainly includes the contraction path and expansion path.The contraction path is used to extract context information,which can reduce the dimension of feature space and increase the number of feature channels at the same time.The shrinkage path in the original network contains four stages,in which each stage contains two convolution operations with a 3 × 3 kernel size and a maximum pooling operation with a 2×2 kernel size.The number of channels in each shrinkage stage is doubled compared with the previous stage.Similar to the shrinkage path,the expansion path also includes four stages,which mainly restore the target details in the original image through the up-sampling operation.The dimension of the feature map in each stage is twice the size of the previous stage,and the number of channels is halved.The last layer of the network is a 1×1 convolution operation,which maps the number of channels to 2.Finally,the network outputs the prediction probability value through thesigmoidfunction.

In this paper,to improve the efficiency of the segmentation network,we design a lightweight U-net.Specifically,the four stages in the shrinkage path and expansion path in the original model are changed into three stages,in which the number of channels in each stage is reduced to one-third of the raw one,and the rest settings remain unchanged.In addition,to strengthen the attention of the U-net model on important feature information,we further introduce the attention mechanism module into the lightweight segmentation network.The common attention mechanism modules are SENet[15](Squeeze-and-Excitation Networks),CBAM[16](Convolutional Block Attention Module),BAM[17](Bottleneck Attention Module and SKNet) [18] (Selective Kernel Networks).We use BAM as the mechanism module,which includes a channel attention module and a spatial attention module.The channel attention module is composed of an average pool layer and two linear layer mapping layers,which outputs a set of corresponding weight coefficients for each feature graph.The spatial attention module includes two reduction layers with 1 × 1 kernel size of convolution operation and two convolution layers with 3×3 kernel size of the convolution operation.The attention mechanism module finally performs element-based summation on the output of channel attention and spatial attention and uses thesigmoidactivation function to calculate the final weight probability value.In this paper,each intermediate output feature of the U-net is input into the BAM attention module to re-strengthen or restrain the extracted feature.The lightweight U-net with the BAM attention module can focus the attention of the extracted features on the important spatial or channel features,increasing the semantic information of the extracted features and improving the segmentation performance.

3.2 Motion Detection Based on Background Modeling

After segmenting the slope region,we use the background modeling and moving object detection algorithm to detect the movement of the slope area.The basic principle of the background subtraction method is to approximate the simulated background with a mathematical background model,and then distinguish the current frame image from the background model to obtain the movement area.Common background models include mixed Gaussian model,K-nearest neighbor model,and codebook model.In this paper,a mixed Gaussian model(GMM)is used to detect the movement of the slope region.

GMM model[19]first defines K(K ≥2)Gaussian distributions.In the detection process,each pixel in the image is input into the K Gaussian distributions respectively.If one of the K Gaussian distributions meets condition(1),the pixel is classified as the background pixel part,otherwise,it is classified as the foreground.

wherePis the current pixel value,μandσrepresent the mean value and standard deviation of Gaussian distribution,respectively.

The process can be formulated as:

whereKis the number of Gaussian distributions,Xtis the pixel value at timet,ωk,tis the weight of K Gaussian models at timet,ηis the probability density function of the kth Gaussian model at timet,andμk,tandrepresents the mean value and variance of the corresponding probability density function,respectively.

To improve the robustness of the motion detection model,is used to represent the reliability parameter of the kth Gaussian distribution at timet,and arrange the K Gaussian distributions from large to small according to the size of the parameter,thus the distributions generated by interference factors are ranked behind,and the distributions that can effectively describe the background change are ranked first.We select the firstBdistributions through the threshold valueT,and the calculation process can be formulated as(3):

The Eq.(4)is used to update the weight of the GMM algorithm:

whereωk,t+1represents the weight of the kth Gaussian distribution at timet+1,andαis the learning rate.When the pixel value meets Eq.(1),Mis set to 1,otherwise set to 0.

For the Gaussian distribution matching the pixel value to be measured,the update rules of mean value and variance are shown in(5):

whereinρis the weight of the learning rateXt+1represents the pixel value at timet+1.After the update,the Gaussian distribution is arranged from large to small according to the reliability parameters.

3.3 Moving Target Recognition Based on Lightweight Convolutional Neural Network

In the actual engineering environment,except for the movement information formed by slope collapse,there are many interference movements,such as birds,pedestrians,leaves fluttering,etc.The movement information caused by these interference sources can also be detected by the background modeling algorithm,resulting in the false detection of slope collapse movement.In order to effectively identify the detected moving target,a lightweight convolutional neural network is proposed to identify the detected moving target to improve the detection accuracy of the proposed method.The architecture of the proposed network is shown in Fig.2.

Figure 2:The proposed lightweight network

The main framework of the proposed lightweight network is based on the squeezenet[20]module,including two convolution layers,one maximum pooling layer,a global average pooling layer,five fire layers,and a softmax probability output layer.The fire module includes two parts:squeeze and expand.Squeeze is a convolution operation with the kernel size of 1×1,and theexpansion is composed of 1×1 and 3×3 convolution operations.The structure of the fire layers is shown in Fig.3:

Figure 3:Fire module[20]

4 Experiments

4.1 Datasets

The experimental data in this paper mainly includes the following three kinds: slope region segmentation data set,motion detection data set and interference elimination data set.The detail of the three datasets is described below.

4.1.1 Slope Region Segmentation Dataset

To effectively segment the slope area,we collected nearly 1000 slope image data from Madu Channel Dangerous Project along the Yellow River in Zhengzhou and Da Yulan in Jiaozuo and used the labelme tool to annotate 800 of them.Some sample data is shown in Fig.4.

Figure 4:Sample images of slope region dataset

4.1.2 Motion Detection Dataset

To verify the effect of the background modeling algorithm on motion detection,we collected video data of dangerous accidents in Madu and Da Yulan.In addition,the landslide motion data is also simulated.The data sample is shown in Fig.5.The first line of the figure is the simulated slope collapse movement scene.The second and the third lines of the figure show the dangerous accidents of Madu and Da Yulan in Jiaozuo,respectively.

4.1.3 Interference Elimination Dataset

Since there may be human,bird,dog,and other interference sources in the slope area,the motion generated by these interference sources would also be detected by the background modeling algorithm,resulting in movement interference.To identify the moving objects and eliminate the disturbing motion,we collected nearly 2000 image data which includes pedestrians,birds,leaves,cats,and dogs from Madu and Da Yulan and divided the dataset into 10 categories,each of which contains 150 to 220 images.The samples of the dataset are shown in Fig.6.We use this dataset to train the designed lightweight convolutional neural network so that it can effectively identify the moving target,eliminate the interference movement,and improve the detection accuracy of the method.

Figure 5:Sample images of motion detection data set

Figure 6:Sample images of interference elimination dataset

4.2 Experimental Parameter Setting

For the slope segmentation experiment,we resize the images into 640 × 320 pixels and use the Pytoch framework to implement the U-net model.The SGD(Stochastic Gradient Descent)algorithm is used for optimizing the parameters.The initial learning rate is set to 0.005,the decay rate is set to 10%,and the number of training iterations is set to 50.

For the lightweight convolutional neural network,the size of input images is set to 50 × 50.The feature dimension of the first layer convolution layer is set to 256,and the feature dimensions of fire1–fire5 are set to 128,256,128,and 128,respectively.Since there are 10 sample categories in the interference elimination dataset,the feature dimension of the prediction layer is set to 10.The network is performed in the Pytorch framework and optimized by Adam (Adaptive moment estimation)algorithm.We set the initial learning rate to 0.01,the decay rate to 10%,and the number of iterations to 500.

4.3 Experimental Results and Analysis

4.3.1 Slope Segmentation Experiment

We use 400 images of the slope region segmentation dataset to train the designed U-net.The rest of the 400 images are used for testing.Some testing results are shown in Fig.7.From the figure,we can observe that the designed model can well segment the slope region.

To verify the effectiveness of the attention mechanism,we further evaluate the performance of the proposed model with the IoU(Intersection-over-Union)value.The test results are shown in Table 1,in which U-net/attentionis the model without attention mechanism and U-net is the model proposed in this paper.From the table,it can be seen that the U-net segmentation network with attention mechanism has reached a higher IoU value.Thus,the effectiveness of the attention mechanism introduced in the U-net model can be obtained.

Figure 7:Segmentation results

Table 1: Comparison results of IoU indicators

In addition,to verify the effectiveness of the proposed lightweight U-net model,we use the dataset to test the original U-net model and the designed lightweight U-net/attentionmodel.The IoU value and the time spent on the segmentation of a single picture are used for comparison.The comparison results are shown in Table 2.From the Table 2,we can see that the IoU values of the two models are almost the same,while the lightweight U-net model has less time-consuming and has higher efficiency,which demonstrates the effectiveness of the proposed lightweight model.

Table 2: Comparison of the two models

4.3.2 Movement Detection Experiment

We used the collected motion detection dataset to test the performance of the background modeling algorithm.The experimental results are shown in Fig.8.The left is the original video with motion information.The right is the corresponding results,in which the red part in each image is the detected motion region.Compared with the movement area in the video,it is concluded that the algorithm can well detect the movement generated in the slope region,verifying the effectiveness of the adopted moving target detection algorithm.

Figure 8:Motion detection results

4.3.3 Interference Elimination Experiment

We randomly select 1500 images to compose the train set for the interference elimination dataset.The remainder of 500 images is used for testing.To verify the rationality of the designed CNN model,we further designed two comparison models.One model added two fire layers based on the proposed model,naming CNN+.The other model removed fire2 and fire3 layers and named CNN-.We used the dataset to train and test the three models and used the top-1 recognition rate as the evaluating indicator.The comparative experimental results are shown in Table 3.From the table,the CNN model has achieved the highest recognition rate.In contrast,whether increasing or reducing the depth of the model will lead to the reduction of the recognition performance of the algorithm,which demonstrates the effectiveness of the proposed CNN model.

Table 3: Comparison of Top-1 results of the three models

4.3.4 Engineering Application Experiment

We embed the proposed method into the monitoring hardware equipment.On this basis,one-pole equipment for river revetment image recognition is developed.The equipment is placed at the Channel Dangerous Project of Madu on the Yellow River for 24-h unattended monitoring.As shown in Fig.9,on July 05,2020,the algorithm effectively captured the slope collapse and notified the relevant defense Bureau staff to reinforce the collapse position through network transmission equipment.

All the above experimental results show that the method proposed in this paper has a good detection effect on engineering slope collapse both in simulated data and in real scene application,which has high practical value.

Figure 9:Actual engineering application effect

5 Conclusion

Aiming at the existing problems of current engineering slope safety inspection,this paper designs an engineering slope collapse detection algorithm based on deep learning technology.Firstly,the slope area is automatically segmented and extracted,then the moving target in the slope region is detected by the background modeling algorithm.Finally,the detected moving target is effectively identified by the designed CNN model.The proposed method has a good performance on the slope collapse detecting both in simulated and real data scenes,which has a good application prospect.Future work will focus on how to use optical flow estimation based on deep learning technology to capture the motion information produced by slope collapse.

Funding Statement:This work was supported in part by the National Science Foundation of Guangxi Province under Grant 2021JJA170199,and in part by the Research Project of Yellow River Engineering Consulting with No.2021ky015.

Conflicts of Interest:The authors declare that they have no conflicts of interest to report regarding the present study.