UPM Institutional Repository

An efficient lung disease classification from X-ray image using Graph Neural Network and Transformer


Citation

Zhihao, Jiang and Manshor, Noridayu and Hamdan, Hazlina and Limi, Chen (2025) An efficient lung disease classification from X-ray image using Graph Neural Network and Transformer. International Journal on Advanced Science, Engineering and Information Technology, 15 (1). pp. 275-282. ISSN 2088-5334; eISSN: 2460-6952

Abstract

This research addresses the need for improved accuracy and efficiency in lung disease classification from X-ray images by developing a novel approach that integrates Graph Neural Networks (GNNs) with Transformer models. Using the extensive ChestXray14 dataset, which includes a wide array of lung disease cases, this study introduces and validates the TransGNN model. TransGNN harnesses the complementary strengths of GNNs and Transformers through advanced attention mechanisms and a dual-branch architecture, effectively capturing the complex and variable characteristics inherent in medical imaging data. The methodological framework includes rigorous data preprocessing, the application of weighted focus loss functions to address significant class imbalances, and extensive data augmentation techniques to bolster the model's robustness during testing. Results show that TransGNN surpasses conventional models by achieving superior classification accuracy across multiple lung diseases, demonstrating substantial improvements in diagnostic precision and reliability. Furthermore, the model's continuous learning mechanisms enable it to adapt seamlessly to new data and advances in medical imaging technology, making it highly versatile. This study highlights the potential of GNNs and Transformers to revolutionize the diagnostic landscape, offering a powerful, precise, and efficient tool for lung disease diagnosis. Future research should aim to incorporate additional patient-specific data, explore more advanced neural network architectures, and validate the model across diverse patient populations to enhance diagnostic accuracy further and expand its practical applicability in clinical settings, paving the way for a new standard in AI-driven medical diagnostics.


Download File

[img] Text
125326.pdf - Published Version
Available under License Creative Commons Attribution Share Alike.

Download (1MB)

Additional Metadata

Item Type: Article
Subject: Computer Science (all)
Subject: Agricultural and Biological Sciences (all)
Subject: Engineering (all)
Divisions: Faculty of Computer Science and Information Technology
DOI Number: https://doi.org/10.18517/ijaseit.15.1.12590
Publisher: Insight Society
Keywords: Graph Neural Networks (GNNs); Lung disease classification; transformers; X-ray images
Sustainable Development Goals (SDGs): SDG 3: Good Health and Well-being, SDG 9: Industry, Innovation and Infrastructure, SDG 4: Quality Education
Depositing User: Ms. Nur Faseha Mohd Kadim
Date Deposited: 07 May 2026 03:00
Last Modified: 07 May 2026 03:00
Altmetrics: http://www.altmetric.com/details.php?domain=psasir.upm.edu.my&doi=10.18517/ijaseit.15.1.12590
URI: http://psasir.upm.edu.my/id/eprint/125326
Statistic Details: View Download Statistic

Actions (login required)

View Item View Item