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

InfMAE:红外模态的基础模型

InfMAE: A Foundation Model in the Infrared Modality

School of Communications and Information Engineering, Chongqing University of Posts and Telecommunications(重庆邮电大学通信与信息工程学院);School of Intelligent Systems Engineering, Sun Yat-sen University(中山大学智能系统工程学院);School of Mathematics and Statistics, Xi’an Jiaotong University(西安交通大学数学与统计学院);School of Computer and Information Technology, Shanxi University(山西大学计算机与信息技术学院)

Fangcen Liu, Chenqiang Gao, Yaming Zhang, Junjie Guo, Jinhao Wang, Deyu Meng

AI总结:

针对红外基础模型设计难题,研究提出红外模态基础模型InfMAE,配套发布Inf30数据集,设计信息感知掩码策略、多尺度编码器及红外解码器,在三项下游任务上性能超越现有监督与自监督方法。

摘要:

近年来,基础模型席卷了计算机视觉领域,推动了不同模态下各类任务的发展。然而,如何设计红外基础模型仍是一个悬而未决的问题。本文提出了红外模态基础模型InfMAE。我们发布了一个名为Inf30的红外数据集,以解决红外视觉领域缺乏自监督学习所需大规模数据的问题。此外,我们设计了一种适用于红外图像的信息感知掩码策略。该掩码策略能在自监督学习过程中让模型更关注红外图像中信息更丰富的区域,有利于学习通用表征。另外,我们采用多尺度编码器来提升预训练编码器在下游任务中的性能。最后,基于红外图像不具备大量细节和纹理信息这一特点,我们设计了红外解码器模块,进一步提升了下游任务性能。大量实验表明,我们提出的InfMAE方法在三项下游任务中表现优于其他有监督方法和自监督学习方法。

英文摘要:

In recent years, the foundation models have swept the computer vision field and facilitated the development of various tasks within different modalities. However, it remains an open question on how to design an infrared foundation model. In this paper, we propose InfMAE, a foundation model in infrared modality. We release an infrared dataset, called Inf30 to address the problem of lacking large-scale data for self-supervised learning in the infrared vision community. Besides, we design an information-aware masking strategy, which is suitable for infrared images. This masking strategy allows for a greater emphasis on the regions with richer information in infrared images during the self-supervised learning process, which is conducive to learning the generalized representation. In addition, we adopt a multi-scale encoder to enhance the performance of the pre-trained encoders in downstream tasks. Finally, based on the fact that infrared images do not have a lot of details and texture information, we design an infrared decoder module, which further improves the performance of downstream tasks. Extensive experiments show that our proposed method InfMAE outperforms other supervised methods and self-supervised learning methods in three downstream tasks.

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

↑