Large-Scale Knowledge Integration for Enhanced Molecular Property Prediction

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ghunaim, Yasir, Hoehndorf, Robert
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916440414617600
author Ghunaim, Yasir
Hoehndorf, Robert
author_facet Ghunaim, Yasir
Hoehndorf, Robert
contents Pre-training machine learning models on molecular properties has proven effective for generating robust and generalizable representations, which is critical for advancements in drug discovery and materials science. While recent work has primarily focused on data-driven approaches, the KANO model introduces a novel paradigm by incorporating knowledge-enhanced pre-training. In this work, we expand upon KANO by integrating the large-scale ChEBI knowledge graph, which includes 2,840 functional groups -- significantly more than the original 82 used in KANO. We explore two approaches, Replace and Integrate, to incorporate this extensive knowledge into the KANO framework. Our results demonstrate that including ChEBI leads to improved performance on 9 out of 14 molecular property prediction datasets. This highlights the importance of utilizing a larger and more diverse set of functional groups to enhance molecular representations for property predictions. Code: github.com/Yasir-Ghunaim/KANO-ChEBI
format Preprint
id arxiv_https___arxiv_org_abs_2410_11914
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Large-Scale Knowledge Integration for Enhanced Molecular Property Prediction
Ghunaim, Yasir
Hoehndorf, Robert
Quantitative Methods
Machine Learning
Chemical Physics
Pre-training machine learning models on molecular properties has proven effective for generating robust and generalizable representations, which is critical for advancements in drug discovery and materials science. While recent work has primarily focused on data-driven approaches, the KANO model introduces a novel paradigm by incorporating knowledge-enhanced pre-training. In this work, we expand upon KANO by integrating the large-scale ChEBI knowledge graph, which includes 2,840 functional groups -- significantly more than the original 82 used in KANO. We explore two approaches, Replace and Integrate, to incorporate this extensive knowledge into the KANO framework. Our results demonstrate that including ChEBI leads to improved performance on 9 out of 14 molecular property prediction datasets. This highlights the importance of utilizing a larger and more diverse set of functional groups to enhance molecular representations for property predictions. Code: github.com/Yasir-Ghunaim/KANO-ChEBI
title Large-Scale Knowledge Integration for Enhanced Molecular Property Prediction
topic Quantitative Methods
Machine Learning
Chemical Physics
url https://arxiv.org/abs/2410.11914