Multi-task learning in mammography has gained attention in recent years as a strategy to improve multi-label breast cancer and density classifications by leveraging shared representations. While previous studies have suggested that combining cancer and density labels in a multi-task learning framework can be beneficial, empirical evidence supporting these claims over single-task baselines remains limited. As breast density is a crucial factor for breast cancer risk but is less explored compared to cancer classification, this study focuses on density classification within the multi-task learning framework. Our findings indicate that single-task models can be beneficial over multi-task models, challenging the assumption that multi-task learning inherently provides performance benefits. Furthermore, we explore various sampling strategies in the multi-task setting to address class imbalance, a prevalent issue in mammography datasets, but find that none surpass the single-task baseline. Our findings suggest that the additional complexity of multi-task learning does not yield substantial benefits for density classification, highlighting the need for further research into alternative strategies for improving mammography classification performance.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Assessing the Efficacy of Multi-task Learning in Mammographic Density Classification: A Study on Class Imbalance and Model Performance

  • Suaiba A. Salahuddin,
  • Elisabeth Wetzer,
  • Kristoffer Wickstrøm,
  • Solveig Thrun,
  • Michael Kampffmeyer,
  • Robert Jenssen

摘要

Multi-task learning in mammography has gained attention in recent years as a strategy to improve multi-label breast cancer and density classifications by leveraging shared representations. While previous studies have suggested that combining cancer and density labels in a multi-task learning framework can be beneficial, empirical evidence supporting these claims over single-task baselines remains limited. As breast density is a crucial factor for breast cancer risk but is less explored compared to cancer classification, this study focuses on density classification within the multi-task learning framework. Our findings indicate that single-task models can be beneficial over multi-task models, challenging the assumption that multi-task learning inherently provides performance benefits. Furthermore, we explore various sampling strategies in the multi-task setting to address class imbalance, a prevalent issue in mammography datasets, but find that none surpass the single-task baseline. Our findings suggest that the additional complexity of multi-task learning does not yield substantial benefits for density classification, highlighting the need for further research into alternative strategies for improving mammography classification performance.