Journal of Jilin University(Engineering and Technology Edition) ›› 2025, Vol. 55 ›› Issue (2): 681-686.doi: 10.13229/j.cnki.jdxbgxb.20230384

Previous Articles    

A weighted isomorphic graph classification algorithm based on causal feature learning

Xiang-jiu CHE(),Yu-ning WU,Quan-le LIU   

  1. College of Computer Science and Technology,Jilin University,Changchun 130012,China
  • Received:2023-04-19 Online:2025-02-01 Published:2025-04-16

Abstract:

Aiming at the problem that the existing neural network methods are not accurate enough to predict the Killip classification of patients with myocardial infarction, a weighted isomorphic graph classification algorithm that can learn causal features is proposed. Use the differentiated learning objectives to separate the graph-level representation features of causal correlation and non-causal correlation, and then use the backdoor adjustment method in causal reasoning to reduce the confusion of non-causal features on classification results. The experimental results show that the average accuracy of the method proposed in this paper reaches 80.52% in the prediction task of the Killip grading of patients with myocardial infarction, which is 3.72% higher than that of the non-graph neural network method, and 0.72% higher than that of the graph neural network method without learning causal characteristics. Therefore, the method proposed can better complete the task of predicting the Killip grading of patients with myocardial infarction.

Key words: computer application technology, graph neural network, causal analysis, Killip classification

CLC Number: 

  • TP391.4

Fig. 1

Overview of CGCN framework"

Fig. 2

Structural causal graph for graph representation learning"

Table 1

Experimental environment configuration"

运行环境CPUGPU显存CUDA 版本深度学习框架语言
Windows 11i5-12500hGeForce RTX 3060 Laptop8 GByte11.0PyTorchPython

Table 2

Experiment results"

模型准确率/%召回率/%
MLP76.8029.19
GCN79.8027.88
CGCN80.5232.83

Fig. 3

Hyperparametric experimental results"

