Interpretable QSPR Modeling using Recursive Feature Machines and Multi-scale Fingerprints

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Shen, Jiaxuan, Zhang, Haitao, Wang, Yunjie, Wang, Yilong, Tao, Song, Qiu, Bo, Shyh-Chang, Ng
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916490708516864
author Shen, Jiaxuan
Zhang, Haitao
Wang, Yunjie
Wang, Yilong
Tao, Song
Qiu, Bo
Shyh-Chang, Ng
author_facet Shen, Jiaxuan
Zhang, Haitao
Wang, Yunjie
Wang, Yilong
Tao, Song
Qiu, Bo
Shyh-Chang, Ng
contents This study pioneers the application of Recursive Feature Machines (RFM) in QSPR modeling, introducing a tailored feature importance analysis approach to enhance interpretability. By leveraging deep feature learning through AGOP, RFM achieves state-of-the-art (SOTA) results in predicting molecular properties, as demonstrated through solubility prediction across nine benchmark datasets. To capture a wide array of structural information, we employ diverse molecular representations, including MACCS keys, Morgan fingerprints, and a custom multi-scale hybrid fingerprint (HF) derived from global descriptors and SMILES local fragmentation techniques. Notably, the HF offers significant advantages over MACCS and Morgan fingerprints in revealing structural determinants of molecular properties. The feature importance analysis in RFM provides robust local and global explanations, effectively identifying structural features that drive molecular behavior and offering valuable insights for drug development. Additionally, RFM demonstrates strong redundancy-filtering abilities, as model performance remains stable even after removing redundant features within custom fingerprints. Importantly, RFM introduces the deep feature learning capabilities of the average gradient outer product (AGOP) matrix into ultra-fast kernel machine learning, to imbue kernel machines with interpretable deep feature learning capabilities. We extend this approach beyond the Laplace Kernel to the Matern, Rational Quadratic, and Gaussian kernels, to find that the Matern and Laplace kernels deliver the best performance, thus reinforcing the flexibility and effectiveness of AGOP in RFM. Experimental results show that RFM-HF surpasses both traditional machine learning models and advanced graph neural networks.
format Preprint
id arxiv_https___arxiv_org_abs_2411_14079
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Interpretable QSPR Modeling using Recursive Feature Machines and Multi-scale Fingerprints
Shen, Jiaxuan
Zhang, Haitao
Wang, Yunjie
Wang, Yilong
Tao, Song
Qiu, Bo
Shyh-Chang, Ng
Biomolecules
This study pioneers the application of Recursive Feature Machines (RFM) in QSPR modeling, introducing a tailored feature importance analysis approach to enhance interpretability. By leveraging deep feature learning through AGOP, RFM achieves state-of-the-art (SOTA) results in predicting molecular properties, as demonstrated through solubility prediction across nine benchmark datasets. To capture a wide array of structural information, we employ diverse molecular representations, including MACCS keys, Morgan fingerprints, and a custom multi-scale hybrid fingerprint (HF) derived from global descriptors and SMILES local fragmentation techniques. Notably, the HF offers significant advantages over MACCS and Morgan fingerprints in revealing structural determinants of molecular properties. The feature importance analysis in RFM provides robust local and global explanations, effectively identifying structural features that drive molecular behavior and offering valuable insights for drug development. Additionally, RFM demonstrates strong redundancy-filtering abilities, as model performance remains stable even after removing redundant features within custom fingerprints. Importantly, RFM introduces the deep feature learning capabilities of the average gradient outer product (AGOP) matrix into ultra-fast kernel machine learning, to imbue kernel machines with interpretable deep feature learning capabilities. We extend this approach beyond the Laplace Kernel to the Matern, Rational Quadratic, and Gaussian kernels, to find that the Matern and Laplace kernels deliver the best performance, thus reinforcing the flexibility and effectiveness of AGOP in RFM. Experimental results show that RFM-HF surpasses both traditional machine learning models and advanced graph neural networks.
title Interpretable QSPR Modeling using Recursive Feature Machines and Multi-scale Fingerprints
topic Biomolecules
url https://arxiv.org/abs/2411.14079