Transfer learning improves supervised image segmentation across imaging protocols
The variation between images obtained with different scanners or different imaging protocols presents a major challenge in automatic segmentation of biomedical images. This variation especially hampers the application of otherwise successful supervised-learning techniques which, in order to perform well, often require a large amount of labeled training data that is exactly representative of the target data. We therefore propose to use transfer learning for image segmentation. Transfer-learning techniques can cope with differences in distributions between training and target data, and therefore may improve performance over supervised learning for segmentation across scanners and scan protocols. We present four transfer classifiers that can train a classification scheme with only a small amount of representative training data, in addition to a larger amount of other training data with slightly different characteristics. The performance of the four transfer classifiers was compared to that of standard supervised classification on two magnetic resonance imaging brain-segmentation tasks with multi-site data: white matter, gray matter, and cerebrospinal fluid segmentation; and white-matter-/MS-lesion segmentation. The experiments showed that when there is only a small amount of representative training data available, transfer learning can greatly outperform common supervised-learning approaches, minimizing classification errors by up to 60%.
|Keywords||Image Segmentation, machine learning, magnetic resonance imaging, pattern recognition, Transfer learning|
|Persistent URL||dx.doi.org/10.1109/TMI.2014.2366792, hdl.handle.net/1765/90076|
|Journal||IEEE Transactions on Medical Imaging|
van Opbroek, A, Ikram, M.A, Vernooij, M.W, & de Bruijne, M. (2015). Transfer learning improves supervised image segmentation across imaging protocols. IEEE Transactions on Medical Imaging, 34(5), 1018–1030. doi:10.1109/TMI.2014.2366792