Transfer learning method in the problem of binary classification of chest X-rays

Kolesnikov Dmitry
3/19/2023
eess.IVcs.CV

No official code found. Best implementation: Koldim2001/transfer_learning_CNN (4 stars, PyTorch, last commit 2023-05-21).

Abstract

The possibility of high-precision and rapid detection of pathologies on chest X-rays makes it possible to detect the development of pneumonia at an early stage and begin immediate treatment. Artificial intelligence can speed up and qualitatively improve the procedure of X-ray analysis and give recommendations to the doctor for additional consideration of suspicious images. The purpose of this study is to determine the best models and implementations of the transfer learning method in the binary classification problem in the presence of a small amount of training data. In this article, various methods of augmentation of the initial data and approaches to training ResNet and DenseNet models for black-and-white X-ray images are considered, those approaches that contribute to obtaining the highest results of the accuracy of determining cases of pneumonia and norm at the testing stage are identified.

View on arXivView PDF

Code Implementations(1)

PyTorch

Бинарная классификация рентгеновских снимков грудной клетки. Определение наличия пневмонии у пациентов при помощи различных CNN архитектур. Использование метода Transfer Learning

40Python, Jupyter NotebookFeb 19, 20233 years ago
arxiv-papersbinary-classificationcnndeep-learningdensenet+5 more

Similar Papers

Cite this paper

@article{dmitry2023transfer,
  title  = {Transfer learning method in the problem of binary classification of chest X-rays},
  author = {Kolesnikov Dmitry},
  year   = {2023},
  eprint = {2303.10601},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2303.10601v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2303.10601.svg)](https://codeofpaper.com/paper/2303.10601)

Discussion