arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2204.09323cs.CVcs.LG
收藏

声纳图像分类的自监督学习

Self-supervised Learning for Sonar Image Classification

German Research Center for Artificial Intelligence(德国人工智能研究中心);Bonn-Rhein-Sieg University of Applied Sciences(波恩-莱茵-锡格应用科学大学);University of Groningen(格罗宁根大学)

Alan Preciado-Grijalva, Bilal Wehbe, Miguel Bande Firvida, Matias Valdenegro-Toro

摘要:

自监督学习已被证明是一种强大的方法来学习图像表示,而不需要大型标记数据集。对于水下机器人来说,设计计算机视觉算法来提高感知能力,如声纳图像分类,是一个很有意义的课题。由于声纳成像的保密性和声纳图像判读的困难性,创建公共的大型标记声纳数据集来训练监督学习算法是具有挑战性的。在这项工作中,我们研究了三种自监督学习方法(RotNet,Denoising Autoencoders和Jigsaw)在不需要人类标签的情况下学习高质量声纳图像表示的潜力。我们提出了预训练和迁移学习的结果在现实生活中的声纳图像数据集。我们的研究结果表明,在所有三种方法中,自监督预训练产生的分类性能与Few-Shot迁移学习设置中的监督预训练相当。代码和自我监督的预训练模型可在https://github.com/agrija9/ssl-sonar-images上获得

英文摘要:

Self-supervised learning has proved to be a powerful approach to learn image representations without the need of large labeled datasets. For underwater robotics, it is of great interest to design computer vision algorithms to improve perception capabilities such as sonar image classification. Due to the confidential nature of sonar imaging and the difficulty to interpret sonar images, it is challenging to create public large labeled sonar datasets to train supervised learning algorithms. In this work, we investigate the potential of three self-supervised learning methods (RotNet, Denoising Autoencoders, and Jigsaw) to learn high-quality sonar image representation without the need of human labels. We present pre-training and transfer learning results on real-life sonar image datasets. Our results indicate that self-supervised pre-training yields classification performance comparable to supervised pre-training in a few-shot transfer learning setup across all three methods. Code and self-supervised pre-trained models are be available at https://github.com/agrija9/ssl-sonar-images

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

↑