SMILE: Cost-sensitive multi-task learning for nuclear segmentation and classification with imbalanced annotations

Xipeng Pan,Jijun Cheng,Feihu Hou,Rushi Lan,Cheng Lu,Lingqiao Li,Zhengyun Feng,Huadeng Wang,Changhong Liang,Zhenbing Liu,Xin Chen,Chu Han,Zaiyi Liu
DOI: https://doi.org/10.1016/j.media.2023.102867
IF: 10.9
2023-06-14
Medical Image Analysis
Abstract:High throughput nuclear segmentation and classification of whole slide images (WSIs) is crucial to biological analysis, clinical diagnosis and precision medicine. With the advances of CNN algorithms and the continuously growing datasets, considerable progress has been made in nuclear segmentation and classification. However, few works consider how to reasonably deal with nuclear heterogeneity in the following two aspects: imbalanced data distribution and diversified morphology characteristics. The minority classes might be dominated by the majority classes due to the imbalanced data distribution and the diversified morphology characteristics may lead to fragile segmentation results. In this study, a cost- S ensitive M ult I -task LE arning ( SMILE ) framework is conducted to tackle the data heterogeneity problem. Based on the most popular multi-task learning backbone in nuclei segmentation and classification, we propose a multi-task correlation attention (MTCA) to perform feature interaction of multiple high relevant tasks to learn better feature representation. A cost-sensitive learning strategy is proposed to solve the imbalanced data distribution by increasing the penalization for the error classification of the minority classes. Furthermore, we propose a novel post-processing step based on the coarse-to-fine marker-controlled watershed scheme to alleviate fragile segmentation when nuclei are with large size and unclear contour. Extensive experiments show that the proposed method achieves state-of-the-art performances on CoNSeP and MoNuSAC 2020 datasets. The code will be available at: https://github.com/panxipeng/nuclear_segandcls .
engineering, biomedical,computer science, interdisciplinary applications, artificial intelligence,radiology, nuclear medicine & medical imaging
What problem does this paper attempt to address?