融合ViT、RoBERTa与CLIP,用双注意力提升图像描述准确性。
Tri-FusionNet: Enhancing Image Description Generation with Transformer-based Fusion Network and Dual Attention Mechanism

- 用双注意力增强ViT编码器,聚焦图像关键区域与语言上下文。
- 在Flickr30k和MS-COCO上,BLEU-4达0.725,CIDEr超1.67,表现优异。
- 适合需要高质量图文生成的场景,如无障碍辅助与多模态应用。
图像描述生成对视觉内容的可访问性与AI理解至关重要。本文提出Tri-FusionNet,一种基于Transformer的图像描述生成模型,融合三种模块:带双注意力机制的视觉变压器(ViT)编码器、鲁棒优化的BERT(RoBERTa)解码器,以及对比语言-图像预训练(CLIP)集成模块。ViT编码器通过双注意力机制强化空间区域与语言上下文的关联,提升特征提取能力;RoBERTa解码器生成精准文本描述;CLIP模块通过对比学习对齐视觉与文本数据,实现模态有效融合。该框架在Flickr30k与Flickr8k数据集上取得0.767–0.456(BLEU)、1.679(CIDEr)、0.478(METEOR)、0.567(ROUGE-L)等指标表现;在MS-COCO上,获得0.893(B-1)、0.821(B-2)、0.794(B-3)、0.725(B-4)的BLEU分数,验证了其生成高质量描述的有效性。
原文摘要 · Abstract (English)
Image description generation is essential for accessibility and AI understanding of visual content. Recent advancements in deep learning have significantly improved natural language processing and computer vision. In this work, we propose Tri-FusionNet, a novel image description generation model that integrates transformer modules: a Vision Transformer (ViT) encoder module with dual-attention mechanism, a Robustly Optimized BERT Approach (RoBERTa) decoder module, and a Contrastive Language-Image Pre-Training (CLIP) integrating module. The ViT encoder, enhanced with dual attention, focuses on relevant spatial regions and linguistic context, improving image feature extraction. The RoBERTa decoder is employed to generate precise textual descriptions. CLIP's integrating module aligns visual and textual data through contrastive learning, ensuring effective combination of both modalities. This fusion of ViT, RoBERTa, and CLIP, along with dual attention, enables the model to produce more accurate, contextually rich, and flexible descriptions. The proposed framework demonstrated competitive performance on the Flickr30k and Flickr8k datasets, with BLEU scores ranging from 0.767 to 0.456 and 0.784 to 0.479, CIDEr scores of 1.679 and 1.483, METEOR scores of 0.478 and 0.358, and ROUGE-L scores of 0.567 and 0.789, respectively. On MS-COCO, the framework obtained BLEU scores of 0.893 (B-1), 0.821 (B-2), 0.794 (B-3), and 0.725 (B-4). The results demonstrate the effectiveness of Tri-FusionNet in generating high-quality image descriptions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。