Evaluating molecular representations in machine learning models for drug response prediction and interpretability

Abstract: Machine learning (ML) is increasingly being used to guide drug discovery processes. When applying ML approaches to chemical datasets, molecular descriptors and fingerprints are typically used to represent compounds as numerical vectors. However, in recent years, end-to-end deep learning (DL) methods that can learn feature representations directly from line notations or molecular graphs have been proposed as alternatives to using precomputed features. This study set out to investigate which compound representation methods are the most suitable for drug sensitivity prediction in cancer cell lines. Twelve different representations were benchmarked on 5 compound screening datasets, using DeepMol, a new chemoinformatics package developed by our research group, to perform these analyses. The results of this study show that the predictive performance of end-to-end DL models is comparable to, and at times surpasses, that of models trained on molecular fingerprints, even when less training data is available. This study also found that combining several compound representation methods into an ensemble can improve performance. Finally, we show that a post hoc feature attribution method can boost the explainability of the DL models.

Location
Deutsche Nationalbibliothek Frankfurt am Main
Extent
Online-Ressource
Language
Englisch

Bibliographic citation
Evaluating molecular representations in machine learning models for drug response prediction and interpretability ; volume:19 ; number:3 ; year:2022 ; extent:13
Journal of integrative bioinformatics ; 19, Heft 3 (2022) (gesamt 13)

Creator
Baptista, Delora
Correia, João
Pereira, Bruno
Rocha, Miguel

DOI
10.1515/jib-2022-0006
URN
urn:nbn:de:101:1-2022093014041498516179
Rights
Open Access; Der Zugriff auf das Objekt ist unbeschränkt möglich.
Last update
15.08.2025, 7:24 AM CEST

Data provider

This object is provided by:
Deutsche Nationalbibliothek. If you have any questions about the object, please contact the data provider.

Associated

  • Baptista, Delora
  • Correia, João
  • Pereira, Bruno
  • Rocha, Miguel

Other Objects (12)