A self-supervised deep learning method for data-efficient training in genomics.

Journal: Communications biology
Published Date:

Abstract

Deep learning in bioinformatics is often limited to problems where extensive amounts of labeled data are available for supervised classification. By exploiting unlabeled data, self-supervised learning techniques can improve the performance of machine learning models in the presence of limited labeled data. Although many self-supervised learning methods have been suggested before, they have failed to exploit the unique characteristics of genomic data. Therefore, we introduce Self-GenomeNet, a self-supervised learning technique that is custom-tailored for genomic data. Self-GenomeNet leverages reverse-complement sequences and effectively learns short- and long-term dependencies by predicting targets of different lengths. Self-GenomeNet performs better than other self-supervised methods in data-scarce genomic tasks and outperforms standard supervised training with ~10 times fewer labeled training data. Furthermore, the learned representations generalize well to new datasets and tasks. These findings suggest that Self-GenomeNet is well suited for large-scale, unlabeled genomic datasets and could substantially improve the performance of genomic models.

Authors

  • Hüseyin Anil Gündüz
    Department of Statistics, LMU Munich, Munich, Germany.
  • Martin Binder
    Department of Statistics, LMU Munich, Munich, Germany.
  • Xiao-Yin To
    Department of Statistics, LMU Munich, Munich, Germany.
  • René Mreches
    Department for Computational Biology of Infection Research, Helmholtz Center for Infection Research, Braunschweig, Germany.
  • Bernd Bischl
    Department of Statistics, Ludwig-Maximilians-University, Munich, Germany.
  • Alice C McHardy
    Computational Biology of Infection Research, Helmholtz Centre for Infection Research, Braunschweig, Germany.
  • Philipp C Münch
    Department for Computational Biology of Infection Research, Helmholtz Center for Infection Research, Braunschweig, Germany.
  • Mina Rezaei
    Department of Statistics, LMU Munich, Munich, Germany. mina.rezaei@stat.uni-muenchen.de.