Online Discriminative Multi-atlas Learning for Isointense Infant Brain Segmentation

Abstract

Multi-atlas labeling in a non-local patch manner has emerged as an important approach to alleviate both the possible misalignment and mis-match among patches for guiding accurate image segmentation. However, the relationship among candidate patches and their intra/inter-class variability are less investigated, which limits the discriminative power of these patches. To address these issues, we present a new online discriminative multi-atlas learning method for labeling the target patch by the best representative candidates in a sparse sense. Specifically, the online multi-kernel learning is firstly adopted to map the patches into a cascade of discriminative kernel spaces for producing corresponding probability maps to model a label of each sample in these spaces. Then the online discriminative dictionary learning is proposed to build the atlas that handles the intra-class compactness and inter-class separability simultaneously. Finally, sparse coding is used to select patches in the dictionary for label propagation. In this way, the multi-atlas information dynamically learned with the context probability maps is iteratively incorporated to build the atlas dictionary, for gradually excluding the misleading candidate patches. The proposed method is validated by experiments on isointense infant brain tissue segmentation, and achieves promising results in comparison with several different labeling strategies.