Deep Relevance Regularization: Interpretable and Robust Tumor Typing of Imaging Mass Spectrometry Data

Abstract

Neural networks have recently been established as a viable classification method for imaging mass spectrometry data for tumor typing. For multi-laboratory scenarios however, certain confounding factors may strongly impede their performance. In this work, we introduce Deep Relevance Regularization, a method of restricting what the neural network can focus on during classification, in order to improve the classification performance. We demonstrate how Deep Relevance Regularization robustifies neural networks against confounding factors on a challenging inter-lab dataset consisting of breast and ovarian carcinoma. We further show that this makes the relevance map -- a way of visualizing the discriminative parts of the mass spectrum -- sparser, thereby making the classifier easier to interpret

References in corpus (11)

• Scikit-learn: Machine Learning in Python
• Methods for Interpreting and Understanding Deep Neural Networks
• Striving for Simplicity: The All Convolutional Net
• Explaining NonLinear Classification Decisions with Deep Taylor Decomposition
• Not Just a Black Box: Learning Important Features Through Propagating Activation Differences
• Visualizing Deep Neural Network Decisions: Prediction Difference Analysis
• Deep Learning for Tumor Classification in Imaging Mass Spectrometry
• Investigating the influence of noise and distractors on the interpretation of neural networks
• VisualBackProp: efficient visualization of CNNs
• Interpretations are useful: penalizing explanations to align neural networks with prior knowledge
• A Forward-Backward Approach for Visualizing Information Flow in Deep Networks