Variational Quantum Attention for Molecular Graph Learning

2026-10-03

AI Summary

This study addresses the limitations of classical attention mechanisms in molecular graph learning by investigating how variational quantum circuits can reshape attention behavior. We propose an edge-aware variational quantum attention mechanism that introduces a novel parameterization scheme encoding central atoms, neighboring atoms, and chemical bonds into quantum states, thereby enabling synergistic modeling of atomic and bond features. This mechanism reveals distinct attribution patterns between quantum and classical attention. Evaluated across five molecular property prediction tasks, our approach achieves performance comparable to GATv2, with particularly significant improvements on the BBBP dataset. Furthermore, a BACE1 case study validates its advantages in maintaining structure–activity relationship consistency.

Abstract

Molecular property prediction is central to computational drug discovery, where graph neural networks learn to weight neighboring atomic environments during message passing. Yet it remains unclear how variational quantum circuits alter learned attention behavior in molecular graphs. We introduce an edge-aware variational quantum attention mechanism for molecular graph learning, in which the receiving atom, neighboring atom, and connecting bond jointly determine the quantum attention state. Across five molecular property and bioactivity prediction tasks, QGAT achieves competitive performance relative to GATv2, with a consistent improvement on BBBP across all six evaluated circuit ansatzes. We further compare how the quantum and classical attention scores weight molecular structure beyond accuracy. In the Verubecestat BACE1 inhibitor series, QGAT achieves a higher Spearman correlation than GATv2 and assigns positive attributions to several structural changes consistent with reported structure-activity relationships (SARs). This case study shows that the two attention mechanisms can exhibit different prediction and attribution behavior across structurally related BACE1 analogues, while broader validation is required to determine how consistently these differences generalize across chemical series and targets. Circuit ablations further show that performance depends on the circuit design. Together, these results show that variational quantum attention can serve as a viable alternative molecular attention parameterization while inducing circuit- and chemistry-dependent behavior distinct from a matched classical scorer.

Problem

• molecular property prediction
• variational quantum circuits
• graph attention
• drug discovery
• structure-activity relationships

Innovation

• Variational Quantum Attention
• Molecular Graph Learning
• Edge-aware Mechanism
• Quantum Circuit Ansatz
• Structure-Activity Relationships