Authors:
Francesca Lizzi
1
;
2
;
3
;
Stefano Atzori
4
;
Giacomo Aringhieri
4
;
Paolo Bosco
2
;
Carolina Marini
4
;
Alessandra Retico
2
;
Antonio C. Traino
4
;
Davide Caramella
3
;
4
and
M. Evelina Fantacci
2
;
3
Affiliations:
1
Scuola Normale Superiore, Pisa, Italy
;
2
Istituto Nazionale di Fisica Nucleare (INFN), Pisa, Italy
;
3
University of Pisa, Pisa, Italy
;
4
Azienda Ospedaliero-Universitaria Pisana (AOUP), Pisa, Italy
Keyword(s):
Convolutional Neural Networks, Breast Density, BI-RADS, Residual Neural Networks.
Abstract:
In this paper, we propose a data-driven method to classify mammograms according to breast density in BI-RADS standard. About 2000 mammographic exams have been collected from the “Azienda Pisana” (AOUP, Pisa, IT). The dataset has been classified according to breast density in the BI-RADS standard. Once the dataset has been labeled by a radiologist, we proceeded by building a Residual Neural Network in order to classify breast density in two ways. First, we classified mammograms using two “super-classes” that are dense and non-dense breast. Second, we trained the residual neural network to classify mammograms according to the four classes of the BI-RADS standard. We evaluated the performance in terms of the accuracy and we obtained very good results compared to other works on similar classification tasks. In the near future, we are going to improve the results by increasing the computing power, by improving the quality of the ground truth and by increasing the number of samples in the
dataset.
(More)