Critiquing Protein Family Classification Models Using Sufficient Input Subsets

J Comput Biol. 2020 Aug;27(8):1219-1231. doi: 10.1089/cmb.2019.0339. Epub 2019 Dec 23.

Abstract

In many application domains, neural networks are highly accurate and have been deployed at large scale. However, users often do not have good tools for understanding how these models arrive at their predictions. This has hindered adoption in fields such as the life and medical sciences, where researchers require that models base their decisions on underlying biological phenomena rather than peculiarities of the dataset. We propose a set of methods for critiquing deep learning models and demonstrate their application for protein family classification, a task for which high-accuracy models have considerable potential impact. Our methods extend the Sufficient Input Subsets (SIS) technique, which we use to identify subsets of features in each protein sequence that are alone sufficient for classification. Our suite of tools analyzes these subsets to shed light on the decision-making criteria employed by models trained on this task. These tools show that while deep models may perform classification for biologically relevant reasons, their behavior varies considerably across the choice of network architecture and parameter initialization. While the techniques that we develop are specific to the protein sequence classification task, the approach taken generalizes to a broad set of scientific contexts in which model interpretability is essential.

Keywords: interpretability; machine learning; model selection; neural networks; protein classification; protein domain.

MeSH terms

  • Computational Biology*
  • Deep Learning
  • Humans
  • Machine Learning
  • Models, Biological*
  • Multigene Family / genetics*
  • Neural Networks, Computer
  • Proteins / classification*
  • Proteins / genetics

Substances

  • Proteins