Port-Hamiltonian Architectural Bias for Long-Range Propagation in Deep Graph Networks
About
The dynamics of information diffusion within graphs is a critical open issue that heavily influences graph representation learning, especially when considering long-range propagation. This calls for principled approaches that control and regulate the degree of propagation and dissipation of information throughout the neural flow. Motivated by this, we introduce (port-)Hamiltonian Deep Graph Networks, a novel framework that models neural information flow in graphs by building on the laws of conservation of Hamiltonian dynamical systems. We reconcile under a single theoretical and practical framework both non-dissipative long-range propagation and non-conservative behaviors, introducing tools from mechanical systems to gauge the equilibrium between the two components. Our approach can be applied to general message-passing architectures, and it provides theoretical guarantees on information conservation in time. Empirical results prove the effectiveness of our port-Hamiltonian scheme in pushing simple graph convolutional architectures to state-of-the-art performance in long-range benchmarks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Diameter prediction | ECHO-Synth | MAE1.627 | 23 | |
| Node Eccentricity Prediction | ECHO-Synth | MAE5.068 | 23 | |
| Single-Source Shortest Path Prediction | ECHO-Synth | MAE1.323 | 23 | |
| Atomic partial charge prediction | ECHO-Charge | MSE2.562 | 11 | |
| Total molecular energy prediction | ECHO-Energy | MSE1.359 | 11 |