Pre-Training on In Vitro and Fine-Tuning on Patient-Derived Data Improves Deep Neural Networks for Anti-Cancer

Paul Prasse1, Pascal Iversen1, Matthias Lienhard2

  • 1Department of Computer Science, University of Potsdam, 14476 Potsdam, Germany.

Cancers
|August 26, 2022
PubMed

Insights

Pre-training machine learning models for drug sensitivity on in vitro data before fine-tuning with patient-derived data enhances accuracy. This approach improves preclinical drug development by leveraging existing data more effectively.

Area of Science:

  • Computational biology
  • Machine learning in oncology
  • Drug discovery

Background:

  • Machine learning models for drug sensitivity are trained on large in vitro cancer cell line databases.
  • In vitro cell lines may not accurately represent human tumors due to long-term culture adaptation.
  • Existing patient-derived data is insufficient for training high-capacity machine learning models.

Purpose of the Study:

  • To improve the accuracy and biological relevance of machine learning models for predicting drug sensitivity.
  • To investigate the efficacy of pre-training deep neural network models on in vitro data before fine-tuning on patient-derived data.

Main Methods:

  • Utilized large-scale in vitro drug-sensitivity databases for pre-training deep neural network models.
  • Fine-tuned pre-trained models using patient-derived drug-response data (cell cultures, xenografts, organoids).
  • Compared the performance of pre-trained models against models trained solely on patient-derived data.

Main Results:

  • Pre-training significantly improved the accuracy of drug-sensitivity prediction models.
  • The biological plausibility of features learned by pre-trained models was enhanced.
  • Pre-trained models demonstrated superior performance compared to models trained exclusively on patient-derived data in most cases.

Conclusions:

  • Pre-training deep neural networks on in vitro drug-sensitivity data is a viable strategy to boost model performance.
  • This approach addresses data scarcity issues with patient-derived datasets.
  • Enhanced models offer improved accuracy and biological relevance for preclinical drug development.