Ambiguity-aware breast tumor cellularity estimation via self-ensemble label distribution learning.

Abstract

In this work, we address the task of tumor cellularity (TC) estimation with a novel framework based on the label distribution learning (LDL) paradigm. We propose a self-ensemble label distribution learning framework (SLDL) to resolve the challenges of existing LDL-based methods, including difficulties for inter-rater ambiguity exploitation, proper and flexible label distribution generation, and accurate TC value recovery. The proposed SLDL makes four main contributions which have been demonstrated to be quite effective in numerous experiments. First, we propose an expertness-aware conditional VAE for diversified single-rater modeling and an attention-based multi-rater fusion strategy that enables effective inter-rater ambiguity exploitation. Second, we propose a template-based label distribution generation method that is tailored for the TC estimation task and constructs label distributions based on the annotation priors. Third, we propose a novel restricted distribution loss, significantly improving the TC value estimation by effectively regularizing the learning with unimodal loss and regression loss. Fourth, to the best of our knowledge, we are the first to simultaneously leverage inter-rater and intra-rater variability to address the label ambiguity issue in the breast tumor cellularity estimation tasks. The experimental results on the public BreastPathQ dataset demonstrate that the SLDL outperforms the existing methods by a large margin and achieves new state-of-the-art results in the TC estimation task. The code will be available from https://github.com/PerceptionComputingLab/ULTRA.