Contrastive Multi-Task Learning with Solvent-Aware Augmentation for Drug Discovery
Accurate prediction of protein-ligand interactions is essential for computer-aided drug discovery. However, existing methods often fail to capture solvent-dependent conformational changes and lack the ability to jointly learn multiple related tasks. To address these limitations, we introduce a pre-training method that incorporates ligand conformational ensembles generated under diverse solvent conditions as augmented input. This design enables the model to learn both structural flexibility and environmental context in a unified manner. The training process integrates molecular reconstruction to capture local geometry, interatomic distance prediction to model spatial relationships, and contrastive learning to build solvent-invariant molecular representations. Together, these components lead to significant improvements, including a 3.7% gain in binding affinity prediction, an 82% success rate on the PoseBusters Astex docking benchmarks, and an area under the curve of 97.1% in virtual screening. The framework supports solvent-aware, multi-task modeling and produces consistent results across benchmarks. A case study further demonstrates sub-angstrom docking accuracy with a root-mean-square deviation of 0.157 angstroms, offering atomic-level insight into binding mechanisms and advancing structure-based drug design.
Code (0)
등록된 구현이 없습니다.
Tasks
Contrastive LearningMulti-Task LearningDrug DiscoverySimilar Papers 제목 키워드 기반
Object-aware Contrastive Learning for Debiased Scene Representation
Contrastive self-supervised learning has shown impressive results in learning visual representations from unlabeled images by enforcing invariance against different data augmentations. However, the learned representation…
Contrastive LearningObjectRepresentation LearningSelf-Supervised LearningGenerative-Contrastive Heterogeneous Graph Neural Network
Heterogeneous Graphs (HGs) effectively model complex relationships in the real world through multi-type nodes and edges. In recent years, inspired by self-supervised learning (SSL), contrastive learning (CL)-based Hetero…
Contrastive LearningData AugmentationGraph Neural NetworkLink Prediction+2Time Series Contrastive Learning with Information-Aware Augmentations
Various contrastive learning approaches have been proposed in recent years and achieve significant empirical success. While effective and prevalent, contrastive learning has been less explored for time series data. A key…
Contrastive LearningOpen-Ended Question AnsweringRepresentation LearningTime SeriesHyperbolic Contrastive Learning with Model-augmentation for Knowledge-aware Recommendation
Benefiting from the effectiveness of graph neural networks (GNNs) and contrastive learning, GNN-based contrastive learning has become mainstream for knowledge-aware recommendation. However, most existing contrastive lear…
Contrastive LearningKnowledge-Aware RecommendationKnowledge GraphsHiTeC: Hierarchical Contrastive Learning on Text-Attributed Hypergraph with Semantic-Aware Augmentation
Contrastive learning (CL) has become a dominant paradigm for self-supervised hypergraph learning, enabling effective training without costly labels. However, node entities in real-world hypergraphs are often associated w…
Self-Supervised LearningRepresentation LearningContrastive Learning