arXiv · 1810.09369
Brain Tumor Image Retrieval via Multitask Learning
Abstract
Classification-based image retrieval systems are built by training convolutional neural networks (CNNs) on a relevant classification problem and using the distance in the resulting feature space as a similarity metric. However, in practical applications, it is often desirable to have representations which take into account several aspects of the data (e.g., brain tumor type and its localization). In our work, we extend the classification-based approach with multitask learning: we train a CNN on brain MRI scans with heterogeneous labels and implement a corresponding tumor image retrieval system. We validate our approach on brain tumor data which contains information about tumor types, shapes and localization. We show that our method allows us to build representations that contain more relevant information about tumors than single-task classification-based approaches.
Explore related subjects
Keep this discovery
Maxim Pisov, Gleb Makarchuk, Valery Kostjuchenko, Alexandra Dalechina, Andrey Golanov, Mikhail Belyaev. 2018-10-22. Brain Tumor Image Retrieval via Multitask Learning. https://arxiv.org/abs/1810.09369
Cite the original work for its findings. Save a collection to share your selection of sources.