Optics and Precision Engineering, Volume. 31, Issue 12, 1816(2023)

Railway few-shot intruding objects detection method with metric meta learning

Baoqing GUO1,2、* and Defen ZHANG1
Author Affiliations
  • 1School of Mechanical, Electronic and Control Engineering, Beijing Jiaotong University, Beijing 00044, China
  • 2Frontiers Science Center for Smart High-speed Railway System, Beijing Jiaotong University, Beijing 100044, China
  • show less

    Object intrusion is among the primary causes of railway accidents. Typically, traditional deep-learning methods require numerous samples for network training; however, intrusion samples in railway settings are scarce and difficult to obtain. Thus, in this paper, a railway few-shot intruding-object detection method based on an improved metric meta-learning network is proposed. To better exploit the features of intruding objects during classification, a feature-extraction network based on the channel attention mechanism is proposed. A network based on fine-tuning of the class center is proposed for class-center correction to solve the problem of individual samples deviating in the feature space of insufficient samples. Additionally, a central correlation loss function based on the center loss and cross entropy is constructed for few-shot network training to improve the compactness of the same-class feature distribution in the feature space. In experiments on a public few-shot dataset called miniImageNet, the accuracy of the proposed method is 7.31% higher than the optimal accuracy of the classical few-shot learning model. In five-way five-shot ablation experiments using a railway dataset, the proposed channel attention mechanism and center-related loss function increase the mean average precision (mAP) by 0.86% and 1.91%, respectively. Additionally, the center fine-tuning and pretraining increase the mAP by 3.05% and 6.70%, respectively, and the total mAP improvement is 7.90%.

    Tools

    Get Citation

    Copy Citation Text

    Baoqing GUO, Defen ZHANG. Railway few-shot intruding objects detection method with metric meta learning[J]. Optics and Precision Engineering, 2023, 31(12): 1816

    Download Citation

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

    Category: Information Sciences

    Received: Jun. 14, 2022

    Accepted: --

    Published Online: Jul. 25, 2023

    The Author Email: GUO Baoqing (bqguo@bjtu.edu.cn)

    DOI:10.37188/OPE.20233112.1816

    Topics