HG-YOLO: Improving Tumor Detection with PP-HGNet and Global Attention Mechanism

Kien Trang, Bao Quoc Vuong, An Hoang Nguyen, Fung Fung Ting, Chee Ming Ting

Research output: Chapter in Book/Report/Conference proceedingConference PaperResearchpeer-review

Abstract

Cancer continues to be a major global health issue, which is defined by uncontrolled cell proliferation and the potential invasion to other areas of the body. The early and precise detection of tumors through medical imaging is essential in improving cancer prognosis and treatment outcomes. This study presents HG-YOLO, a novel YOLO-based architecture enhanced by the integration of the PP-HGNet backbone and a Global Attention Mechanism (GAM). HG-YOLO capitalizes on the robust feature extraction capabilities of PP-HGNet and the attention-enhancing properties of GAM to improve the potential features. This combination aims to improve the sensitivity and precision of tumor localization, especially in complex cases where tumors are small or poorly delineated. Overall, the model is assessed using the Brain Tumor Detection 2020 (Br35H) dataset with the Magnetic Resonance Imaging (MRI) images. Comparative studies show that our HG-YOLO outperforms the prior versions in terms of Precision, Recall, mAP50 and mAP50-95 - giving 0.934, 0.915, 0.953, and 0.728, respectively.

Original languageEnglish
Title of host publicationComputational Science and Its Applications – ICCSA 2025 - 25th International Conference Istanbul, Turkey, June 30 – July 3, 2025 Proceedings, Part I
EditorsOsvaldo Gervasi, Beniamino Murgante, Chiara Garau, Yeliz Karaca, David Taniar, Ana Maria A. C. Rocha, Bernady O. Apduhan
Place of PublicationCham Switzerland
PublisherSpringer
Pages332-344
Number of pages13
ISBN (Electronic)9783031970009
ISBN (Print)9783031969997
DOIs
Publication statusPublished - 2025
EventInternational Conference on Computational Science and Applications 2025 - Istanbul, Türkiye
Duration: 30 Jun 20253 Jul 2025
Conference number: 25th
https://link.springer.com/book/10.1007/978-3-031-97000-9 (Proceedings)
https://iccsa.org/ (Website)

Publication series

NameLecture Notes in Computer Science
PublisherSpringer
Volume15648
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

ConferenceInternational Conference on Computational Science and Applications 2025
Abbreviated titleICCSA 2025
Country/TerritoryTürkiye
CityIstanbul
Period30/06/253/07/25
Internet address

Keywords

  • brain tumor
  • deep learning
  • MRI
  • PP-HGNet
  • YOLO

Cite this