A Semi-supervised Molecular Learning Framework for Activity Cliff Estimation
A Semi-supervised Molecular Learning Framework for Activity Cliff Estimation
Fang Wu
Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence
Main Track. Pages 6080-6088.
https://doi.org/10.24963/ijcai.2024/672
Machine learning (ML) enables accurate and fast molecular property predictions, which is of interest in drug discovery and material design. Their success is based on the principle of similarity at its heart, assuming that similar molecules exhibit close properties. However, activity cliffs challenge this principle, and their presence leads to a sharp decline in the performance of existing ML algorithms, particularly graph-based methods. To overcome this obstacle under a low-data scenario, we propose a novel semi-supervised learning (SSL) method dubbed SemiMol, which employs predictions on numerous unannotated data as pseudo-signals for subsequent training. Specifically, we introduce an additional instructor model to evaluate the accuracy and trustworthiness of proxy labels because existing pseudo-labeling approaches require probabilistic outputs to reveal the model's confidence and fail to be applied in regression tasks. Moreover, we design a self-adaptive curriculum learning algorithm to progressively move the target model toward hard samples at a controllable pace. Extensive experiments on 30 activity cliff datasets demonstrate that SemiMol significantly enhances graph-based ML architectures and outpasses state-of-the-art pretraining and SSL baselines.
Keywords:
Multidisciplinary Topics and Applications: MTA: Health and medicine
Multidisciplinary Topics and Applications: MTA: Life sciences
Humans and AI: HAI: Applications