返回
Multi-Task Fusion for Improving Mammography Screening Data Classification
DOI:10.1109/TMI.2021.3129068.png)
摘要
En 中文
Machine learning and deep learning methods have become essential for computer-assisted prediction in medicine, with a growing number of applications also in the field of mammography. Typically these algorithms are trained for a specific task, e.g., the classification of lesions or the prediction of a mammogram's pathology status. To obtain a comprehensive view of a patient, models which were all trained for the same task(s) are subsequently ensembled or combined. In this work, we propose a pipeline approach, where we first train a set of individual, task-specific models and subsequently investigate the fusion thereof, which is in contrast to the standard model ensembling strategy. We fuse model predictions and high-level features from deep learning models with hybrid patient models to build stronger predictors on patient level. To this end, we propose a multi-branch deep learning model which efficiently fuses features across different tasks and mammograms to obtain a comprehensive patient-level prediction. We train and evaluate our full pipeline on public mammography data, i.e., DDSM and its curated version CBIS-DDSM, and report an AUC score of 0.962 for predicting the presence of any lesion and 0.791 for predicting the presence of malignant lesions on patient level. Overall, our fusion approaches improve AUC scores significantly by up to 0.04 compared to standard model ensembling. Moreover, by providing not only global patient-level predictions but also task-specific model results that are related to radiological features, our pipeline aims to closely support the reading workflow of radiologists.
Keyword:
Cancer
Feature extraction
Lesions
Task analysis
Breast
Predictive models
Standards
Mammography
DDSM
CBIS-DDSM
deep learning
model fusion
期刊
IF:
9.8
论文数:
6.2K
被引数:
3.7W
机构
暂无机构信息
引用论文
Degradation of reactive dyes I. A comparative study of ozonation, enzymatic and photochemical processes活性染料的降解I.臭氧、酶和光化学过程的比较研究
Chemosphere
IF0
Trp-dependent auxin biosynthesis in Arabidopsis: involvement of cytochrome P450s CYP79B2 and CYP79B3
Multi-task transfer learning deep convolutional neural network: application to computer-aided diagnosis of breast cancer on mammograms多任务迁移学习深度卷积神经网络在乳腺x线照片乳腺癌计算机辅助诊断中的应用
Artificial intelligence for breast cancer detection in mammography and digital breast tomosynthesis: State of the art乳房x线照相术和数字乳房断层合成中用于乳腺癌检测的人工智能: 最新技术
Robust breast cancer detection in mammography and digital breast tomosynthesis using an annotation-efficient deep learning approach使用注释有效的深度学习方法在乳房x线照相术和数字乳房断层合成中进行稳健的乳腺癌检测
NATURE MEDICINE
IF50
Changes in cancer detection and false-positive recall in mammography using artificial intelligence: a retrospective, multireader study使用人工智能的乳房x线照相术中癌症检测和假阳性回忆的变化: 一项回顾性多读者研究
LANCET DIGITAL HEALTH
IF24.1

