Attention Mechanisms in Physics-Inspired Graph Neural Networks for the Max-Cut Problem

Main Article Content

Temuujin Delgertsogt
Dalaijargal Purevsuren
Gantulga Gombojav

Keywords

Physics-Inspired Graph Neural Networks, Combinatorial Optimization, Graph Attention Networks, oversmoothing

Abstract

Physics-Inspired Graph Neural Networks (PI-GNNs) reformulate MAX-CUT as QUBO energy minimization, training a GNN to produce soft binary node assignments without labeled data. The baseline PI-GCN uses static, degree-normalized aggregation, while its attention-augmented counterpart PI-GAT---built on GATv2---introduces additional hyperparameters whose effects remain uncharacterized. This paper addresses that gap through controlled experiments on five Gset benchmark graphs (800--10,000 nodes). The first experiment examines how attention dropout p_attn affects solution quality and convergence stability across 25 seeds per configuration. The second sweeps depth from 1 to 20 layers, quantifying oversmoothing via Dirichlet energy, cosine similarity, and mean absolute deviation. On dense graphs (average degree 10), dropout in [0.10, 0.20] reduces inter-run variance by 30--41% while improving best-cut value; on the sparse graph G70 (average degree 2, diameter 34), any dropout is detrimental. PI-GAT delays oversmoothing onset by one to two layers versus PI-GCN, at 1.5-1.8x computational overhead per layer. These findings establish graph density and diameter as the primary structural determinants of whether dynamic attention benefits the PI-GNN framework.

Abstract 4 | FULL PDF Downloads 1

References

