Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces Bipartite Cholesky Graph Networks (BCGNs) for predicting molecular correlation energies, addressing the computational bottleneck of handling the electron repulsion integral (ERI) tensor. By leveraging the density-fitted Cholesky decomposition of the ERI, BCGNs construct a bipartite graph that explicitly models orbital and auxiliary interaction nodes, preserving higher-order interaction structures. Experiments on diatomic molecules demonstrate that BCGNs achieve significantly lower mean absolute error compared to methods that compress the ERI tensor into low-rank scalar features, while also revealing insights into zero-shot generalization performance based on orbital environment similarity.
Ditch the low-rank compression: a bipartite graph network built on Cholesky decomposition unlocks more accurate quantum chemistry predictions by preserving higher-order electron interaction structures.
Accurate prediction of molecular correlation energies from first principles requires resolving the {O}(N^4) electron repulsion integral (ERI) tensor. Existing graph neural network approaches to the electronic structure problem often compress this tensor into low-rank scalar features, discarding higher-order interaction structures relevant to electron correlation. In this work, we demonstrate that tensor factorization of the ERI naturally induces a structured bipartite message-passing architecture that preserves access to higher-order interaction structure more effectively than compressed orbital representations. By utilizing the density-fitted Cholesky decomposition of the ERI tensor, we derive a bipartite graph network that models orbital degrees of freedom and auxiliary interaction nodes as distinct sets, maintaining interaction topology at a reduced theoretical complexity of {O}(N^3). Evaluated on 132 geometries of six diatomic molecules with Full Configuration Interaction (FCI) reference energies, our factorized representation achieves an in-distribution Mean Absolute Error (MAE) of 0.0296 Ha under five-fold cross-validation, a substantial improvement over compressed-integral baselines. Leave-one-molecule-out validation reveals that zero-shot generalization varies by nearly a factor of four across molecular species and correlates with the structural similarity of the held-out molecule's orbital environment to the training distribution, rather than with nuclear charge asymmetry alone.