Training Neural Networks on Data Sources with Unknown Reliability

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Capstick, Alexander, Palermo, Francesca, Cui, Tianyu, Barnaghi, Payam
Natura: Preprint
Pubblicazione: 2022
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917921553383424
author Capstick, Alexander
Palermo, Francesca
Cui, Tianyu
Barnaghi, Payam
author_facet Capstick, Alexander
Palermo, Francesca
Cui, Tianyu
Barnaghi, Payam
contents When data is generated by multiple sources, conventional training methods update models assuming equal reliability for each source and do not consider their individual data quality. However, in many applications, sources have varied levels of reliability that can have negative effects on the performance of a neural network. A key issue is that often the quality of the data for individual sources is not known during training. Previous methods for training models in the presence of noisy data do not make use of the additional information that the source label can provide. Focusing on supervised learning, we aim to train neural networks on each data source for a number of steps proportional to the source's estimated reliability by using a dynamic re-weighting strategy motivated by likelihood tempering. This way, we allow training on all sources during the warm-up and reduce learning on less reliable sources during the final training stages, when it has been shown that models overfit to noise. We show through diverse experiments that this can significantly improve model performance when trained on mixtures of reliable and unreliable data sources, and maintain performance when models are trained on reliable sources only.
format Preprint
id arxiv_https___arxiv_org_abs_2212_02895
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Training Neural Networks on Data Sources with Unknown Reliability
Capstick, Alexander
Palermo, Francesca
Cui, Tianyu
Barnaghi, Payam
Machine Learning
When data is generated by multiple sources, conventional training methods update models assuming equal reliability for each source and do not consider their individual data quality. However, in many applications, sources have varied levels of reliability that can have negative effects on the performance of a neural network. A key issue is that often the quality of the data for individual sources is not known during training. Previous methods for training models in the presence of noisy data do not make use of the additional information that the source label can provide. Focusing on supervised learning, we aim to train neural networks on each data source for a number of steps proportional to the source's estimated reliability by using a dynamic re-weighting strategy motivated by likelihood tempering. This way, we allow training on all sources during the warm-up and reduce learning on less reliable sources during the final training stages, when it has been shown that models overfit to noise. We show through diverse experiments that this can significantly improve model performance when trained on mixtures of reliable and unreliable data sources, and maintain performance when models are trained on reliable sources only.
title Training Neural Networks on Data Sources with Unknown Reliability
topic Machine Learning
url https://arxiv.org/abs/2212.02895