Deep Bayesian Self-Training

Fabio De Sousa Ribeiro, Francesco Calivá, Mark Swainson, Kjartan Gudmundsson, Georgios Leontidis* (Corresponding Author), Stefanos Kollias

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

20 Citations (Scopus)
17 Downloads (Pure)

Abstract

Supervised deep learning has been highly successful in recent years, achieving state-of-the-art results in most tasks. However, with the ongoing uptake of such methods in industrial applications, the requirement for large amounts of annotated data is often a challenge. In most real-world problems, manual annotation is practically intractable due to time/labour constraints; thus, the development of automated and adaptive data annotation systems is highly sought after. In this paper, we propose both a (1) deep Bayesian self-training methodology for automatic data annotation, by leveraging predictive uncertainty estimates using variational inference and modern neural network (NN) architectures, as well as (2) a practical adaptation procedure for handling high label variability between different dataset distributions through clustering of NN latent variable representations. An experimental study on both public and private datasets is presented illustrating the superior performance of the proposed approach over standard self-training baselines, highlighting the importance of predictive uncertainty estimates in safety-critical domains.
Original languageEnglish
Pages (from-to)4275-4291
Number of pages17
JournalNeural Computing and Applications
Volume32
Early online date10 Jul 2019
DOIs
Publication statusPublished - May 2020

Bibliographical note

Acknowledgements
The authors would like to thank Mr. George Marandianos, Mrs. Mamatha Thota and Mr. Samuel Bond-Taylor for manually annotating datasets used in this study and of course the reviewers for their constructive feedback that helped to improve the manuscript. We would also like to thank Professor Luc Bidaut for enabling this collaboration.

Funding
The research presented in this paper was funded by Engineering and Physical Sciences Research Council (Reference Number EP/R005524/1) and Innovate UK (Reference Number 102908), in collaboration with the Olympus Automation Limited Company, for the project Automated Robotic Food Manufacturing System.

Keywords

  • Machine Learning
  • Deep Learning
  • Deep learning
  • Representation learning
  • Bayesian CNN
  • Variational inference
  • Clustering
  • Self-training
  • Adaptation
  • Uncertainty weighting

Fingerprint

Dive into the research topics of 'Deep Bayesian Self-Training'. Together they form a unique fingerprint.

Cite this