arrow
Return

Kernel-elastic autoencoder for molecular design

delete2024-04-25
delete1
delete
OA
AI
H
Haote Li
Y
Yu Shee
B
Brandon Allen
F
Federica Maschietto
A
Anton Morgunov
V
Víctor S. Batista *
DOI:10.1093/pnasnexus/pgae168delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
We introduce the kernel-elastic autoencoder (KAE), a self-supervised generative model based on the transformer architecture with enhanced performance for molecular design. KAE employs two innovative loss functions: modified maximum mean discrepancy (m-MMD) and weighted reconstruction ( L WCEL ). The m-MMD loss has significantly improved the generative performance of KAE when compared to using the traditional Kullback-Leibler loss of VAE, or standard maximum mean discrepancy. Including the weighted reconstruction loss L WCEL , KAE achieves valid generation and accurate reconstruction at the same time, allowing for generative behavior that is intermediate between VAE and autoencoder not available in existing generative approaches. Further advancements in KAE include its integration with conditional generation, setting a new state-of-the-art benchmark in constrained optimizations. Moreover, KAE has demonstrated its capability to generate molecules with favorable binding affinities in docking applications, as evidenced by AutoDock Vina and Glide scores, outperforming all existing candidates from the training dataset. Beyond molecular design, KAE holds promise to solve problems by generation across a broad spectrum of applications.
Keywords:
generative modeling
molecular optimization
molecular docking

Journal

P
PNAS Nexus
IF:
3.8
Papers:
2.1K
Citations:
3.2K

Organization

Y
Yale University
Scholars:
6.5W
Papers: 6.0W
Citations: 10.0W