高级检索

基于高斯混合特征改进的自编码器机场跑道FOD精准检测

Gaussian Mixture Feature Improved Autoencoder for FOD Detection on Airport Runway

  • 摘要: 基于自编码器的图像重构方法可以检测机场跑道上类型未知的外来物. 针对自编码器良好的泛化能力导致异常重构的问题, 提出一种基于高斯混合特征改进的自编码器异常检测模型. 首先通过编码器获取输入图像的编码特征, 由高斯混合模型识别并剔除其中潜在的异常特征以进行重构; 在训练阶段, 通过联合估计网络对自编码器和高斯混合模型进行端到端优化, 建立准确的正常样本模型; 在测试阶段, 重构图像由剔除异常信息的编码特征生成, 倾向于接近正常图像, 因此异常的重构误差较大; 为了进一步提高异常检测精度, 使用多层级语义差异进行异常检测. 在中国某机场的跑道道面数据集FODAD和公共基准异常检测数据集MVTec AD上的实验结果表明, 所提模型在FODAD数据集上平均AUC达0.981, 在MVTec AD数据集上平均AUC达0.985, 均优于对比的同类方法.

     

    Abstract: Foreign Object Debris (FOD) detection on airport runway is essential for ensuring flight safety since FOD poses a safety threat to aircraft during takeoff or landing. Image Reconstruction based on autoencoder (AE) is a commonly used method for anomaly detection, which assumes that the model trained only using normal samples cannot accurately reconstruct previously unseen anomalies, such as unknown types of FOD. However, deep neural networks have strong generalization capability and accurately reconstruct these anomalies, thereby compromising anomaly detection performance. If the features fed into the decoder do not contain anomalous information, the likelihood of accurately reconstructing anomalies will be significantly reduced. Based on this observation, the Gaussian Mixture Feature Improved Autoencoder (GMFIAE) is proposed for anomaly detection. Given an input image, GMFIAE first obtains the encoding features through the encoder, and then the Gaussian mixture model (GMM) identifies and eliminates potential abnormal features for reconstruction. At the training stage, GMFIAE combines the estimation network to perform end-to-end optimization of AE and GMM to build an accurate normal model. At the test stage, the reconstructed image is obtained from the encoding feature that removes abnormal information, which will tend to be close to the normal image, resulting in a large reconstruction error for anomalies. Furthermore, the multiple hierarchical semantic difference is employed to improve the accuracy of the anomaly detection. The experimental validation is carried out on the runway pavement dataset FODAD from an airport in China and the public anomaly detection dataset MVTec AD. The experimental results demonstrate that the proposed method can achieve accurate anomaly detection.

     

/

返回文章
返回