Interpreting linear support vector machine models with heat map molecule coloring

publication · 6 years ago
by Lars Rosenbaum, Georg Hinselmann, Andreas Jahn, Andreas Zell (University of Tübingen)

Background: Model-based virtual screening plays an important role in the early drug discovery stage. The out- comes of high-throughput screenings are a valuable source for machine learning algorithms to infer such models. Besides a strong performance, the interpretability of a machine learning model is a desired property to guide the optimization of a compound in later drug discovery stages. Linear support vector machines showed to have a convincing performance on large-scale data sets. The goal of this study is to present a heat map molecule coloring technique to interpret linear support vector machine models. Based on the weights of a linear model, the visualization approach colors each atom and bond of a compound according to its importance for activity.

Results: We evaluated our approach on a toxicity data set, a chromosome aberration data set, and the maximum unbiased validation data sets. The experiments show that our method sensibly visualizes structure-property and structure-activity relationships of a linear support vector machine model. The coloring of ligands in the binding pocket of several crystal structures of a maximum unbiased validation data set target indicates that our approach assists to determine the correct ligand orientation in the binding pocket. Additionally, the heat map coloring enables the identi cation of substructures important for the binding of an inhibitor.

Conclusions: In combination with heat map coloring, linear support vector machine models can help to guide the modi cation of a compound in later stages of drug discovery. Particularly substructures identi ed as important by our method might be a starting point for optimization of a lead compound. The heat map coloring should be considered as complementary to structure based modeling approaches. As such, it helps to get a better understanding of the binding mode of an inhibitor.

Visit publication