Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Journal articles
  4. Metric learning for kernel ridge regression: assessment of molecular similarity
 
research article

Metric learning for kernel ridge regression: assessment of molecular similarity

Fabregat, Raimon  
•
van Gerwen, Puck  
•
Haeberle, Matthieu  
Show more
September 1, 2022
Machine Learning: Science and Technology

Supervised and unsupervised kernel-based algorithms widely used in the physical sciences depend upon the notion of similarity. Their reliance on pre-defined distance metrics-e.g. the Euclidean or Manhattan distance-are problematic especially when used in combination with high-dimensional feature vectors for which the similarity measure does not well-reflect the differences in the target property. Metric learning is an elegant approach to surmount this shortcoming and find a property-informed transformation of the feature space. We propose a new algorithm for metric learning specifically adapted for kernel ridge regression (KRR): metric learning for kernel ridge regression (MLKRR). It is based on the Metric Learning for Kernel Regression framework using the Nadaraya-Watson estimator, which we show to be inferior to the KRR estimator for typical physics-based machine learning tasks. The MLKRR algorithm allows for superior predictive performance on the benchmark regression task of atomisation energies of QM9 molecules, as well as generating more meaningful low-dimensional projections of the modified feature space.

  • Files
  • Details
  • Metrics
Loading...
Thumbnail Image
Name

Fabregat_2022_Mach._Learn.__Sci._Technol._3_035015.pdf

Type

Publisher

Version

Published version

Access type

openaccess

License Condition

CC BY

Size

1.97 MB

Format

Adobe PDF

Checksum (MD5)

4ad50e61aeb3cf5023c701c7673703a2

Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés