Application of Transfer Learning and Ensemble Learning in Image-level Classification for Breast Histopathology

04/18/2022
by   Yuchao Zheng, et al.
20

Background: Breast cancer has the highest prevalence in women globally. The classification and diagnosis of breast cancer and its histopathological images have always been a hot spot of clinical concern. In Computer-Aided Diagnosis (CAD), traditional classification models mostly use a single network to extract features, which has significant limitations. On the other hand, many networks are trained and optimized on patient-level datasets, ignoring the application of lower-level data labels. Method: This paper proposes a deep ensemble model based on image-level labels for the binary classification of benign and malignant lesions of breast histopathological images. First, the BreakHis dataset is randomly divided into a training, validation and test set. Then, data augmentation techniques are used to balance the number of benign and malignant samples. Thirdly, considering the performance of transfer learning and the complementarity between each network, VGG-16, Xception, Resnet-50, DenseNet-201 are selected as the base classifiers. Result: In the ensemble network model with accuracy as the weight, the image-level binary classification achieves an accuracy of 98.90%. In order to verify the capabilities of our method, the latest Transformer and Multilayer Perception (MLP) models have been experimentally compared on the same dataset. Our model wins with a 5%-20% advantage, emphasizing the ensemble model's far-reaching significance in classification tasks. Conclusion: This research focuses on improving the model's classification performance with an ensemble algorithm. Transfer learning plays an essential role in small datasets, improving training speed and accuracy. Our model has outperformed many existing approaches in accuracy, providing a method for the field of auxiliary medical diagnosis.

READ FULL TEXT

page 2

page 3

page 9

page 15

page 17

research
11/01/2022

ViT-DeiT: An Ensemble Model for Breast Cancer Histopathological Images Classification

Breast cancer is the most common cancer in the world and the second most...
research
02/17/2021

Ensemble Transfer Learning of Elastography and B-mode Breast Ultrasound Images

Computer-aided detection (CAD) of benign and malignant breast lesions be...
research
09/16/2021

A Medical Pre-Diagnosis System for Histopathological Image of Breast Cancer

This paper constructs a novel intelligent medical diagnosis system, whic...
research
07/11/2021

BCNet: A Deep Convolutional Neural Network for Breast Cancer Grading

Breast cancer has become one of the most prevalent cancers by which peop...
research
02/11/2020

Optimal Transfer Learning Model for Binary Classification of Funduscopic Images through Simple Heuristics

Deep learning models have the capacity to fundamentally revolutionize me...
research
12/29/2020

COIN: Contrastive Identifier Network for Breast Mass Diagnosis in Mammography

Computer-aided breast cancer diagnosis in mammography is a challenging p...

Please sign up or login with your details

Forgot password? Click here to reset