English

GANs 'N Lungs: improving pneumonia prediction

Image and Video Processing 2019-08-02 v1 Computer Vision and Pattern Recognition

Abstract

We propose a novel method to improve deep learning model performance on highly-imbalanced tasks. The proposed method is based on CycleGAN to achieve balanced dataset. We show that data augmentation with GAN helps to improve accuracy of pneumonia binary classification task even if the generative network was trained on the same training dataset.

Keywords

Cite

@article{arxiv.1908.00433,
  title  = {GANs 'N Lungs: improving pneumonia prediction},
  author = {Tatiana Malygina and Elena Ericheva and Ivan Drokin},
  journal= {arXiv preprint arXiv:1908.00433},
  year   = {2019}
}

Comments

Accepted as an extended abstract for MIDL 2019 [arXiv:1907.08612]

R2 v1 2026-06-23T10:37:22.940Z