Detail publikačního výsledku

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

Anglický název

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

Druh

Stať ve sborníku mimo WoS a Scopus

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.

Anglický 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

Klíčová slova v angličtině

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.

Rok RIV

2022

Vydáno

07.06.2021

Kniha

Computational Science and AI in Industry (CSAI 2021)

Strany od

1

Strany do

1

Strany počet

1

Plný text v Digitální knihovně

http://hdl.handle.net/

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"
}