Detail publikace

Data pre-processing effect on classification accuracy of convolutional neural networks for train type identification

KRČ, R. PODROUŽEK, J. VUKUŠIČ, I. PLÁŠEK, O.

Originální název

Data pre-processing effect on classification accuracy of convolutional neural networks for train type identification

Typ

článek ve sborníku mimo WoS a Scopus

Jazyk

angličtina

Originální abstrakt

Accelerometer data collected by in-situ measurements near the common crossing from two locations in the Czech Republic were used for training and validation of machine learning models. Several architectures of convolutional neural networks (CNN), successfully applied in the previous research for electrical grid analysis, were evaluated in this paper for the problem of locomotive type identification with regards to the number of parameters and the size of the available dataset, which was limited in this case. Therefore, time-series pre-processing techniques aiming to improve classification accuracy by removing noise were incorporated, including Butterworth low-pass and high-pass filters as well as Wavelet threshold filter. Results for raw and filtered data are presented as mean confusion matrices to evaluate the statistical significance of the adopted methods.

Klíčová slova

Railway Switches and Crossings, Train Identification System, Convolutional Neural Network, Accelerometer Data, Time Series Processing, Wavelet Transform

Autoři

KRČ, R.; PODROUŽEK, J.; VUKUŠIČ, I.; PLÁŠEK, O.

Vydáno

7. 6. 2021

Strany od

1

Strany do

1

Strany počet

1

BibTex

@inproceedings{BUT171770,
  author="Rostislav {Krč} and Jan {Podroužek} and Ivan {Vukušič} and Otto {Plášek}",
  title="Data pre-processing effect on classification accuracy of convolutional neural networks for train type identification",
  booktitle="Computational Science and AI in Industry (CSAI 2021)",
  year="2021",
  pages="1--1"
}