International Conference on Computer and Knowledge Engineering

Home / 14th International Conference on Computer and Knowledge Engineering

YOLOatt-Med: YOLO-Based Attention Mechanism for Medical Image Classification

Authors :

Fatemeh Naserizadeh¹ Erfan Akbarnezhad Sany² Parsa Sinichi³ Seyyed Abed Hosseini⁴

1- Faculty of Electrical and Computer Engineering, Malek Ashtar University of Technology, Tehran, Iran 2- Faculty of Computer Engineering, Quchan University of Technology, Quchan, Iran 3- Faculty of Computer Engineering, Quchan University of Technology, Quchan, Iran 4- Department of Electrical Engineering, Mashhad Branch, Islamic Azad University, Mashhad, Iran

Keywords :

Diagnosis،YOLOv8،Convolutional Block Attention Module،Classification،Lung cancer

Abstract :

Early and accurate diagnosis of respiratory diseases can significantly reduce mortality rates. This study presents a novel approach for classifying respiratory diseases from medical images using computer vision. The method integrates YOLOv8 (you only look once version 8), a powerful object detection model, with the convolutional block attention module (CBAM) to enhance feature extraction and improve classification accuracy. The proposed model was evaluated on two datasets: chest computed tomography (CT) scans for lung cancer classification and chest X-rays for pneumonia classification. The results demonstrate that the CBAM-YOLOv8 model outperforms existing state-of-the-art methods. This improvement highlights the effectiveness of the proposed method in focusing on critical image features for accurate disease classification.

Papers List

List of archived papers

Machine Learning-Driven Prediction of Anti-Alzheimer Drug Efficacy Using PubChem Molecular Fingerprints

Mohammad Javad Sadeghi - Mohammad Javad Nemati - AliAsghar Zare - Mohammadreza Shams

Paddy Plant Stress Identification Using Few-Shot Learning Framework

Ervin Gubin Moung - Pavindrah Naidu a/l Narayanasamy Naiidu - Maisarah Mohd Sufian - Valentino Liaw - Ali Farzamnia - Lorita Angeline

A Dual-Branch Attention-Enhanced CNN for Corn Leaf Disease Classification via RGB-HLS Color Space Fusion

Mohammad Ali Salehi Rad - Kamran Kazemi - Mohammad Sadegh Helfroush - Tahereh Golshaeian

Multi-Fusion Ensemble CNN for Drug–Target Binding Affinity Prediction Using Transformer-Based Molecular and Protein Representations

Betsabeh Tanoori

Automated software design using Machine Learning With Natural Language Processing

Fahimeh Khedmatkon - Seyed Mohammad Hossein Hasheminejad - Jaleh Shoshtarian Malak

Optimizing Question-Answering Framework Through Integration of Text Summarization Model and Third-Generation Generative Pre-Trained Transformer

Ervin Gubin Moung - Toh Sin Tong - Maisarah Mohd Sufian - Valentino Liaw - Ali Farzamnia - Farashazillah Yahya

EfficientNetB0’s Hybrid Approach for Brain Tumor Classification from MRI Images Using Deep Learning and Bagging Trees

Yeganeh Modaresnia - Farhad Abedinzadeh Torghabeh - Seyyed Abed Hosseini

Extracting Major Topics of COVID-19 Related Tweets

Faezeh Azizi - Hamed Vahdat-Nejad - Hamideh Hajiabadi - Mohammad Hossein Khosravi

Area-Efficient VLSI Implementation of Bit-Serial Multiplier Using Polynomial Basis over GF(2m)

Saeideh Nabipour - Javad Javidan - Gholamreza Zare Fatin

Camouflage Object Segmentation with Attention-Guided Pix2Pix and Boundary Awareness

Erfan Akbarnezhad Sany - Fatemeh Naserizadeh - Parsa Sinichi - Seyyed Abed Hosseini