MMDRFuse: Distilled Mini-Model with Dynamic Refresh for Multi-Modality Image Fusion

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Deng, Yanglin, Xu, Tianyang, Cheng, Chunyang, Wu, Xiao-Jun, Kittler, Josef
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916373277442048
author Deng, Yanglin
Xu, Tianyang
Cheng, Chunyang
Wu, Xiao-Jun
Kittler, Josef
author_facet Deng, Yanglin
Xu, Tianyang
Cheng, Chunyang
Wu, Xiao-Jun
Kittler, Josef
contents In recent years, Multi-Modality Image Fusion (MMIF) has been applied to many fields, which has attracted many scholars to endeavour to improve the fusion performance. However, the prevailing focus has predominantly been on the architecture design, rather than the training strategies. As a low-level vision task, image fusion is supposed to quickly deliver output images for observation and supporting downstream tasks. Thus, superfluous computational and storage overheads should be avoided. In this work, a lightweight Distilled Mini-Model with a Dynamic Refresh strategy (MMDRFuse) is proposed to achieve this objective. To pursue model parsimony, an extremely small convolutional network with a total of 113 trainable parameters (0.44 KB) is obtained by three carefully designed supervisions. First, digestible distillation is constructed by emphasising external spatial feature consistency, delivering soft supervision with balanced details and saliency for the target network. Second, we develop a comprehensive loss to balance the pixel, gradient, and perception clues from the source images. Third, an innovative dynamic refresh training strategy is used to collaborate history parameters and current supervision during training, together with an adaptive adjust function to optimise the fusion network. Extensive experiments on several public datasets demonstrate that our method exhibits promising advantages in terms of model efficiency and complexity, with superior performance in multiple image fusion tasks and downstream pedestrian detection application. The code of this work is publicly available at https://github.com/yanglinDeng/MMDRFuse.
format Preprint
id arxiv_https___arxiv_org_abs_2408_15641
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MMDRFuse: Distilled Mini-Model with Dynamic Refresh for Multi-Modality Image Fusion
Deng, Yanglin
Xu, Tianyang
Cheng, Chunyang
Wu, Xiao-Jun
Kittler, Josef
Computer Vision and Pattern Recognition
In recent years, Multi-Modality Image Fusion (MMIF) has been applied to many fields, which has attracted many scholars to endeavour to improve the fusion performance. However, the prevailing focus has predominantly been on the architecture design, rather than the training strategies. As a low-level vision task, image fusion is supposed to quickly deliver output images for observation and supporting downstream tasks. Thus, superfluous computational and storage overheads should be avoided. In this work, a lightweight Distilled Mini-Model with a Dynamic Refresh strategy (MMDRFuse) is proposed to achieve this objective. To pursue model parsimony, an extremely small convolutional network with a total of 113 trainable parameters (0.44 KB) is obtained by three carefully designed supervisions. First, digestible distillation is constructed by emphasising external spatial feature consistency, delivering soft supervision with balanced details and saliency for the target network. Second, we develop a comprehensive loss to balance the pixel, gradient, and perception clues from the source images. Third, an innovative dynamic refresh training strategy is used to collaborate history parameters and current supervision during training, together with an adaptive adjust function to optimise the fusion network. Extensive experiments on several public datasets demonstrate that our method exhibits promising advantages in terms of model efficiency and complexity, with superior performance in multiple image fusion tasks and downstream pedestrian detection application. The code of this work is publicly available at https://github.com/yanglinDeng/MMDRFuse.
title MMDRFuse: Distilled Mini-Model with Dynamic Refresh for Multi-Modality Image Fusion
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.15641