从压缩视角重设计分割解码器,提升精度与鲁棒性
Rethinking Decoders for Transformer-based Semantic Segmentation: A Compression Perspective

- 将解码器视为主成分分析,用自注意力构建理想特征子空间
- 交叉注意力实现低秩近似,生成对应类别的正交基
- 点积输出紧凑分割掩码,模型轻量且更鲁棒
基于Transformer的语义分割方法通常采用通过交叉注意力从图像嵌入中提取额外嵌入、利用自注意力精炼图像嵌入与额外嵌入,并通过点积将图像嵌入投影到额外嵌入的解码器。尽管表现优异,但这些设计缺乏理论解释,限制了系统性改进。本文提出从压缩视角重新思考分割问题,发现解码器与主成分分析(PCA)存在根本联系。由此提出白盒全注意力解码器DEPICT:1)自注意力将图像嵌入精炼为对齐监督信号且保留大部分信息的理想主子空间;2)交叉注意力寻找精炼后嵌入的低秩近似,得到主子空间的一组正交基,对应预定义类别;3)点积操作生成图像嵌入的紧凑表示作为分割掩码。在ADE20K数据集上的实验表明,DEPICT持续优于黑箱解码器Segmenter,且模型轻量、更鲁棒。
原文摘要 · Abstract (English)
State-of-the-art methods for Transformer-based semantic segmentation typically adopt Transformer decoders that are used to extract additional embeddings from image embeddings via cross-attention, refine either or both types of embeddings via self-attention, and project image embeddings onto the additional embeddings via dot-product. Despite their remarkable success, these empirical designs still lack theoretical justifications or interpretations, thus hindering potentially principled improvements. In this paper, we argue that there are fundamental connections between semantic segmentation and compression, especially between the Transformer decoders and Principal Component Analysis (PCA). From such a perspective, we derive a white-box, fully attentional DEcoder for PrIncipled semantiC segemenTation (DEPICT), with the interpretations as follows: 1) the self-attention operator refines image embeddings to construct an ideal principal subspace that aligns with the supervision and retains most information; 2) the cross-attention operator seeks to find a low-rank approximation of the refined image embeddings, which is expected to be a set of orthonormal bases of the principal subspace and corresponds to the predefined classes; 3) the dot-product operation yields compact representation for image embeddings as segmentation masks. Experiments conducted on dataset ADE20K find that DEPICT consistently outperforms its black-box counterpart, Segmenter, and it is light weight and more robust.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。