arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1709.01591cs.CV

Improving Landmark Localization with Semi-Supervised Learning

MILA-University of Montreal(蒙特利尔大学MILA研究所);NVIDIA(英伟达);Ecole Polytechnique of Montreal(蒙特利尔高等理工学院);CIFAR(加拿大高等研究院);Facebook AI Research(脸书人工智能研究院)

Sina Honari, Pavlo Molchanov, Stephen Tyree, Pascal Vincent, Christopher Pal, Jan Kautz

英文摘要:

We present two techniques to improve landmark localization in images from partially annotated datasets. Our primary goal is to leverage the common situation where precise landmark locations are only provided for a small data subset, but where class labels for classification or regression tasks related to the landmarks are more abundantly available. First, we propose the framework of sequential multitasking and explore it here through an architecture for landmark localization where training with class labels acts as an auxiliary signal to guide the landmark localization on unlabeled data. A key aspect of our approach is that errors can be backpropagated through a complete landmark localization model. Second, we propose and explore an unsupervised learning technique for landmark localization based on having a model predict equivariant landmarks with respect to transformations applied to the image. We show that these techniques, improve landmark prediction considerably and can learn effective detectors even when only a small fraction of the dataset has landmark labels. We present results on two toy datasets and four real datasets, with hands and faces, and report new state-of-the-art on two datasets in the wild, e.g. with only 5\% of labeled images we outperform previous state-of-the-art trained on the AFLW dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