Active Learning for Interactive 3D Image Segmentation
We propose a novel method for applying active learning strategies to interactive 3D image segmentation. Active learning has been recently introduced to the field of image segmentation. However, so far discussions have focused on 2D images only. Here, we frame interactive 3D image segmentation as a classification problem and incorporate active learning in order to alleviate the user from choosing where to provide interactive input. Specifically, we evaluate a given segmentation by constructing an “uncertainty field” over the image domain based on boundary, regional, smoothness and entropy terms. We then calculate and highlight the plane of maximal uncertainty in a batch query step. The user can proceed to guide the labeling of the data on the query plane, hence actively providing additional training data where the classifier has the least confidence. We validate our method against random plane selection showing an average DSC improvement of 10% in the first five plane suggestions (batch queries). Furthermore, our user study shows that our method saves the user 64% of their time, on average.
KeywordsImage Segmentation Hyperspectral Image Query Plane Query Suggestion Active Learning Strategy
- 5.Li, J., Bioucas-Dias, J., Plaza, A.: Supervised hyperspectral image segmentation using active learning. In: 2010 2nd Workshop on Hyperspectral Image and Signal Processing: Evolution in Remote Sensing, pp. 1–4 (2010)Google Scholar
- 9.Pavlopoulou, C., Kak, A.C., Brodley, C.: Application of semi-supervised and active learning to interactive contour delineation. In: ICML 2003 Workshop on the Continuum from Labeled to Unlabeled Data in Machine Learning and Data Mining, pp. 26–33 (2003)Google Scholar
- 11.Settles, B.: Active learning literature survey. Tech. Rep. 1648, University of Wisconsin-Madison (2010)Google Scholar
- 12.Top, A., Hamarneh, G., Abugharbieh, R.: Spotlight: Automated confidence-based user guidance for increasing efficiency in interactive 3D image segmentation. In: Menze, B., Langs, G., Tu, Z., Criminisi, A. (eds.) MICCAI 2010. LNCS, vol. 6533, pp. 204–213. Springer, Heidelberg (2011)CrossRefGoogle Scholar