Learning Cross-Modality Representations from Multi-Modal Images

Research output: Contribution to journalJournal articleResearchpeer-review

Documents

Machine learning algorithms can have difficulties adapting to data from different sources, for example from different imaging modalities. We present and analyze three techniques for unsupervised cross-modality feature learning, using a shared autoencoder-like convolutional network that learns a common representation from multi-modal data. We investigate a form of feature normalization, a learning objective that minimizes crossmodality differences, and modality dropout, in which the network is trained with varying subsets of modalities. We measure the same-modality and cross-modality classification accuracies and explore whether the models learn modality-specific or shared features. This paper presents experiments on two public datasets, with knee images from two MRI modalities, provided by the Osteoarthritis Initiative, and brain tumor segmentation on four MRI modalities from the BRATS challenge. All three approaches improved the cross-modality classification accuracy, with modality dropout and per-feature normalization giving the largest improvement. We observed that the networks tend to learn a combination of cross-modality and modality-specific features. Overall, a combination of all three methods produced the most cross-modality features and the highest cross-modality classification accuracy, while maintaining most of the samemodality accuracy.

Original languageEnglish
JournalIEEE Transactions on Medical Imaging
Volume38
Issue number2
Pages (from-to)638-648
ISSN0278-0062
DOIs
Publication statusPublished - 2019

    Research areas

  • Autoencoders, Deep learning, Representation learning, Transfer learning

Number of downloads are based on statistics from Google Scholar and www.ku.dk


No data available

ID: 203054362