Multiscale Hypergraph Masked Autoencoder with Δ-Property Alignment for Novel Molecular Representation Learning

Molecular representation learning often focuses on recognizing local structural patterns but struggles to capture functional groups and cross-scale interactions in a chemically meaningful way. Here, we propose the Multiscale Hypergraph Convolutional Masked Autoencoder (MSHG-MAE), a hypergraph-based self-supervised framework for drug-like molecular representation learning. We model each molecule as a unified molecular hypergraph with atom nodes and multitype hyperedges, including bond, ring, functional group, conjugated system, and hydrogen bond hyperedges, and use multiscale hypergraph convolutions to jointly capture dependencies at the atomic, substructural, and molecular levels. In pretraining, a semantics-aware masked autoencoding objective applies biased masking to atoms and hyperedges and reconstructs masked node features and hyperedge attributes, encouraging the model to internalize functional-group semantics. Furthermore, we introduce Δ-Property Alignment (Δ-PropAlign), which aligns embedding differences with proxy property differences so that the learned representations remain sensitive to interpretable structure-property changes. Experiments on multiple public molecular property benchmarks under scaffold splits show that MSHG-MAE consistently outperforms baselines on regression tasks and achieves competitive results on classification benchmarks. On three physicochemical regression benchmarks under scaffold splits (ESOL, FreeSolv, and Lipophilicity), MSHG-MAE with Δ-PropAlign achieves RMSEs of 0.465, 0.780, and 0.501, respectively, corresponding to approximately 40% and 47% lower RMSE than Uni-Mol on ESOL and FreeSolv and approximately 27% lower RMSE than D-MPNN on Lipophilicity (Table 5, scaffold split 80/10/10, 3 seeds), while not degrading structural or functional-group geometry. These results indicate that Δ-PropAlign improves the consistency between embedding differences and property differences without sacrificing structural or functional-group organization in the learned representations. The core code is publicly available on GitHub (https://github.com/Irzos/MSHG-MAE), and the data sets used in this study are publicly available from ZINC20 and MoleculeNet.

Authors

Institutions

Publication Details

Journal
Journal of Chemical Information and Modeling
Published
2026-03-16
DOI
https://doi.org/10.1021/acs.jcim.5c02994
Primary Topic
Machine Learning in Materials Science
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Multiscale Hypergraph Masked Autoencoder with Δ-Property Alignment for Novel Molecular Representation Learning

Yang Wang, Ziyan Zhu, Yanmin Zhang, Xian Wei
Journal of Chemical Information and Modeling
Machine Learning in Materials Science
article

Multiscale Hypergraph Masked Autoencoder with Δ-Property Alignment for Novel Molecular Representation Learning

Yang Wang, Ziyan Zhu, Yanmin Zhang, Xian Wei
article en

Abstract

Molecular representation learning often focuses on recognizing local structural patterns but struggles to capture functional groups and cross-scale interactions in a chemically meaningful way. Here, we propose the Multiscale Hypergraph Convolutional Masked Autoencoder (MSHG-MAE), a hypergraph-based self-supervised framework for drug-like molecular representation learning. We model each molecule as a unified molecular hypergraph with atom nodes and multitype hyperedges, including bond, ring, functional group, conjugated system, and hydrogen bond hyperedges, and use multiscale hypergraph convolutions to jointly capture dependencies at the atomic, substructural, and molecular levels. In pretraining, a semantics-aware masked autoencoding objective applies biased masking to atoms and hyperedges and reconstructs masked node features and hyperedge attributes, encouraging the model to internalize functional-group semantics. Furthermore, we introduce Δ-Property Alignment (Δ-PropAlign), which aligns embedding differences with proxy property differences so that the learned representations remain sensitive to interpretable structure-property changes. Experiments on multiple public molecular property benchmarks under scaffold splits show that MSHG-MAE consistently outperforms baselines on regression tasks and achieves competitive results on classification benchmarks. On three physicochemical regression benchmarks under scaffold splits (ESOL, FreeSolv, and Lipophilicity), MSHG-MAE with Δ-PropAlign achieves RMSEs of 0.465, 0.780, and 0.501, respectively, corresponding to approximately 40% and 47% lower RMSE than Uni-Mol on ESOL and FreeSolv and approximately 27% lower RMSE than D-MPNN on Lipophilicity (Table 5, scaffold split 80/10/10, 3 seeds), while not degrading structural or functional-group geometry. These results indicate that Δ-PropAlign improves the consistency between embedding differences and property differences without sacrificing structural or functional-group organization in the learned representations. The core code is publicly available on GitHub (https://github.com/Irzos/MSHG-MAE), and the data sets used in this study are publicly available from ZINC20 and MoleculeNet.

Journal of Chemical Information and Modeling
China Pharmaceutical University (CN), Affiliated Hospital of Youjiang Medical University for Nationalities (CN)
Openalex Percentile: Top 21%
Machine Learning in Materials Science
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.