1 Kipf T N, Welling M. Semi-supervised classification with graph convolutional networks[J/OL].[2023-03-05]. arXiv preprint arXiv:.
2 Wu Y X, Wang X, Zhang A, et al. Discovering invariant rationales for graph neural networks[J/OL]. [2023-03-06].arXiv preprint arXiv:2201.12872v1 .
3 Sui Y, Wang X, Wu J, et al. Causal attention for interpretable and generalizable graph classification[C]∥ Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, Washington, USA, 2022: 1696-1705.
4 Fan S, Wang X, Mo Y, et al. Debiasing graph neural networks via learning disentangled causal substructure[C]∥Advances in Neural Information Processing Systems, New York, USA, 2022: 3602078.
5 徐冰冰, 岑科廷, 黄俊杰, 等. 图卷积神经网络综述[J]. 计算机学报, 2020, 43(5): 755-780.
Xu Bing-bing, Cen Ke-ting, Huang Jun-jie, et al. A survey on graph convolutional neural network[J]. Chinese Journal of Computers, 2020, 43(5): 755-780.
6 王兆慧, 沈华伟, 曹婍, 等. 图分类研究综述[J]. 软件学报, 2022, 33(1): 171-192.
Wang Zhao-hui, Shen Hua-wei, Cao Qi, et al. Survey on graph classification[J]. Journal of Software, 2022, 33(1): 171-192.
7 Geirhos R, Jacobsen J H, Michaelis C, et al. Shortcut learning in deep neural networks[J]. Nature Machine Intelligence, 2020, 2(11): 665-673.
8 Geirhos R, Rubisch P, Michaelis C, et al. Imagenet-trained CNNs are biased towards texture; increasing shape bias improves accuracy and robustness[J/OL].[2023-03-07]. arXiv preprint arXiv:.
9 Arjovsky M, Bottou L, Gulrajani I, et al. Invariant risk minimization[J/OL].[2023-03-08] .arXiv preprint, arXiv:.
10 Knyazev B, Taylor G W, Amer M. Understanding attention and generalization in graph neural networks[C]∥Advances in Neural Information Processing Systems, Vancouver, Canada, 2019: 190502850.
11 Pearl J, Mackenzie D. The Book of Why: The New Science of Cause and Effect[M]. New York: Basic books, 2018.
12 Jia Z, Lin Y, Wang J, et al. Hetemotionnet: two-stream heterogeneous graph recurrent neural network for multi-modal emotion recognition[C]∥Proceedings of the 29th ACM International Conference on Multimedia, Chengdu, China, 2021: 1047-1056.
[1] Xi-jun ZHANG,Guang-jie YU,Yong CUI,Ji-yang SHANG. Short-term traffic flow prediction based on clustering algorithm and graph neural network [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(6): 1593-1600.
[2] Li-ming LIANG,Long-song ZHOU,Jiang YIN,Xiao-qi SHENG. Fusion multi-scale Transformer skin lesion segmentation algorithm [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(4): 1086-1098.
[3] Dondrub LHAKPA,Duoji ZHAXI,Jie ZHU. Tibetan text normalization method [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(12): 3577-3588.
[4] Na CHE,Yi-ming ZHU,Jian ZHAO,Lei SUN,Li-juan SHI,Xian-wei ZENG. Connectionism based audio-visual speech recognition method [J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(10): 2984-2993.
[5] Ya-hui ZHAO,Fei-yu LI,Rong-yi CUI,Guo-zhe JIN,Zhen-guo ZHANG,De LI,Xiao-feng JIN. Korean⁃Chinese translation quality estimation based on cross⁃lingual pretraining model [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(8): 2371-2379.
[6] Shan XUE,Ya-liang ZHANG,Qiong-ying LYU,Guo-hua CAO. Anti⁃unmanned aerial vehicle system object detection algorithm under complex background [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(3): 891-901.
[7] Zhen WANG,Xiao-han YANG,Nan-nan WU,Guo-kun LI,Chuang FENG. Ordinal cross entropy Hashing based on generative adversarial network [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(12): 3536-3546.
[8] Feng-feng ZHOU,Zhen-wei YAN. A model for identifying neuropeptides by feature selection based on hybrid features [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(11): 3238-3245.
[9] Li-yan DONG,Wei-ye LIANG,Yue-qun WANG,Yong-li LI. Global potential information combined graph neural networks for session-based recommendation [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(10): 2964-2972.
[10] Jun-jie WANG,Yuan-jun NONG,Li-te ZHANG,Pei-chen ZHAI. Visual relationship detection method based on construction scene [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(1): 226-233.
[11] Bing ZHU,Zi-wei LI,Qi LI. Building segmentation method of remote sensing image based on improved SegNet [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(1): 248-254.
[12] Gui-he QIN,Jun-feng HUANG,Ming-hui SUN. Text input based on two⁃handed keyboard in virtual environment [J]. Journal of Jilin University(Engineering and Technology Edition), 2022, 52(8): 1881-1888.
[13] Fu-heng QU,Tian-yu DING,Yang LU,Yong YANG,Ya-ting HU. Fast image codeword search algorithm based on neighborhood similarity [J]. Journal of Jilin University(Engineering and Technology Edition), 2022, 52(8): 1865-1871.
[14] Tian BAI,Ming-wei XU,Si-ming LIU,Ji-an ZHANG,Zhe WANG. Dispute focus identification of pleading text based on deep neural network [J]. Journal of Jilin University(Engineering and Technology Edition), 2022, 52(8): 1872-1880.
[15] Ming LIU,Yu-hang YANG,Song-lin ZOU,Zhi-cheng XIAO,Yong-gang ZHANG. Application of enhanced edge detection image algorithm in multi-book recognition [J]. Journal of Jilin University(Engineering and Technology Edition), 2022, 52(4): 891-896.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!