A Robust Training Method for Pathological Cellular Detector via Spatial Loss Calibration
Hansheng Li,Yuxin Kang,Wentao Yang,Zhuoyue Wu,Xiaoshuang Shi,Feihong Liu,Jianye Liu,Lingyu Hu,Qian Ma,Lei Cui,Jun Feng,Lin Yang
DOI: https://doi.org/10.3389/fmed.2021.767625
IF: 3.9
2021-12-14
Frontiers in Medicine
Abstract:Computer-aided diagnosis of pathological images usually requires detecting and examining all positive cells for accurate diagnosis. However, cellular datasets tend to be sparsely annotated due to the challenge of annotating all the cells. However, training detectors on sparse annotations may be misled by miscalculated losses, limiting the detection performance. Thus, efficient and reliable methods for training cellular detectors on sparse annotations are in higher demand than ever. In this study, we propose a training method that utilizes regression boxes' spatial information to conduct loss calibration to reduce the miscalculated loss. Extensive experimental results show that our method can significantly boost detectors' performance trained on datasets with varying degrees of sparse annotations. Even if 90% of the annotations are missing, the performance of our method is barely affected. Furthermore, we find that the middle layers of the detector are closely related to the generalization performance. More generally, this study could elucidate the link between layers and generalization performance, provide enlightenment for future research, such as designing and applying constraint rules to specific layers according to gradient analysis to achieve “scalpel-level” model training.
medicine, general & internal