对比ViT与CNN在遥感图像语义分割中的表现,发现新损失函数让CNN更优。
Heuristical Comparison of Vision Transformers Against Convolutional Neural Networks for Semantic Segmentation on Remote Sensing Imagery
- 设计加权融合损失函数,同时优化mIoU、Dice分数并降低类别熵。
- 基于iSAID数据集,改进后的CNN在精度上超越了迁移学习的ViT模型。
- 适合关注遥感图像分割性能优化的研究者和工程实践者。
视觉变换器(ViT)近年来在计算机视觉领域引发研究热潮,在图像分类和分割任务中表现突出。以iSAID数据集为例,其前20名基准中有超过80%采用ViT架构或其注意力机制。本文针对遥感航拍图像语义分割,从三个关键方面进行启发式比较:首先,研究一种加权融合损失函数,旨在最大化平均交并比(mIoU)和Dice分数,同时最小化熵或类别表示损失;其次,将Meta的MaskFormer(基于ViT)与通用的UNet(CNN)在mIoU、Dice分数、训练效率和推理时间上进行迁移学习对比;第三,分析二者与当前最先进分割模型的权衡关系。实验表明,新提出的联合加权损失函数显著提升了CNN模型性能,使其超越了基于ViT的迁移学习方案。代码已公开于:https://github.com/ashimdahal/ViT-vs-CNN-Image-Segmentation。
原文摘要 · Abstract (English)
Vision Transformers (ViT) have recently brought a new wave of research in the field of computer vision. These models have performed particularly well in image classification and segmentation. Research on semantic and instance segmentation has accelerated with the introduction of the new architecture, with over 80% of the top 20 benchmarks for the iSAID dataset based on either the ViT architecture or the attention mechanism behind its success. This paper focuses on the heuristic comparison of three key factors of using (or not using) ViT for semantic segmentation of remote sensing aerial images on the iSAID dataset. The experimental results observed during this research were analyzed based on three objectives. First, we studied the use of a weighted fused loss function to maximize the mean Intersection over Union (mIoU) score and Dice score while minimizing entropy or class representation loss. Second, we compared transfer learning on Meta's MaskFormer, a ViT-based semantic segmentation model, against a generic UNet Convolutional Neural Network (CNN) based on mIoU, Dice scores, training efficiency, and inference time. Third, we examined the trade-offs between the two models in comparison to current state-of-the-art segmentation models. We show that the novel combined weighted loss function significantly boosts the CNN model's performance compared to transfer learning with ViT. The code for this implementation can be found at: https://github.com/ashimdahal/ViT-vs-CNN-Image-Segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。