Transfer learning of medical image segmentation based on optimal transport feature selection

ABSTRACT

In the unsupervised domain adaptive transfer learning process，domain-independent features lead to the degradation of model segmentation performance，but there is no effective feature selection method for transfer learning segmentation model at present. To solve this problem，a general feature selection module for transfer learning was proposed based on optimal transport，which can be applied to various unsupervised domain adaptive image segmentation models. In this module，the optimal sample subsets of two domains are selected by weighted optimal transport of segmentation accuracy，and then the features of sample subsets are subjected to entropy regularized optimal transport，so as to obtain a descending list of similarity between two domains to remove domain-independent features. The universal feature selection module is applied to three unsupervised domain adaptive models to solve the problem of Covid-19 image segmentation，which improves the model performance to a certain extent. © 2022 Editorial Board of Jilin University. All rights reserved.