Abstract:Effective transfer learning for molecular property prediction has shown considerable strength in addressing insufficient labeled molecules. Many existing methods either disregard the quantitative relationship between source and target properties, risking negative transfer, or require intensive training on target tasks. To quantify transferability concerning task-relatedness, we propose Principal Gradient-based Measurement (PGM) for transferring molecular property prediction ability. First, we design an optimization-free scheme to calculate a principal gradient for approximating the direction of model optimization on a molecular property prediction dataset. We have analyzed the close connection between the principal gradient and model optimization through mathematical proof. PGM measures the transferability as the distance between the principal gradient obtained from the source dataset and that derived from the target dataset. Then, we perform PGM on various molecular property prediction datasets to build a quantitative transferability map for source dataset selection. Finally, we evaluate PGM on multiple combinations of transfer learning tasks across 12 benchmark molecular property prediction datasets and demonstrate that it can serve as fast and effective guidance to improve the performance of a target task. This work contributes to more efficient discovery of drugs, materials, and catalysts by offering a task-relatedness quantification prior to transfer learning and understanding the relationship between chemical properties. Transfer learning is known to enhance molecular property prediction in limited data sets, however, negative transfer due to insufficient relatedness between source and target tasks continues to be a major challenge. Here, the authors develop a principal gradient-based measurement to evaluate the quantitative transferability from the source property to the target property before applying transfer learning, significantly improving the transfer learning performance.

Feasibility and Transferability of Transfer Learning: A Mathematical Framework

Feasibility of Transfer Learning: A Mathematical Framework

Risk of Transfer Learning and its Applications in Finance

A Mathematical Framework for Quantifying Transferability in Multi-source Transfer Learning

Transfer Learning for Portfolio Optimization

A Bayesian Approach to (Online) Transfer Learning: Theory and Algorithms

A Visual Analytical Approach for Transfer Learning in Classification.

To transfer or not transfer: Unified transferability metric and analysis

Towards Understanding Transfer Learning Algorithms Using Meta Transfer Features

Transferring knowledge from human-demonstration trajectories to reinforcement learning

Deep Model Transferability from Attribution Maps

Transferring Knowledge across Learning Processes

An Information-Theoretic Analysis for Transfer Learning: Error Bounds and Applications

Transferability Estimation Based on Principal Gradient Expectation.

Ensemble Transfer Learning Algorithm

Fast and Effective Molecular Property Prediction with Transferability Map

Features are fate: a theory of transfer learning in high-dimensional regression

Frustratingly Easy Transferability Estimation

Towards Understanding the Transferability of Deep Representations

Transferability in Deep Learning: A Survey

Understanding Transfer Learning via Mean-field Analysis