CS-GIN: A cosine similarity-based graph isomorphism network for multi-class anemia classification

Open

Iswanto Suwarno, Alfian Ma'arif, Purwono, Nia Maharani Raharja, Tri Ratnaningsih, Tanzilal Mustaqim, Dimas Chaerul Ekty Saputra

2026 Array Vol. 31 Article Cited by 0 Quartile

Abstract

Accurate anemia diagnosis remains challenging due to the heterogeneous characteristics of anemia subtypes and the complex interactions among hematological biomarkers. Conventional machine learning approaches generally treat patient records as independent observations, limiting their ability to exploit clinically meaningful relationships among patients with similar hematological profiles. To address this limitation, this study proposes a Cosine Similarity-based Graph Isomorphism Network (CS-GIN) for multi-class anemia classification. A hematological dataset consisting of 513 patient records was transformed into a patient similarity graph using cosine similarity, where patients were represented as graph nodes and similarity relationships were represented as graph edges. A parameter sensitivity analysis was first conducted to determine the optimal cosine similarity threshold, followed by comparisons with Euclidean distance- and K-nearest neighbor (KNN)-based graph construction strategies. The resulting graphs were evaluated using Graph Convolutional Network (GCN), Graph Attention Network (GAT), conventional Graph Isomorphism Network (GIN), and the proposed CS-GIN under stratified five-fold cross-validation. Experimental results demonstrated that CS-GIN achieved the best performance among all graph neural network models, obtaining an average accuracy of 90.24%, outperforming GCN (84.39%), GAT (84.59%), and conventional GIN (83.23%). Statistical significance was confirmed using the Friedman and Nemenyi tests. Explainability analyses using SHAP and GNNExplainer consistently identified Ferritin, HbA2, RET-He, and MCV as the most influential biomarkers, while graph topology and embedding analyses demonstrated that the learned representations effectively preserved meaningful patient similarity relationships. These findings demonstrate that similarity-aware graph construction substantially enhances graph representation learning and provides an effective, interpretable framework for graph-based multi-class anemia classification. © 2026 The Authors

Affiliations

Department of Electrical Engineering, Universitas Muhammadiyah Yogyakarta, Yogyakarta, 55183, Indonesia; Department of Electrical Engineering, Faculty of Industrial Technology, Universitas Ahmad Dahlan, Yogyakarta, 55191, Indonesia; Department of Informatics, Faculty of Science and Technology, Universitas Harapan Bangsa, Banyumas, 53182, Indonesia; Department of Information Engineering, UIN Sunan Kalijaga Yogyakarta, Yogyakarta, 55281, Indonesia; Department of Clinical Pathology and Laboratory Medicine, Faculty of Medicine, Public Health and Nursing, Universitas Gadjah Mada, Special Region of Yogyakarta, Yogyakarta, 55281, Indonesia; Informatics Study Program, Telkom University, Surabaya Campus, East Java, Surabaya, 60231, Indonesia; Center of Excellence for Motion Technology for Safety Health and Wellness, Research Institute of Sustainable Society, Telkom University, Surabaya Campus, East Java, Surabaya, 60231, Indonesia

Research at a Glance

Premium content — register to unlock

Research at a Glance

Register to unlock

Topics & SDG Alignment

Premium content — register to unlock

Topics & SDG Alignment

Register to unlock

Collaboration

Premium content — register to unlock

Collaboration

Register to unlock

Author Profile (Selected)

Premium content — register to unlock

Author Profile (Selected)

Register to unlock

References Overview

Premium content — register to unlock

References Overview

Register to unlock

Journal & Source

Premium content — register to unlock

Journal & Source

Register to unlock

Metadata & Integrity

Premium content — register to unlock

Metadata & Integrity

Register to unlock