Transporting Labels via Hierarchical Optimal Transport for Semi-Supervised Learning
Abstract
Semi-Supervised Learning (SSL) based on Convolutional Neural Networks (CNNs) have recently been proven as powerful tools for standard tasks such as image classification when there is not a sufficient amount of labeled data available during the training. In this work, we consider the general setting of the SSL problem for image classification, where the labeled and unlabeled data come from the same underlying distribution. We propose a new SSL method that adopts a hierarchical Optimal Transport (OT) technique to find a mapping from empirical unlabeled measures to corresponding labeled measures by leveraging the minimum amount of transportation cost in the label space. Based on this mapping, pseudo-labels for the unlabeled data are inferred, which are then used along with the labeled data for training the CNN. We evaluated and compared our method with state-of-the-art SSL approaches on standard datasets to demonstrate the superiority of our SSL method.
Cite
Text
Taherkhani et al. "Transporting Labels via Hierarchical Optimal Transport for Semi-Supervised Learning." Proceedings of the European Conference on Computer Vision (ECCV), 2020. doi:10.1007/978-3-030-58548-8_30Markdown
[Taherkhani et al. "Transporting Labels via Hierarchical Optimal Transport for Semi-Supervised Learning." Proceedings of the European Conference on Computer Vision (ECCV), 2020.](https://mlanthology.org/eccv/2020/taherkhani2020eccv-transporting/) doi:10.1007/978-3-030-58548-8_30BibTeX
@inproceedings{taherkhani2020eccv-transporting,
title = {{Transporting Labels via Hierarchical Optimal Transport for Semi-Supervised Learning}},
author = {Taherkhani, Fariborz and Dabouei, Ali and Soleymani, Sobhan and Dawson, Jeremy and Nasrabadi, Nasser M.},
booktitle = {Proceedings of the European Conference on Computer Vision (ECCV)},
year = {2020},
doi = {10.1007/978-3-030-58548-8_30},
url = {https://mlanthology.org/eccv/2020/taherkhani2020eccv-transporting/}
}