vix.ing · top · new · best · stats · spec

Knowledge-enhanced Relation Graph and Task Sampling for Few-shot Molecular Property Prediction

2024/05/24 by Zeyu Wang, Tianyi Jiang, Wang, Zeyu +11 · 1 citation
Biochemistry, Genetics and Molecular Biology · Computer Science · Materials Science · #Artificial Intelligence (cs.AI) #Computational Drug Discovery Methods #FOS: Biological sciences #FOS: Computer and information sciences #Machine Learning (cs.LG) #Machine Learning in Materials Science #Protein Structure and Dynamics #Quantitative Methods (q-bio.QM)

paper · pdf · doi:10.48550/arxiv.2405.15544

openalex publication_date 2024/05/24 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Recently, few-shot molecular property prediction (FSMPP) has garnered increasing attention. Despite impressive breakthroughs achieved by existing methods, they often overlook the inherent many-to-many relationships between molecules and properties, which limits their performance. For instance, similar substructures of molecules can inspire the exploration of new compounds. Additionally, the relationships between properties can be quantified, with high-related properties providing more information in exploring the target property than those low-related. To this end, this paper proposes a novel meta-learning FSMPP framework (KRGTS), which comprises the Knowledge-enhanced Relation Graph module and the Task Sampling module. The knowledge-enhanced relation graph module constructs the molecule-property multi-relation graph (MPMRG) to capture the many-to-many relationships between molecules and properties. The task sampling module includes a meta-training task sampler and an auxiliary task sampler, responsible for scheduling the meta-training process and sampling high-related auxiliary tasks, respectively, thereby achieving efficient meta-knowledge learning and reducing noise introduction. Empirically, extensive experiments on five datasets demonstrate the superiority of KRGTS over a variety of state-of-the-art methods. The code is available in https://github.com/Vencent-Won/KRGTS-public.

Cited by

Related