UniRGB-IR: A Unified Framework for Visible-Infrared Semantic Tasks via Adapter Tuning
arXiv:2404.17360 · doi:10.1145/3746027.3754806
Abstract
Semantic analysis on visible (RGB) and infrared (IR) images has gained significant attention due to their enhanced accuracy and robustness under challenging conditions including low-illumination and adverse weather. However, due to the lack of pre-trained foundation models on the large-scale infrared image datasets, existing methods prefer to design task-specific frameworks and directly fine-tune them with pre-trained foundation models on their RGB-IR semantic relevance datasets, which results in poor scalability and limited generalization. To address these limitations, we propose UniRGB-IR, a scalable and efficient framework for RGB-IR semantic tasks that introduces a novel adapter mechanism to effectively incorporate rich multi-modal features into pre-trained RGB-based foundation models. Our framework comprises three key components: a vision transformer (ViT) foundation model, a Multi-modal Feature Pool (MFP) module, and a Supplementary Feature Injector (SFI) module. The MFP and SFI modules cooperate with each other as an adpater to effectively complement the ViT features with the contextual multi-scale features. During training process, we freeze the entire foundation model to inherit prior knowledge and only optimize the MFP and SFI modules. Furthermore, to verify the effectiveness of our framework, we utilize the ViT-Base as the pre-trained foundation model to perform extensive experiments. Experimental results on various RGB-IR semantic tasks demonstrate that our method can achieve state-of-the-art performance.
10 pages, 6 figures, Accepted by ACM MM 2025
References in corpus (7)
- SSD: Single Shot MultiBox Detector
- UIU-Net: U-Net in U-Net for Infrared Small Object Detection
- SwinNet: Swin Transformer drives edge-aware RGB-D and RGB-T salient object detection
- Multi-interactive Dual-decoder for RGB-thermal Salient Object Detection
- CAVER: Cross-Modal View-Mixed Transformer for Bi-Modal Salient Object Detection
- Weakly Aligned Feature Fusion for Multimodal Object Detection
- Parameter-Inverted Image Pyramid Networks for Visual Perception and Multimodal Understanding