• Home
  • Search
  • Shape-Consistent One-Shot Unsupervised Domain Adaptation for Rail Surface Defect Segmentation
  • Cite Icon56
  • https://doi.org/10.1109/tii.2022.3233654Copy DOI Icon

Shape-Consistent One-Shot Unsupervised Domain Adaptation for Rail Surface Defect Segmentation

Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Deep neural networks have greatly improved the performance of rail surface defect segmentation when the test samples have the same distribution as the training samples. However, in practical inspection scenarios, the rail surface exhibits variations in appearance due to different service times and natural conditions. Conventional deep learning models show limited generalization in scenes with distribution differences. To address this problem, we propose a novel one-shot unsupervised domain adaptation framework. Specifically, we introduce a shape-consistent style transfer module that performs pixel-level distribution alignment between the training and test images. Based on the one-shot test image, the training image is reconstructed to have the same appearance as the test image. Meanwhile, we employ a multitask learning strategy to prevent content distortion of the reconstructed images. To improve the robustness of the model to distribution differences, we design an edge-aware defect segmentation model and train the model using the reconstructed training images. The experimental results show that our method effectively improves the robustness of the model to distribution differences and achieves satisfying results in the task of rail surface defect segmentation.

Similar Papers
  • Conference Article
  • Citations2

Neural Net-Based Algorithms Comparison for Facial Gender Classification

  • Dec 01, 2009
  • Liu Zunxiong +2
  • Research Article
  • Citations16

The impact of 2D cine MR imaging parameters on automated tumor and organ localization for MR-guided real-time adaptive radiotherapy

  • Nov 22, 2018
  • Physics in Medicine & Biology
  • Martin J Menten +7
  • Research Article
  • Citations4

Sparse representation with multi-manifold analysis for texture classification from few training images

  • Jul 11, 2014
  • Image and Vision Computing
  • Xiangping Sun +3
  • Research Article
  • Citations4

Novel Deep Neural Network Model for Handwritten Digit Classification and Recognition

  • Feb 14, 2021
  • International Journal of Advanced Research in Science, Communication and Technology
  • Ayush Kumar Agrawal +1
  • Research Article
  • Citations131

Robust segmentation of arterial walls in intravascular ultrasound images using Dual Path U-Net

  • Mar 23, 2019
  • Ultrasonics
  • Ji Yang +2
  • Conference Article
  • Citations3

The Effect of Image Dimension and Exposure Fusion Framework Enhancement in Pneumonia Detection Using Residual Neural Network

  • Sep 17, 2022
  • Agung W Setiawan
  • PDF
  • Research Article
  • Citations3

IMAGE LABELING FOR LIDAR INTENSITY IMAGE USING K-NN OF FEATURE OBTAINED BY CONVOLUTIONAL NEURAL NETWORK

  • Jun 10, 2016
  • The International Archives of the Photogrammetry, Remote Sensing and Spatial Information Sciences
  • Masaki Umemura +3
  • Research Article
  • Citations8

Weakly Supervised Semantic and Attentive Data Mixing Augmentation for Fine-Grained Visual Categorization

  • Jan 01, 2022
  • IEEE Access
  • Mengqi He +2
  • Research Article

Improved segmentation of cardiac structures in echocardiograms for diastolic function evaluation.

  • Mar 01, 2026
  • Medical physics
  • Chengcong Lv +5
  • PDF
  • Research Article

Smart Detection of Macro Nutrient Deficiency in Soybean Plant using Convolutional Neural Network

  • Nov 12, 2025
  • LEGUME RESEARCH - AN INTERNATIONAL JOURNAL
  • R Bhavani
  • Conference Article
  • Citations178

Weakly supervised semantic segmentation with a multi-image model

  • Nov 01, 2011
  • Alexander Vezhnevets +2
  • Book Chapter
  • Citations5

University of Glasgow at ImageCLEF 2009 Robot Vision Task: A Rule Based Approach

  • Jan 01, 2010
  • Yue Feng +2
  • Research Article
  • Citations19

Can deep convolutional neural networks support relational reasoning in the same-different task?

  • Sep 12, 2022
  • Journal of Vision
  • Guillermo Puebla +1
  • Conference Article
  • Citations11

Robust face recognition using automatic age normalization

  • May 29, 2000
  • A Lanitis +1
  • Research Article
  • Citations938

Toward automatic simulation of aging effects on face images

  • Apr 01, 2002
  • IEEE Transactions on Pattern Analysis and Machine Intelligence
  • A Lanitis +2
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.