Deakin University
Browse

File(s) under embargo

A novel myocarditis detection combining deep reinforcement learning and an improved differential evolution algorithm

journal contribution
posted on 2024-03-12, 02:44 authored by J Yang, T Sadiq, J Xiong, M Awais, U Aslam Bhatti, Roohallah AlizadehsaniRoohallah Alizadehsani, JM Gorriz
AbstractMyocarditis is a serious cardiovascular ailment that can lead to severe consequences if not promptly treated. It is triggered by viral infections and presents symptoms such as chest pain and heart dysfunction. Early detection is crucial for successful treatment, and cardiac magnetic resonance imaging (CMR) is a valuable tool for identifying this condition. However, the detection of myocarditis using CMR images can be challenging due to low contrast, variable noise, and the presence of multiple high CMR slices per patient. To overcome these challenges, the approach proposed incorporates advanced techniques such as convolutional neural networks (CNNs), an improved differential evolution (DE) algorithm for pre‐training, and a reinforcement learning (RL)‐based model for training. Developing this method presented a significant challenge due to the imbalanced classification of the Z‐Alizadeh Sani myocarditis dataset from Omid Hospital in Tehran. To address this, the training process is framed as a sequential decision‐making process, where the agent receives higher rewards/penalties for correctly/incorrectly classifying the minority/majority class. Additionally, the authors suggest an enhanced DE algorithm to initiate the backpropagation (BP) process, overcoming the initialisation sensitivity issue of gradient‐based methods like back‐propagation during the training phase. The effectiveness of the proposed model in diagnosing myocarditis is demonstrated through experimental results based on standard performance metrics. Overall, this method shows promise in expediting the triage of CMR images for automatic screening, facilitating early detection and successful treatment of myocarditis.

History

Journal

CAAI Transactions on Intelligence Technology

Location

London, Eng.

ISSN

2468-6557

eISSN

2468-2322

Language

eng

Publication classification

C1 Refereed article in a scholarly journal

Publisher

Wiley

Usage metrics

    Research Publications

    Exports

    RefWorks
    BibTeX
    Ref. manager
    Endnote
    DataCite
    NLM
    DC