Complex network robustness prediction using attention-augmented CNN

Assessing the strength of complex networks is crucial for evaluating their functionality and monitoring system security. Two primary perspectives for measuring network robustness are connectivity and controllability, which quantify how well a network maintains connectivity and controllability after experiencing complete attacks. However, conventional robustness calculation methods that use simulation attacks can be impractical for large-scale networks. Recent research has explored the use of deep learning methods for network robustness prediction, including PCR (short for predictor of controllability robustness), which uses a VGG (short for visual geometry group)-based convolutional neural network (CNN) and a linear filter. However, PCR has limitations, including inaccurate edge importance differentiation and fluctuating output curves. This paper presents ATTRP, an attention-based robustness predictor, which addresses these limitations. ATTRP uses a novel convolutional neural network with channel and spatial attention mechanisms to improve prediction accuracy and a robustness filter based on Savitzky–Golay smoothing to rectify the robustness curve. Experimental studies demonstrate that ATTRP outperforms other predictors, with much lower errors, while its computational speed is far superior to that of traditional attacking simulation. Overall, our research provides a more effective and efficient means of predicting network robustness, with potential applications in monitoring system security.