Unsupervised Learning for Robust Fitting: A Reinforcement Learning Approach

Giang Truong, Huu Le, David Suter, Erchuan Zhang, Syed Zulqarnain Gilani; Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2021, pp. 10348-10357

Abstract


Robust model fitting is a core algorithm in a large number of computer vision applications. Solving this problem efficiently for highly contaminated datasets is, however, still challenging due to its underlying computational complexity. Recent attention has been focused on learning-based algorithms. However, most approaches are supervised (which require a large amount of labelled training data). In this paper, we introduce a novel unsupervised learning framework that learns to directly solve robust model fitting. Unlike other methods, our work is agnostic to the underlying input features, and can be easily generalized to a wide variety of LP-type problems with quasi-convex residuals. We empirically show that our method outperforms existing unsupervised learning approaches, and achieves competitive results compared to traditional methods on several important computer vision problems.

Related Material


[pdf] [supp] [arXiv]
[bibtex]
@InProceedings{Truong_2021_CVPR, author = {Truong, Giang and Le, Huu and Suter, David and Zhang, Erchuan and Gilani, Syed Zulqarnain}, title = {Unsupervised Learning for Robust Fitting: A Reinforcement Learning Approach}, booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)}, month = {June}, year = {2021}, pages = {10348-10357} }