Rxivist logo

Exploration of Chemical Space with Partial Labeled Noisy Student Self-Training for Improving Deep Learning: Application to Drug Metabolism

By Yang Liu, Hansaim Lim, Lei Xie

Posted 06 Aug 2020
bioRxiv DOI: 10.1101/2020.08.06.239988

Motivation Drug discovery is time-consuming and costly. Machine learning, especially deep learning, shows a great potential in accelerating the drug discovery process and reducing the cost. A big challenge in developing robust and generalizable deep learning models for drug design is the lack of data with high quality and balanced labels. To address this challenge, we developed a self-training method PLANS that exploits millions of unlabeled chemical compounds as well as partially labeled pharmacological data to improve the generalizability of neural network models. Result We evaluated the self-training with PLANS for Cytochrome P450 binding activity prediction task, and proved that our method could significantly improve the performance of the neural network model with a large margin. Compared with the baseline model, the PLANS-trained neural network model improved accuracy, precision, recall, and F1 score by 13.4%, 12.5%, 8.3%, and 10.3%, respectively. The self-training with PLANS is model agnostic, and can be applied to any deep learning architectures. Thus, PLANS provides a general solution to utilize unlabeled and partially labeled data to improve the predictive modeling for drug discovery. ### Competing Interest Statement The authors have declared no competing interest.

Download data

  • Downloaded 63 times
  • Download rankings, all-time:
    • Site-wide: 119,904
    • In bioinformatics: 9,666
  • Year to date:
    • Site-wide: None
  • Since beginning of last month:
    • Site-wide: 90,277

Altmetric data

Downloads over time

Distribution of downloads per paper, site-wide


Sign up for the Rxivist weekly newsletter! (Click here for more details.)