CVApr 26, 2024

UniRGB-IR: A Unified Framework for Visible-Infrared Semantic Tasks via Adapter Tuning

arXiv:2404.17360v425 citationsh-index: 6MM
Originality Incremental advance
AI Analysis

This work addresses scalability and generalization issues in RGB-IR semantic analysis for applications like low-illumination and adverse weather conditions, representing an incremental improvement over existing methods.

The paper tackles the problem of poor scalability and limited generalization in visible-infrared (RGB-IR) semantic tasks by proposing UniRGB-IR, a unified framework that uses adapter tuning to incorporate multi-modal features into pre-trained RGB models, achieving state-of-the-art performance across various tasks.

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.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes