Weighted graph convolutional network with feature mask for low back pain prediction

Zhixin Li,Shiyi Shen,Fanqi Shang,Zhan Huan,Jiuzhen Liang,Ying Chen
DOI: https://doi.org/10.1016/j.bspc.2024.106879
IF: 5.1
2024-09-15
Biomedical Signal Processing and Control
Abstract:Low back pain (LBP) is one of the most common physical disabilities worldwide, imposing a heavy medical economic burden. To help medical experts make disease predictions at an early stage, this paper proposes an LBP abnormality diagnosis model based on a weighted graph convolutional network with feature mask. Unlike popular neural network architectures, we represent the biomechanical characteristics of the spine as a weighted feature graph. The potential relationship between patients is mined through edge connections, and the importance of features for different patients is dynamically captured through the internal feature weighting of nodes. Finally, the importance score of each feature is calculated according to its identification and independence to generate the feature importance mask. It realizes feature sparsification and reinforces important features at the same time, so as to improve the accuracy and stability of the model. The proposed model is verified on a public dataset available in the Kaggle repository, and the classification accuracy achieves 90.2%, which is 2.14% higher than the current best result. Furthermore, when compared for stability against several commonly used machine learning algorithms, the standard deviation of accuracy over ten experiments is only about 1/3 of that of other algorithms.
engineering, biomedical
What problem does this paper attempt to address?