Search papers, labs, and topics across Lattice.
This study investigates the impact of explicitly encoding a molecule's Bemis-Murcko scaffold on self-supervised molecular embedding learning, comparing the effectiveness of Euclidean and Lorentz contrastive objectives. The findings reveal that scaffold-supervised models consistently organize molecular embeddings based on structural similarities, enhancing property prediction across various tasks, although the degree of improvement varies by property. Notably, the Lorentz geometry yields stronger scaffold supervision effects, indicating that structural core relationships significantly influence embedding organization, albeit with task-dependent utility.
Explicitly teaching models about molecular scaffolds can reshape how embeddings are organized, leading to improved property predictions in self-supervised learning.
Molecular self-supervised learning uses chemical structures to guide which molecular embeddings should be similar. We study whether explicitly encoding a molecule's Bemis-Murcko scaffold and using it to supervise the molecular embedding changes what the model learns. We further test whether this effect depends on the embedding geometry by comparing Euclidean and Lorentz contrastive objectives. Across two augmentation strengths, scaffold-supervised models consistently organize molecules according to both identical and structurally related scaffolds. The resulting embeddings also improve molecular property prediction on several tasks, while the exact gains depend on the predicted property. The effect of scaffold supervision on molecular organization is stronger under Lorentz objectives, but neither geometry provides a consistent overall advantage. These results show that explicitly teaching the relation between a molecule and its structural core can reliably shape the organization of molecular embedding space, while the extent of usefulness of this organization remains task dependent.