A Novel Method for Myocardial Image Classification using Data Augmentation

Abstract

Myocarditis is an important public health concern since it can cause heart failure and abrupt death. It can be diagnosed with magnetic resonance imaging (MRI) of the heart, a non-invasive imaging technology with the potential for operator bias. The study provides a deep learning-based model for myocarditis detection using CMR images to support medical professionals. The proposed architecture comprises a convolutional neural network (CNN), a fully-connected decision layer, a generative adversarial network (GAN)-based algorithm for data augmentation, an enhanced DE for pre-training weights, and a reinforcement learning-based method for training. We present a new method of employing produced images for data augmentation based on GAN to improve the classification performance of the provided CNN. Unbalanced data is one of the most significant classification issues, as negative samples are more than positive, decimating system performance. To solve this issue, we offer an RL-based training method that learns minority class examples with attention. In addition, we tackle the challenges associated with the training step, which typically relies on gradient-based techniques for the learning process; however, these methods often face issues like sensitivity to initialization. To start the BP process, we present an improved differential evolution (DE) technique that leverages a clustering-based mutation operator. It recognizes a successful cluster for DE and applies an original updating strategy to produce potential solutions. We assess our suggested model on the Z-Alizadeh Sani myocarditis dataset and show that it outperforms other methods. Keywords—Myocarditis; generative adversarial network; data augmentation; differential evolution