Entity-Guided Multi-Task Learning for Infrared and Visible Image Fusion

TL;DR

This paper introduces EGMT, a novel multi-task learning framework that leverages entity-level textual information and cross-modal interactions to significantly improve infrared and visible image fusion quality.

Contribution

The paper proposes a new entity-guided multi-task learning approach that extracts semantic-rich textual information and enhances feature interaction for better image fusion results.

Findings

EGMT outperforms state-of-the-art methods in preserving salient targets.

It improves texture detail and semantic consistency in fused images.

The approach effectively utilizes entity-level textual supervision.

Abstract

Existing text-driven infrared and visible image fusion approaches often rely on textual information at the sentence level, which can lead to semantic noise from redundant text and fail to fully exploit the deeper semantic value of textual information. To address these issues, we propose a novel fusion approach named Entity-Guided Multi-Task learning for infrared and visible image fusion (EGMT). Our approach includes three key innovative components: (i) A principled method is proposed to extract entity-level textual information from image captions generated by large vision-language models, eliminating semantic noise from raw text while preserving critical semantic information; (ii) A parallel multi-task learning architecture is constructed, which integrates image fusion with a multi-label classification task. By using entities as pseudo-labels, the multi-label classification task provides semantic supervision, enabling the model to achieve a deeper understanding of image content and significantly improving the quality and semantic density of the fused image; (iii) An entity-guided cross-modal interactive module is also developed to facilitate the fine-grained interaction between visual and entity-level textual features, which enhances feature representation by capturing cross-modal dependencies at both inter-visual and visual-entity levels. To promote the wide application of the entity-guided image fusion framework, we release the entity-annotated version of four public datasets (i.e., TNO, RoadScene, M3FD, and MSRS). Extensive experiments demonstrate that EGMT achieves superior performance in preserving salient targets, texture details, and semantic consistency, compared to the state-of-the-art methods. The code and dataset will be publicly available at https://github.com/wyshao-01/EGMT.