SCAKD: a knowledge distillation framework based on spatial-corner attention for infrared and visible image fusion

Jiakun Zhao,Yige Cai
DOI: https://doi.org/10.1007/s00371-024-03356-9
IF: 2.835
2024-05-08
The Visual Computer
Abstract:Infrared and visible image fusion is a kind of image enhancement technology, aiming at combining the characteristics of infrared image and visible image to synthesize them into an image that contains more information and is more suitable for human and computer perception and recognition. Most of the existing fusion methods focus on improving the fusion effect but ignore the model size and computing cost, which is not conducive to the actual deployment of such models on resource-constrained platforms such as mobile terminals. To solve this problem, we propose a new knowledge distillation framework for infrared and visible image fusion. Based on the spatial and corner attention features, the framework uses the middle-layer feature mapping to construct the distillation loss function and guides the lightweight student model to learn the fusion ability of the heavyweight teacher model. After distillation training, the student model can achieve a similar level of integration with the teacher model. A large number of qualitative and quantitative experiments show that our method has better fusion performance and lower model complexity than the existing methods.
computer science, software engineering
What problem does this paper attempt to address?