Tensor Channel Equivariant Graph Neural Networks for Molecular Polarizability Prediction

• URL: http://arxiv.org/abs/2605.16891v1
• Date: Sat, 16 May 2026 09:07:07 GMT
• Title: Tensor Channel Equivariant Graph Neural Networks for Molecular Polarizability Prediction
• Abstract summary: We introduce a tensor-channel equivariant graph neural network for direct prediction of molecular polarizability tensors.<n>Building on the efficient PaiNN architecture, we augment the hidden representation with explicit symmetric rank-2 tensor channels.<n>Our model propagates tensor structure throughout message passing using geometrically motivated tensor bases.
• Score: 4.089404117494378
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: We introduce a tensor-channel equivariant graph neural network for direct prediction of molecular polarizability tensors. Building on the efficient PaiNN architecture, we augment the hidden representation with explicit symmetric rank-2 tensor channels aligned with the decomposition of polarizability into isotropic and anisotropic components. In contrast to approaches that construct tensor outputs only at readout, our model propagates tensor structure throughout message passing using geometrically motivated tensor bases. This yields a target-aligned architecture for tensor-valued molecular prediction. On optimized QM7-X geometries, the proposed model achieves lower full-tensor and anisotropic error than both a PaiNN-style readout baseline and a dielectric MACE baseline under matched training conditions and at nearly identical parameter count. In this controlled setting, it also outperforms MACE while remaining substantially faster at inference. Ablation studies show that the gain does not arise from increased capacity alone, but from the combination of explicit tensor propagation and a traceless target parameterization matched to the anisotropic part of the polarizability tensor. Among the tensor bases considered, the strongest results are obtained from interactions between learned directional features, indicating that these are particularly effective for modeling molecular polarizability. Rotational equivariance tests further confirm that all compared models are numerically equivariant, so the observed improvements are attributable to better learning of the target tensor itself. Overall, our results show that for structured tensor-valued targets, propagating target-aligned tensor features can outperform both readout-only tensor construction and a more general higher-order equivariant model in the present training setting.

Related papers

• Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generation [56.361076943802594]
  CanonFlow achieves state-of-the-art performance on the challenging GEOM-DRUG dataset, and the advantage remains large in few-step generation.
  arXiv  Detail & Related papers  (2026-02-16T18:58:55Z)
• Symmetry aware Reynolds Averaged Navier Stokes turbulence models with equivariant neural networks [0.6626421278252587]
  We introduce tensor-based, symmetry aware closures using equivariant neural networks (ENNs)<n>We present an algorithm for enforcing algebraic contraction relations among tensor components.
  arXiv  Detail & Related papers  (2025-11-12T22:04:05Z)
• Score-Based Model for Low-Rank Tensor Recovery [49.158601255093416]
  Low-rank tensor decompositions (TDs) provide an effective framework for multiway data analysis.<n>Traditional TD methods rely on predefined structural assumptions, such as CP or Tucker decompositions.<n>We propose a score-based model that eliminates the need for predefined structural or distributional assumptions.
  arXiv  Detail & Related papers  (2025-06-27T15:05:37Z)
• Rao-Blackwell Gradient Estimators for Equivariant Denoising Diffusion [55.95767828747407]
  In domains such as molecular and protein generation, physical systems exhibit inherent symmetries that are critical to model.<n>We present a framework that reduces training variance and provides a provably lower-variance gradient estimator.<n>We also present a practical implementation of this estimator incorporating the loss and sampling procedure through a method we call Orbit Diffusion.
  arXiv  Detail & Related papers  (2025-02-14T03:26:57Z)
• TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular Potentials [4.169915659794567]
  We introduceNet, an innovative O(3)-equivariant message-passing neural network architecture. By using tensor atomic embeddings, feature mixing is simplified through matrix product operations. The accurate prediction of vector and tensor molecular quantities on top of potential energies and forces is possible.
  arXiv  Detail & Related papers  (2023-06-10T16:41:18Z)
• A Nested Matrix-Tensor Model for Noisy Multi-view Clustering [5.132856740094742]
  We propose a nested matrix-tensor model which extends the spiked rank-one tensor model of order three. We show that our theoretical results allow us to anticipate the exact accuracy of the proposed clustering approach. Our analysis unveils unexpected and non-trivial phase transition phenomena depending on the model parameters.
  arXiv  Detail & Related papers  (2023-05-31T16:13:46Z)
• Optimizing Orthogonalized Tensor Deflation via Random Tensor Theory [5.124256074746721]
  This paper tackles the problem of recovering a low-rank signal tensor with possibly correlated components from a random noisy tensor. Non-orthogonal components may alter the tensor deflation mechanism, thereby preventing efficient recovery. An efficient tensor deflation algorithm is proposed by optimizing the parameter introduced in the deflation mechanism.
  arXiv  Detail & Related papers  (2023-02-11T22:23:27Z)
• Error Analysis of Tensor-Train Cross Approximation [88.83467216606778]
  We provide accuracy guarantees in terms of the entire tensor for both exact and noisy measurements. Results are verified by numerical experiments, and may have important implications for the usefulness of cross approximations for high-order tensors.
  arXiv  Detail & Related papers  (2022-07-09T19:33:59Z)
• Edge-based Tensor prediction via graph neural networks [6.021787236982659]
  Message-passing neural networks (MPNN) have shown extremely high efficiency and accuracy in predicting the physical properties of molecules and crystals. There is currently a lack of a general MPNN framework for directly predicting the tensor properties of the crystals. In this work, we directly designed the edge-based tensor prediction graph neural network (ETGNN) model on the basis of the invariant graph neural network to predict tensors.
  arXiv  Detail & Related papers  (2022-01-15T06:43:15Z)
• Lorentz Group Equivariant Neural Network for Particle Physics [58.56031187968692]
  We present a neural network architecture that is fully equivariant with respect to transformations under the Lorentz group. For classification tasks in particle physics, we demonstrate that such an equivariant architecture leads to drastically simpler models that have relatively few learnable parameters.
  arXiv  Detail & Related papers  (2020-06-08T17:54:43Z)
• Multi-View Spectral Clustering Tailored Tensor Low-Rank Representation [105.33409035876691]
  This paper explores the problem of multi-view spectral clustering (MVSC) based on tensor low-rank modeling. We design a novel structured tensor low-rank norm tailored to MVSC. We show that the proposed method outperforms state-of-the-art methods to a significant extent.
  arXiv  Detail & Related papers  (2020-04-30T11:52:12Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.