Pneumonia Detection on Chest X-rays with Attention Mask Data Augmentation

Fengwei Lai, Seiichiro Kamata, Zhengbo Luo*

*この研究の対応する著者

研究成果: Conference contribution

抄録

Each year, pneumonia affects about 450 million people globally and results in about 4 million deaths. In developing countries, pneumonia remains a leading cause of death in chronic patients and older adults. Chest X-rays are currently the best available method for diagnosing pneumonia, which is very critical for the treatment of patients. Deep learning-based image recognition technology can significantly improve the efficiency of pneumonia detection. Data Augmentation is a technology that artificially expands the training data set by using limited data to generate more comparable data. Excellent data augmentation methods can effectively improve the performance of neural networks and are currently widely used in various fields of deep learning. We propose a novel data augmentation method called Attention Mask in this paper, which provides accurate predictions and a more explainable attention focus comparing with many traditional data augmentation methods, such as random erasing and hide-and-seek. We guide the weight and focal point of the model with the attention mechanism to avoid the model from relying too much on superficial features. After data augmentation, the self-ensemble of different stage models also makes the entire system more stable. The experiments show that the attention is reasonably diverted, which is extremely helpful in classifying the target that is fallibility correctly. On Chest X-Ray Images (Pneumonia) dataset for classification, our method notably improves performance over baselines.

本文言語English
ホスト出版物のタイトルProceedings of 2020 2nd International Conference on Video, Signal and Image Processing, VSIP 2020
出版社Association for Computing Machinery
ページ10-16
ページ数7
ISBN(電子版)9781450388931
DOI
出版ステータスPublished - 2020 4 12
イベント2nd International Conference on Video, Signal and Image Processing, VSIP 2020 - Virtual, Online, Indonesia
継続期間: 2020 12 42020 12 6

出版物シリーズ

名前ACM International Conference Proceeding Series

Conference

Conference2nd International Conference on Video, Signal and Image Processing, VSIP 2020
国/地域Indonesia
CityVirtual, Online
Period20/12/420/12/6

ASJC Scopus subject areas

  • ソフトウェア
  • 人間とコンピュータの相互作用
  • コンピュータ ビジョンおよびパターン認識
  • コンピュータ ネットワークおよび通信

フィンガープリント

「Pneumonia Detection on Chest X-rays with Attention Mask Data Augmentation」の研究トピックを掘り下げます。これらがまとまってユニークなフィンガープリントを構成します。

引用スタイル