Reconstruction of missing data in transferred generative adversarial networks with small sample data

Jing He, Hongrun Chen and Zhenwen Sheng

PLOS ONE, 2025, vol. 20, issue 6, 1-17

Abstract: Under special working conditions, data collection systems of heavy-duty trains may be faced with a small sample size and missing data when executing measurement, operation, or maintenance tasks. Existing generative modeling methods are ineffective in reconstructing missing data with such a small sample. Hence, we set up a frame of migration learning generative adversarial network for small data samples, in which a new variational autoencoder semantic fusion generative adversarial network (VAE-FGAN) is developed to reconstruct missing data. First a GRU module is introduced in the encoder to fuse the underlying features of the data with higher-level features, which enables the VAE-FGAN to learn the correlation between the measured data through unsupervised training. Second, an SE-NET attention mechanism is introduced into the whole generative network to enhance the expression of the feature extraction network on data features. Finally, parameters are shared through migration learning and pre-training, thereby eliminating the difficulty in training the model due to the small size of certain operation and maintenance data. Experimental results show that the reconstruction accuracy indices MAE and MAPE can be kept below 1.5 when measured data is missing; the reconstructed data also fits well to the distribution trend of the measured data.

Date: 2025
References: View complete reference list from CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0322323 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 22323&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0322323

DOI: 10.1371/journal.pone.0322323

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().