S 3 R: Shape and Semantics-based Selective Regularization for Explainable Continual Segmentation across Multiple Sites
Jingyang Zhang,Ran Gu,Peng Xue,Mianxin Liu,Hao Zheng,Yefeng Zheng,Lei Ma,Guotai Wang,Lixu Gu
DOI: https://doi.org/10.1109/tmi.2023.3260974
IF: 10.6
2023-01-01
IEEE Transactions on Medical Imaging
Abstract:In clinical practice, it is desirable for medical image segmentation models to be able to continually learn on a sequential data stream from multiple sites, rather than a consolidated dataset, due to storage cost and privacy restrictions. However, when learning on a new site, existing methods struggle with a weak memorizability for previous sites with complex shape and semantic information, and a poor explainability for the memory consolidation process. In this work, we propose a novel Shape and Semantics-based Selective Regularization (S<sup>3</sup>R) method for explainable cross-site continual segmentation to maintain both shape and semantic knowledge of previously learned sites. Specifically, S<sup>3</sup>R method adopts a selective regularization scheme to penalize changes of parameters with high Joint Shape and Semantics-based Importance (JSSI) weights, which are estimated based on the parameter sensitivity to shape properties and reliable semantics of the segmentation object. This helps to prevent the related shape and semantic knowledge from being forgotten. Moreover, we propose an Importance Activation Mapping (IAM) method for memory interpretation, which indicates the spatial support for important parameters to visualize the memorized content. We have extensively evaluated our method on prostate segmentation and optic cup and disc segmentation tasks. Our method outperforms other comparison methods in reducing model forgetting and increasing explainability. Our code is available at https://github.com/jingyzhang/S3R.
engineering, biomedical,imaging science & photographic technology, electrical & electronic,computer science, interdisciplinary applications,radiology, nuclear medicine & medical imaging