Tanimoto Random Features for Scalable Molecular Machine Learning
Abstract
The Tanimoto coefficient is commonly used to measure the similarity between molecules represented as discrete fingerprints, either as a distance metric or a positive definite kernel. While many kernel methods can be accelerated using random feature approximations, at present there is a lack of such approximations for the Tanimoto kernel. In this paper we propose two kinds of novel random features to allow this kernel to scale to large datasets, and in the process discover a novel extension of the kernel to real-valued vectors. We theoretically characterize these random features, and provide error bounds on the spectral norm of the Gram matrix. Experimentally, we show that these random features are effective at approximating the Tanimoto coefficient of real-world datasets and are useful for molecular property prediction and optimization tasks.
Keywords
Cite
@article{arxiv.2306.14809,
title = {Tanimoto Random Features for Scalable Molecular Machine Learning},
author = {Austin Tripp and Sergio Bacallado and Sukriti Singh and José Miguel Hernández-Lobato},
journal= {arXiv preprint arXiv:2306.14809},
year = {2023}
}
Comments
Camera-ready version presented at NeurIPS 2023. Updates include: notation changes, better description of features in section 4, updated experiments, link to code