Laser & Optoelectronics Progress, Volume. 61, Issue 12, 1237006(2024)

Anomaly Detection Algorithm of Ballastless Track Bed Based on Image Inpainting

Wan Jiang, Kai Yang*, Chunrong Qiu, and Liming Xie
Author Affiliations
  • School of Physical Science and Technology, Southwest Jiaotong University, Chengdu 610031, Sichuan, China
  • show less

    Accurate detection of foreign objects on ballastless railway track beds is crucial for ensuring train safety. Although unsupervised anomaly detection algorithms based on deep learning address the effect of insufficient abnormal data on detection, the "generalization" ability of the encoder can proficiently reconstruct anomalous instances, thereby affecting detection accuracy. To solve this problem, this study proposes an anomaly detection framework for ballastless track beds utilizing image inpainting. First, inpainting was employed to obscure and subsequently restore the image using training on nonanomalous and incomplete image data, aiming to improve the model's contextual semantic understanding and enhance its reconstruction ability. Second, the maximum value obtained from the average anomaly map of the test and reconstructed images, which was analyzed across multiple scales, was utilized as the reconstruction error to calculate the anomaly score. This step aimed to widen the reconstruction error boundary between the abnormal and normal images. Finally, experimental results show a notable advantage of the proposed algorithm over alternative methods on public datasets, such as MNIST, CIFAR-10, and the ballastless track bed dataset.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Wan Jiang, Kai Yang, Chunrong Qiu, Liming Xie. Anomaly Detection Algorithm of Ballastless Track Bed Based on Image Inpainting[J]. Laser & Optoelectronics Progress, 2024, 61(12): 1237006

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Digital Image Processing

    Received: May. 15, 2023

    Accepted: Sep. 18, 2023

    Published Online: May. 29, 2024

    The Author Email: Kai Yang (yangkai_swjtu@163.com)

    DOI:10.3788/LOP231318

    CSTR:32186.14.LOP231318

    Topics