Automated Wound Segmentation using Attention Mechanism based on Enhanced MobileNetV2
Date
2024
Authors
Muhtasim, D.A.
Pavel, M.I.
Paul, A.
Shakib, A.W.
Gazi, S.S.
Hasan, M.M.
Editors
Advisors
Journal Title
Journal ISSN
Volume Title
Type:
Conference paper
Citation
2024 IEEE 3rd International Conference on Robotics, Automation, Artificial-Intelligence and Internet-of-Things, RAAICON 2024 - Proceedings, 2024, pp.96-101
Statement of Responsibility
Conference Name
RAAICON 2024 - Proceedings (29 Nov 2024 - 30 Nov 2024 : Dhaka, Bangladesh)
Abstract
Wound healing is a morphogenetic reaction to injury that restores physiological and anatomical function. It is crucial in fields like obesity treatment, cancer, trauma, burns, and diabetic foot ulcer (DFU) care. Wound care experts rely on affordable, high-resolution cell phone cameras for image documentation; however, lack of information can hinder proper wound management, documentation, and diagnosis, highlighting the need for improved therapeutic techniques and better automated image segmentation methods. Recent deep learning approaches for automatic feature extraction from images face challenges due to limited training data and low neural networks depths, which hinder accuracy by failing to capture complex patterns. This research introduces an improved MobileNetV2 model which utilized transfer learning, two additional dense layers, and fusion with spatial attention mechanism to improve the accuracy of the wound segmentation task. The proposed approach achieved a dice score of 92.40% on the Foot Ulcer Dataset and 96.92% on the Medetec Wound Dataset. In comparison to existing state-of- the-art deep learning-based VGG16, SegNet, U-Net, and MobileNetV2+CCL techniques, the proposed attention-based MobileNetV2 technique obtained the highest dice score.
School/Discipline
Dissertation Note
Provenance
Description
Access Status
Rights
Copyright 2024 IEEE.