A closer look onto breast density with weakly supervised dense-tissue masks
Mickael Tardy Bruno Scheffer Diana Mateus
eess.IV


The matched product of set-theoretical solutions associated with shelves
    

Francesco CATINOaCONTACT  F. Catino.  Email: francesco.catino@unisalento.it,
Address: Dipartimento di Matematica e Fisica “Ennio De Giorgi",
Università del Salento -
Via Provinciale Lecce - Arnesano , 73100 Lecce (Italy), 
Ilaria COLAZZOa, 
and 
Paola STEFANELLIa

aDipartimento di Matematica e Fisica “Ennio De Giorgi",
	Università del Salento, Lecce, Italy

    Received: date / Accepted: date
=============================================================================================================================================================================================================================================================================================================================================================================





This work focuses on the automatic quantification of the breast density from digital mammography imaging. Using only categorical image-wise labels we train a model capable of predicting continuous density percentage as well as providing a pixel wise support frit for the dense region. In particular we propose a weakly supervised loss linking the density percentage to the mask size. 



Deep Learning, Weakly supervised segmentation, Regression, Breast density






§ PURPOSE

Breast density is a biomarker for breast cancer development risk that suggests that the risk of cancer development increases with denser breasts. Moreover, the detection of cancer in dense tissues and, more generally, in dense breasts is often considered more challenging due to the similar visual aspects of normal and abnormal tissues, which complicates the interpretation of mammographic images. For the above reasons, we argue that computer-aided decision systems for early breast cancer detection should both, quantitatively evaluate the breast density, and evaluate the spatial distribution of the dense tissues.



§ METHODS

In clinical practice, breast density is usually assessed image-wise using a classification grid like the BI-RADS (Breast imaging-reporting and data system) <cit.>. In the present work, we propose to estimate breast density at the pixel level while using only image-wise ground truth from the BI-RADS scale. Our goal is to generate a breast density mask, identifying pixels associated with the tissue that contributed to the density class. To achieve our goal, we propose a novel loss linking the sought breast density mask to the globally estimated breast density (fig. <ref>). We formulate the problem as a weakly supervised binary semantic segmentation. Our approach is related to recent efforts to reduce the amount of supervision <cit.>. 

In practice, we rely on a modified U-Net architecture <cit.> and on an extended 12-class density grid that improves the density resolution compared to traditional BI-RADS classification (4th edition). Compared to the state-of-the-art, our classification and segmentation scheme does not rely on the model13s attention but uses a loss function efficiently correlating a tissue mask with the target breast density values. Moreover, the output is constrained with the breast binary mask removing useless activations.




§ RESULTS

The database for training and tests consists of 1232 and 370 images respectively. We got promising results with a mean absolute error (MAE) of 6.7% for the density regression estimate (see tab. <ref>) and an accuracy of 78% for 4-class BI-RADS density classification (tab. <ref>). Our comparison baseline is a VGG-like regression model trained on the same dataset.

To validate the segmentation performance, we collected regions of interest on several images (16) and calculated the Dice=0.65. Overall we obtain clinically meaningful segmentation masks offering valuable insights into the spatial distribution of the dense tissues (fig. <ref>). In comparison, we demonstrate the inefficiency of the attention-based techniques for the breast density mask generation.

In addition we validate our approach on the INBreast <cit.> database. Without any additional training and a simple preprocessing we obtained 65 % accuracy and MAE = 13%. We note that our results are comparable to other works on the same dataset (64.53%, <cit.>, 67.8% <cit.>. 



§ CONCLUSIONS

Our approach to link breast density classification to the spatial distribution of dense tissue has a positive effect on classification scores while providing an additional output mask of the dense regions. These results are interesting given the considerably low requirements on ground truth (just a class instead of an image mask) and the size of the training dataset. 
















Research funding is provided by Hera-MI, SAS and Association Nationale de la Recherche et de la Technologie via CIFRE grant no. 2018/0308.





