Paper Reading AI Learner

A Novel Lightweight Transformer with Edge-Aware Fusion for Remote Sensing Image Captioning

2025-06-11 06:24:02
Swadhin Das, Divyansh Mundra, Priyanshu Dayal, Raksha Sharma

Abstract

Transformer-based models have achieved strong performance in remote sensing image captioning by capturing long-range dependencies and contextual information. However, their practical deployment is hindered by high computational costs, especially in multi-modal frameworks that employ separate transformer-based encoders and decoders. In addition, existing remote sensing image captioning models primarily focus on high-level semantic extraction while often overlooking fine-grained structural features such as edges, contours, and object boundaries. To address these challenges, a lightweight transformer architecture is proposed by reducing the dimensionality of the encoder layers and employing a distilled version of GPT-2 as the decoder. A knowledge distillation strategy is used to transfer knowledge from a more complex teacher model to improve the performance of the lightweight network. Furthermore, an edge-aware enhancement strategy is incorporated to enhance image representation and object boundary understanding, enabling the model to capture fine-grained spatial details in remote sensing images. Experimental results demonstrate that the proposed approach significantly improves caption quality compared to state-of-the-art methods.

Abstract (translated)

基于Transformer的模型通过捕捉长距离依赖和上下文信息,在遥感图像描述生成中取得了优异的成绩。然而,由于计算成本高,尤其是在使用独立的Transformer编码器和解码器的多模态框架中,这些模型的实际部署受到了阻碍。此外,现有的遥感图像描述生成模型主要关注高层次语义提取,而往往忽视了边缘、轮廓和对象边界等细粒度结构特征。 为了解决这些问题,提出了一种轻量级的Transformer架构,通过减少编码器层的维度并使用GPT-2的精简版本作为解码器来实现。采用知识蒸馏策略,将更复杂的教师模型的知识转移到轻量级网络中以提高性能。此外,还引入了边缘感知增强策略,用以提升图像表示和对象边界理解能力,使模型能够捕捉到遥感图像中的细粒度空间细节。 实验结果表明,所提出的方法在图像描述质量方面显著优于现有的先进方法。

URL

https://arxiv.org/abs/2506.09429

PDF

https://arxiv.org/pdf/2506.09429.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot