论文标题
可解释的深度学习回归用于MRI的乳房密度估计
Interpretable deep learning regression for breast density estimation on MRI
论文作者
论文摘要
乳房密度是纤维岩组织(FGT)与总乳房体积之间的比率,可以通过放射科医生进行定性评估,并通过计算机算法进行定量评估。这些算法通常依赖于乳房和FGT体积的分割。在这项研究中,我们提出了一种直接评估MRI乳房密度并提供这些评估的方法。 我们使用回归卷积神经网络(CNN)评估了506例乳腺癌患者的乳房密度。 CNN的输入是128 x 128素的乳房MRI切片,输出是0(脂肪乳房)和1(浓密的乳房)之间的连续密度值。我们使用350名患者训练CNN,75例进行验证,81例进行独立测试。我们调查了为什么使用深沙普利添加剂解释(Shap)来达到CNN的预测密度。 CNN在测试集中预测的密度与地面真相密度显着相关(n = 81例,Spearman的Rho = 0.86,p <0.001)。在检查CNN基于其基于其预测的内容时,我们发现FGT中的体素通常具有正壳值,脂肪组织中的体素通常具有负壳值负值,而在非胸皮组织中的体素通常在零接近零。这意味着密度的预测基于我们期望的结构,即FGT和脂肪组织。 总而言之,我们提出了一种可解释的深度学习回归方法,用于对MRI的乳房密度估计,并有令人鼓舞的结果。
Breast density, which is the ratio between fibroglandular tissue (FGT) and total breast volume, can be assessed qualitatively by radiologists and quantitatively by computer algorithms. These algorithms often rely on segmentation of breast and FGT volume. In this study, we propose a method to directly assess breast density on MRI, and provide interpretations of these assessments. We assessed breast density in 506 patients with breast cancer using a regression convolutional neural network (CNN). The input for the CNN were slices of breast MRI of 128 x 128 voxels, and the output was a continuous density value between 0 (fatty breast) and 1 (dense breast). We used 350 patients to train the CNN, 75 for validation, and 81 for independent testing. We investigated why the CNN came to its predicted density using Deep SHapley Additive exPlanations (SHAP). The density predicted by the CNN on the testing set was significantly correlated with the ground truth densities (N = 81 patients, Spearman's rho = 0.86, P < 0.001). When inspecting what the CNN based its predictions on, we found that voxels in FGT commonly had positive SHAP-values, voxels in fatty tissue commonly had negative SHAP-values, and voxels in non-breast tissue commonly had SHAP-values near zero. This means that the prediction of density is based on the structures we expect it to be based on, namely FGT and fatty tissue. To conclude, we presented an interpretable deep learning regression method for breast density estimation on MRI with promising results.