Mobile Freeze-Net with Attention-based Loss Function for Covid-19 Detection from an Imbalanced CXR Dataset
- Title
- Mobile Freeze-Net with Attention-based Loss Function for Covid-19 Detection from an Imbalanced CXR Dataset
- Creator
- Roy S.; Khurana R.
- Description
- In this paper, we present a novel framework, that is, Mobile Freeze-Net along with Attention-based Loss Function, for Covid-19 detection from a Chest X-Ray (CXR) dataset. First, we have observed that by freezing 50% of a Mobile Net-V2 model (means fine-tuning 50% layers from ImageNet dataset) has automatically removed the class imbalance problem from the CXR dataset considerably. We call this 50% frozen Mobile Net-V2 model as Mobile Freeze-Net. Secondly, we have proposed an Attention-based Loss function, which provides more attention to the class, having higher inter-class similarity. We have computed attention weights for each class from the statistical inference of the dataset itself, by employing a Monte-Carlo method and thereafter, we have incorporated those weights into WCCE loss function of Mobile Freeze-Net model. By utilizing Mobile freeze-Net, we have achieved testing accuracy, F1 score, precision and recall of 93%, 94%, 93% and 94% respectively. This is approximately 3-4% improvement compared to 100% fine tuning of Mobile-Net V2. Furthermore, we have achieved approximate 1-2% improvement of Mobile Freeze-Net, after incorporating Attention-based Loss function. For the validity of the proposed framework, we have conducted experiments with 10-fold cross validation. All these experimental results suggest that our proposed framework has outperformed other existing models considerably. 2023 Owner/Author(s).
- Source
- Proceedings of the ACM Symposium on Applied Computing, pp. 611-613.
- Date
- 2023-01-01
- Publisher
- Association for Computing Machinery
- Subject
- attention-based loss function; chest X-ray (CXR) images; class imbalance; Covid-19 detection; mobile freeze-net
- Coverage
- Roy S., Christ (Deemed to Be University), Bangalore, India; Khurana R., Bharati Vidyapeeth's College of Engineering, New Delhi, India
- Rights
- Restricted Access
- Relation
- ISBN: 978-145039517-5
- Format
- Online
- Language
- English
- Type
- Conference paper
Collection
Citation
Roy S.; Khurana R., “Mobile Freeze-Net with Attention-based Loss Function for Covid-19 Detection from an Imbalanced CXR Dataset,” CHRIST (Deemed To Be University) Institutional Repository, accessed February 27, 2025, https://archives.christuniversity.in/items/show/19612.