English
Related papers

Related papers: Pretraining Graph Transformers with Atom-in-a-Mole…

200 papers

Chemical Language Models (CLMs) pre-trained on large scale molecular data are widely used for molecular property prediction. However, the common belief that increasing training resources such as model size, dataset size, and training…

Machine Learning · Computer Science 2026-05-14 Tatsuya Sagawa , Ryosuke Kojima

Molecular dynamics (MD) simulations underpin modern computational drug discovery, materials science, and biochemistry. Recent machine learning models provide high-fidelity MD predictions without the need to repeatedly solve quantum…

Machine Learning · Computer Science 2026-04-24 Luke Thompson , Davy Guan , Dai Shi , Slade Matthews , Junbin Gao , Andi Han

Recent work on Graph Neural Networks has demonstrated that self-supervised pretraining can further enhance performance on downstream graph, link, and node classification tasks. However, the efficacy of pretraining tasks has not been fully…

Machine Learning · Computer Science 2023-03-28 Jonathan Pilault , Michael Galkin , Bahare Fatemi , Perouz Taslakian , David Vasquez , Christopher Pal

Molecular representation is a critical element in our understanding of the physical world and the foundation for modern molecular machine learning. Previous molecular machine learning models have employed strings, fingerprints, global…

Machine Learning · Computer Science 2025-05-28 Daniil A. Boiko , Thiago Reschützegger , Benjamin Sanchez-Lengeling , Samuel M. Blau , Gabe Gomes

When training transformers on graph-structured data, incorporating information about the underlying topology is crucial for good performance. Topological masking, a type of relative position encoding, achieves this by upweighting or…

There is a perceived dichotomy between structure-based and descriptor-based molecular representations used for predictive chemistry tasks. Here, we study the performance, generalizability, and interpretability of the recently proposed…

Chemical Physics · Physics 2022-03-14 Thijs Stuyver , Connor W. Coley

Autoencoders are effective deep learning models that can function as generative models and learn latent representations for downstream tasks. The use of graph autoencoders - with both encoder and decoder implemented as message passing…

Machine Learning · Computer Science 2025-03-04 Magnus Cunow , Gerrit Großmann

In the technical report, we provide our solution for OGB-LSC 2022 Graph Regression Task. The target of this task is to predict the quantum chemical property, HOMO-LUMO gap for a given molecule on PCQM4Mv2 dataset. In the competition, we…

Machine Learning · Computer Science 2023-08-17 Yusong Wang , Shaoning Li , Zun Wang , Xinheng He , Bin Shao , Tie-Yan Liu , Tong Wang

Understanding molecular structure and related knowledge is crucial for scientific research. Recent studies integrate molecular graphs with their textual descriptions to enhance molecular representation learning. However, they focus on the…

Artificial Intelligence · Computer Science 2025-06-02 Yibo Li , Yuan Fang , Mengmei Zhang , Chuan Shi

Designing a single neural network architecture that performs competitively across a range of molecule property prediction tasks remains largely an open challenge, and its solution may unlock a widespread use of deep learning in the drug…

Machine Learning · Computer Science 2021-02-10 Łukasz Maziarka , Tomasz Danel , Sławomir Mucha , Krzysztof Rataj , Jacek Tabor , Stanisław Jastrzębski

Pre-trained model representations have demonstrated state-of-the-art performance in speech recognition, natural language processing, and other applications. Speech models, such as Bidirectional Encoder Representations from Transformers…

Audio and Speech Processing · Electrical Eng. & Systems 2023-03-07 Vikramjit Mitra , Vasudha Kowtha , Hsiang-Yun Sherry Chien , Erdrin Azemi , Carlos Avendano

According to density functional theory, any chemical property can be inferred from the electron density, making it the most informative attribute of an atomic structure. In this work, we demonstrate the use of established physical methods…

Materials Science · Physics 2023-09-12 Ethan M. Sunshine , Muhammed Shuaibi , Zachary W. Ulissi , John R. Kitchin

The advent of the Transformer architecture has propelled the growth of natural language processing (NLP) models, leading to remarkable achievements in numerous NLP tasks. Yet, the absence of specialized hardware like expansive GPU memory…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-03-18 Xiaofeng Wu , Jia Rao , Wei Chen

Molecular pretrained representations (MPR) has emerged as a powerful approach for addressing the challenge of limited supervised data in applications such as drug discovery and material design. While early MPR methods relied on 1D sequences…

Biomolecules · Quantitative Biology 2025-03-19 Shuqi Lu , Xiaohong Ji , Bohang Zhang , Lin Yao , Siyuan Liu , Zhifeng Gao , Linfeng Zhang , Guolin Ke

Transformer-based deep learning methods have become the standard approach for modeling diverse data such as sequences, images, and graphs. These methods rely on self-attention, which treats data as an unordered set of elements. This ignores…

Machine Learning · Computer Science 2025-10-15 Aakash Lahoti , Tanya Marwah , Ratish Puduppully , Albert Gu

Learning effective protein representations is critical in a variety of tasks in biology such as predicting protein function or structure. Existing approaches usually pretrain protein language models on a large number of unlabeled amino acid…

Machine Learning · Computer Science 2023-01-31 Zuobai Zhang , Minghao Xu , Arian Jamasb , Vijil Chenthamarakshan , Aurelie Lozano , Payel Das , Jian Tang

Pre-trained language models have recently emerged as a powerful tool for fine-tuning a variety of language tasks. Ideally, when models are pre-trained on large amount of data, they are expected to gain implicit knowledge. In this paper, we…

Computation and Language · Computer Science 2023-06-22 Mohamad Ballout , Ulf Krumnack , Gunther Heidemann , Kai-Uwe Kühnberger

In the computational prediction of chemical compound properties, molecular descriptors and fingerprints encoded to low dimensional vectors are used. The selection of proper molecular descriptors and fingerprints is both important and…

Machine Learning · Computer Science 2020-10-23 Sangrak Lim , Yong Oh Lee

The success of deep learning heavily relies on large-scale data with comprehensive labels, which is more expensive and time-consuming to fetch in 3D compared to 2D images or natural languages. This promotes the potential of utilizing models…

Computer Vision and Pattern Recognition · Computer Science 2023-02-03 Runpei Dong , Zekun Qi , Linfeng Zhang , Junbo Zhang , Jianjian Sun , Zheng Ge , Li Yi , Kaisheng Ma

Retrosynthesis prediction is a fundamental problem in organic synthesis, where the task is to identify precursor molecules that can be used to synthesize a target molecule. A key consideration in building neural models for this task is…

Machine Learning · Computer Science 2021-06-07 Vignesh Ram Somnath , Charlotte Bunne , Connor W. Coley , Andreas Krause , Regina Barzilay