Current browse context:
q-bio.QM
Change to browse by:
References & Citations
Quantitative Biology > Quantitative Methods
Title: Using convolutional neural networks for the classification of breast cancer images
(Submitted on 31 Aug 2021 (v1), last revised 27 Oct 2022 (this version, v2))
Abstract: In this study I compare different architectures of convolutional neural networks and different hardware acceleration devices for the detection of breast cancer from microscopic images of sentinel lymph nodes tissue. Convolutional models with increasing depth are trained and tested on a public data set of more than 300,000 images of lymph node tissue, on four different hardware acceleration cards, using an off-the-shelf deep learning framework. The impact of transfer learning, data augmentation and hyperparameters fine-tuning are also tested. Hardware acceleration device performance can improve training time by a factor of five to twelve, depending on the model used. On the other hand, increasing convolutional depth will augment the training time by a factor of four to six times, depending on the acceleration device used. Increasing the depth of the model, as could be expected, clearly improves performance, while data augmentation and transfer learning do not. Fine-tuning the hyperparameters of the model notably improves the results, with the best model showing a performance comparable to state-of-the-art models.
Submission history
From: Eric Bonnet [view email][v1] Tue, 31 Aug 2021 07:53:41 GMT (387kb,D)
[v2] Thu, 27 Oct 2022 14:15:20 GMT (387kb,D)
Link back to: arXiv, form interface, contact.