Enhancing Classification Performance in Knee Magnetic Resonance Imaging Using Adversarial Data Augmentation

Zhongbo Yan, Xu Yang, Chak Fong Chong, Yapeng Wang

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

The utilization of adversarial data augmentation has demonstrated the potential capability to enhance the classification performance in training deep neural networks to perform computer vision tasks. In this paper, we investigate the effectiveness of this approach as a strategy for enlarging a knee Magnetic Resonance Imaging (MRI) dataset by adding adversarial perturbation. Specifically, we use the Fast Gradient Sign Method (FGSM) to perturb a subset of the training dataset as extra training images to re-train a baseline model that was trained under the same configuration as the top-ranked model on the MRNet leaderboard. Particularly, unlike most of the current work, we investigate the impact of two hyperparameters (attack magnitude and the proportion of data to be re-trained) on the performance of area under the ROC curve (AUC), accuracy, sensitivity, and specificity. Additionally, our results show that adversarial data augmentation can further improve the well-trained baseline model's AUC by 0.26%, as well as provide a slight improvement in specificity at the same classification threshold. These findings underscore the potential advantages of adversarial data augmentation as a technique for optimizing the decision boundaries of deep learning models. The code of this work will be available on GitHub after the paper is published.

Original languageEnglish
Title of host publicationICSESS 2023 - Proceedings of 2023 IEEE 14th International Conference on Software Engineering and Service Science
EditorsLi Wenzheng
PublisherIEEE Computer Society
Pages19-24
Number of pages6
ISBN (Electronic)9798350336269
DOIs
Publication statusPublished - 2023
Event14th IEEE International Conference on Software Engineering and Service Science, ICSESS 2023 - Beijing, China
Duration: 17 Oct 202318 Oct 2023

Publication series

NameProceedings of the IEEE International Conference on Software Engineering and Service Sciences, ICSESS
ISSN (Print)2327-0586
ISSN (Electronic)2327-0594

Conference

Conference14th IEEE International Conference on Software Engineering and Service Science, ICSESS 2023
Country/TerritoryChina
CityBeijing
Period17/10/2318/10/23

Keywords

  • Adversarial Data Augmentation
  • Binary Classification
  • FGSM
  • Medical Image Data Classification

Fingerprint

Dive into the research topics of 'Enhancing Classification Performance in Knee Magnetic Resonance Imaging Using Adversarial Data Augmentation'. Together they form a unique fingerprint.

Cite this