[1] I. Alkhouri, M. Wu, C. Yu, J. Liu, R. Wang and A. Velasquez, “A scalable lift-and-project differentiable approach for the maximum cut problem,” arXiv preprint arXiv:2509.18612, 2025. [Online]. Available: https://arxiv.org/abs/2509.18612
[2] P. Almasan, J. Suárez-Varela, K. Rusek, P. Barlet-Ros and A. Cabellos-Aparicio, “Deep reinforcement learning meets graph neural networks: Exploring a routing optimization use case,” Computer Communications, vol. 196, pp. 184–194, 2022. [Online]. Available: https://doi.org/10.1016/j.comcom.2022.09.029
[3] F. Barahona, M. Grötschel, M. Jünger and G. Reinelt, “An application of combinatorial optimization to statistical physics and circuit layout design,” Operations Research, vol. 36, no. 3, pp. 493–513, 1988. [Online]. Available: https://doi.org/10.1287/opre.36.3.493
[4] I. Bello, H. Pham, Q. V. Le, M. Norouzi and S. Bengio, “Neural combinatorial optimization with reinforcement learning,” arXiv preprint arXiv:1611.09940, 2016. [Online]. Available: https://arxiv.org/abs/1611.09940
[5] U. Benlic and J.-K. Hao, “Breakout local search for the max-cut problem,” Engineering Applications of Artificial Intelligence, vol. 26, no. 3, pp. 1162–1173, 2013. [Online]. Available: https://doi.org/10.1016/j.engappai.2012.09.001
[6] B. Bhattacharyya, M. Capriotti and R. Tate, “Solving general QUBOs with warm-start QAOA via a reduction to max-cut,” arXiv preprint arXiv:2504.06253, 2025. [Online]. Available: https://arxiv.org/abs/2504.06253
[7] S. Brody, U. Alon and E. Yahav, “How attentive are graph attention networks?” in Proc. Int. Conf. Learning Representations (ICLR), 2022. [Online]. Available: https://openreview.net/forum?id=F72ximsx7C1
[8] B. A. Cipra, “An introduction to the Ising model,” The American Mathematical Monthly, vol. 94, no. 10, pp. 937–959, 1987. [Online]. Available: https://doi.org/10.1080/00029890.1987.12000742
[9] J. Falla, Q. Langfitt, Y. Alexeev and I. Safro, “Graph representation learning for parameter transferability in quantum approximate optimization algorithm,” Quantum Machine Intelligence, vol. 6, no. 2, art. 46, 2024. [Online]. Available: https://doi.org/10.1007/s42484-024-00178-9
[10] P. Festa, P. M. Pardalos, M. G. C. Resende and C. C. Ribeiro, “Randomized heuristics for the max-cut problem,” Optimization Methods and Software, vol. 17, no. 6, pp. 1033–1058, 2002. [Online]. Available: https://doi.org/10.1080/1055678021000090033
[11] M. X. Goemans and D. P. Williamson, “Improved approximation algorithms for maximum cut and satisfiability problems using semidefinite programming,” Journal of the ACM, vol. 42, no. 6, pp. 1115–1145, 1995. [Online]. Available: https://doi.org/10.1145/227683.227684
[12] C. Helmberg and F. Rendl, “A spectral bundle method for semidefinite programming,” SIAM Journal on Optimization, vol. 10, no. 3, pp. 673–696, 2000. [Online]. Available: https://doi.org/10.1137/S1052623497328987
[13] D. Karalias and A. Loukas, “Erdos goes neural: An unsupervised learning framework for combinatorial optimization on graphs,” in Advances in Neural Information Processing Systems (NeurIPS), vol. 33, 2020, pp. 6659–6672. [Online]. Available: https://arxiv.org/abs/2006.10643
[14] R. M. Karp, “Reducibility among combinatorial problems,” in Complexity of Computer Computations, R. E. Miller and J. W. Thatcher, Eds. New York: Plenum Press, 1972, pp. 85–103. [Online]. Available: https://doi.org/10.1007/978-1-4684-2001-2_9
[15] E. Khalil, H. Dai, Y. Zhang, B. Dilkina and L. Song, “Learning combinatorial optimization algorithms over graphs,” in Advances in Neural Information Processing Systems (NeurIPS), vol. 30, 2017, pp. 6348–6358. [Online]. Available: https://arxiv.org/abs/1704.01665
[16] S. Khot, G. Kindler, E. Mossel and R. O’Donnell, “Optimal inapproximability results for Max-Cut and other 2-variable CSPs?” SIAM Journal on Computing, vol. 37, no. 1, pp. 319–357, 2007. [Online]. Available: https://doi.org/10.1137/S0097539705447372
[17] T. N. Kipf and M. Welling, “Semi-supervised classification with graph convolutional networks,” in Proc. Int. Conf. Learning Representations (ICLR), 2017. [Online]. Available: https://arxiv.org/abs/1609.02907
[18] J. Li, Q. Zhang, W. Liu, A. B. Chan and Y.-G. Fu, “Another perspective of over-smoothing: Alleviating semantic over-smoothing in deep GNNs,” IEEE Transactions on Neural Networks and Learning Systems, vol. 36, no. 4, pp. 6897–6910, 2025. [Online]. Available: https://doi.org/10.1109/TNNLS.2024.3402317
[19] Q. Li, Z. Han and X.-M. Wu, “Deeper insights into graph convolutional networks for semi-supervised classification,” in Proc. AAAI Conf. Artificial Intelligence, vol. 32, 2018, pp. 3538–3545. [Online]. Available: https://doi.org/10.1609/aaai.v32i1.11604
[20] G. Maliakal, I. Alkhouri, A. Velasquez, A. M. Alessio and S. Ravishankar, “A dataless reinforcement learning approach to rounding hyperplane optimization for max-cut,” arXiv preprint arXiv:2505.13405, 2025. [Online]. Available: https://arxiv.org/abs/2505.13405
[21] G. Palubeckis, “Multistart tabu search strategies for the unconstrained binary quadratic optimization problem,” Annals of Operations Research, vol. 131, no. 1–4, pp. 259–282, 2004. [Online]. Available: https://doi.org/10.1023/B:ANOR.0000039522.58036.68
[22] D. Pugacheva, A. Ermakov, I. Lyskov, I. Makarov and Y. Zotov, “Enhancing GNNs performance on combinatorial optimization by recurrent feature update,” arXiv preprint arXiv:2407.16468, 2024. [Online]. Available: https://arxiv.org/abs/2407.16468
[23] Y. Qiu, Y. Xue, A. Wang, Y. Wang, Q. Shi and Z.-Q. Luo, “ROS: A GNN-based relax-optimize-and-sample framework for max-k-cut problems,” arXiv preprint arXiv:2412.05146, 2024. [Online]. Available: https://arxiv.org/abs/2412.05146
[24] Y. Rong, W. Huang, T. Xu and J. Huang, “DropEdge: Towards deep graph convolutional networks on node classification,” in Proc. Int. Conf. Learning Representations (ICLR), 2020. [Online]. Available: https://openreview.net/forum?id=Hkx1qkrKPr
[25] R. A. Rossi and N. K. Ahmed, “The network data repository with interactive graph analytics and visualization,” in Proc. AAAI Conf. Artificial Intelligence, 2015, pp. 4292–4293. [Online]. Available: https://networkrepository.com
[26] T. K. Rusch, M. M. Bronstein and S. Mishra, “A survey on oversmoothing in graph neural networks,” arXiv preprint arXiv:2303.10993, 2023. [Online]. Available: https://arxiv.org/abs/2303.10993
[27] M. J. A. Schuetz, J. K. Brubaker and H. G. Katzgraber, “Combinatorial optimization with physics-inspired graph neural networks,” Nature Machine Intelligence, vol. 4, no. 4, pp. 367–377, 2022. [Online]. Available: https://doi.org/10.1038/s42256-022-00468-6
[28] M. J. A. Schuetz, J. K. Brubaker, Z. Zhu and H. G. Katzgraber, “Graph coloring with physics-inspired graph neural networks,” Physical Review Research, vol. 4, no. 4, art. 043131, 2022. [Online]. Available: https://doi.org/10.1103/PhysRevResearch.4.043131
[29] A. Vaswani, N. Shazeer, N. Parmar, J. Uszkoreit, L. Jones, A. N. Gomez, Ł. Kaiser and I. Polosukhin, “Attention is all you need,” in Advances in Neural Information Processing Systems, vol. 30, 2017. [Online]. Available: https://arxiv.org/abs/1706.03762
[30] P. Veličković, G. Cucurull, A. Casanova, A. Romero, P. Liò and Y. Bengio, “Graph attention networks,” in Proc. Int. Conf. Learning Representations (ICLR), 2018. [Online]. Available: https://arxiv.org/abs/1710.10903
[31] Y. Wang and X. Liang, “Application of reinforcement learning methods combining graph neural networks and self-attention mechanisms in supply chain route optimization,” Sensors, vol. 25, no. 3, art. 955, 2025. [Online]. Available: https://doi.org/10.3390/s25030955
[32] Y. Ye, “The Gset dataset,” 2003. [Online]. Available:https://web.stanford.edu/~yyye/yyye/Gset/

Similar Articles

You may also start an advanced similarity search for this article.