无人系统技术2024,Vol.7Issue(2) :73-81.DOI:10.19942/j.issn.2096-5915.2024.02.18

基于SAM的无人艇多源图像分割研究

Research on Multi-sensor Image Segmentation of USV Based on SAM

杨博 谢杨柳 张天航 杨远鹏 李祥生 马向峰 韩玮
无人系统技术2024,Vol.7Issue(2) :73-81.DOI:10.19942/j.issn.2096-5915.2024.02.18

基于SAM的无人艇多源图像分割研究

Research on Multi-sensor Image Segmentation of USV Based on SAM

杨博 1谢杨柳 1张天航 1杨远鹏 1李祥生 1马向峰 1韩玮1
扫码查看

作者信息

  • 1. 中国船舶集团有限公司系统工程研究院,北京 100094
  • 折叠

摘要

针对无人艇多源传感器的图像分割问题,开展了结合图像分割大模型与大模型低成本微调技术的研究,以增强感知大模型在无人系统上的实用性.首先将分割大模型应用于无人艇可见光、红外和声学图像分割,并对其性能进行了比较.经对比发现,分割大模型对可见光和红外图像分割效果好,但对声学图像的分割性能存在严重下降.随后利用低秩自适应算法对分割大模型编码器部分进行了微调训练,以适应声学图像分割任务.微调后的分割大模型对声学图像的分割性能得到显著提升,Dice相似系数平均值相比初始分割大模型提高了52.88%,证明了方法的有效性.作为一种大模型的微调技术,低秩自适应算法使用低秩近似来降低权重矩阵维度,大幅减少模型中可训练的参数量,在提升训练效率的同时降低了训练成本.

Abstract

A study was conducted on image segmentation obtained by different types of sensors on Unmanned Surface Vessel(USV),combining the image segmentation large model with low-cost fine-tuning techniques to enhance the practicality of large models in unmanned systems.This article first applies the Segment Anything Model(SAM)to the segmentation tasks of visible light,infrared,and sonar images of USV.The comparison results show that SAM has good segmentation performances in visible light and infrared images,but there is a serious decline in sonar image segmentation performance.Subsequently,this article used Low-rank Adaptation(LoRA)algorithm to fine tune the SAM encoder to adapt to sonar image segmentation tasks.The segmentation performance of the fine tuned SAM on sonar images has been significantly improved,with Dice improving by 52.88 compared to the initial SAM,proving the effectiveness of the method.Compared to the scheme of retraining all parameters of the large model,LoRA uses low-rank approximation to reduce the dimensionality of the weight matrix,thereby significantly reducing the number of trainable parameters in the model.It performs better in terms of training efficiency and cost reduction.

关键词

无人艇/多源/传感器/图像分割/分割大模型/低秩自适应算法

Key words

Unmanned Surface Vessel(USV)/Multi-source/Sensor/Image Segmentation/Seg-ment Anything Model(SAM)/Low-Rank Adaptation(LoRA)

引用本文复制引用

出版年

2024
无人系统技术

无人系统技术

ISSN:
参考文献量23
段落导航相关论文