视觉 Transformer 把图像块序列交给自注意力
说明图像分块、位置编码和自注意力怎样形成视觉表示,并分析数据规模、分辨率与计算成本的约束。
视觉 Transformer 把图像块序列交给自注意力
视觉 Transformer 将图像切成固定大小的小块,把每个图像块映射为向量,再像处理标记序列一样交给 Transformer 编码器。它没有先规定卷积核只能查看局部邻域,而是让不同位置通过自注意力直接交换信息。
这种转换不是把图像真的当成文字。图像块携带连续空间中的颜色和纹理,顺序本身没有词法含义。模型能否理解空间关系,取决于位置表示、训练目标、数据规模和分辨率处理。
图像块把二维输入变成标记序列
设输入图像大小为高乘宽,块边长为固定值。图像被划分为互不重叠的块,每块展平并线性投影到统一维度。块越小,空间细节越多,序列长度也按面积快速增长;块越大,计算较省,却可能在第一步就混合细小对象和边界。
分类任务常在序列前加入一个可学习的分类标记。经过多层编码后,分类标记的表示送入分类头。也可以汇聚所有块表示,或为检测、分割保留空间排列。分类标记只是汇聚设计,不是必须具备的视觉概念。
图像尺寸若不能整除块大小,需要裁剪、填充或调整。上线时改变缩放比例会改变每个块覆盖的真实区域,训练与推理必须采用一致规则,并检查对象形变和边缘损失。
位置编码补回空间次序
自注意力本身对标记排列没有天然空间方向。视觉 Transformer 因此为每个块加入位置表示,使模型能区分左上与右下、相邻与远隔。位置表示可以学习,也可以由坐标函数产生。
当推理分辨率与训练分辨率不同,块数量会变化,学习到的位置向量往往需要插值。插值使形状匹配,不保证空间关系完全迁移。宽高比变化、极端裁剪或超出训练尺度时,应通过独立数据验证,而不是仅确认程序可以运行。
二维位置还不足以表达所有视觉结构。边缘连续性、平移等变和局部纹理在卷积中由结构直接提供,在纯注意力模型中更多依靠数据学习。混合架构或层次化窗口可以重新引入局部约束,但这属于不同的设计取舍。
自注意力建立全局依赖也带来成本
每个标记生成查询、键和值。查询与所有键的相似度决定对值的加权汇聚,多头机制在不同表示子空间并行进行。一个图像块可以在单层中直接关联远处块,这对整体形状和长距离关系有价值。
标准注意力的计算和内存随标记数平方增长。把块边长减半,标记数约增至四倍,注意力矩阵规模约增至十六倍。高分辨率视觉任务因此常采用窗口、稀疏注意力、分层下采样或混合特征,而不是无限缩小图像块。
注意力权重反映模型内部的加权关系,不自动构成对象边界或因果解释。可视化某个头的权重可以帮助诊断,但不能据此证明模型使用了正确证据。
预训练规模改变架构比较
原始 Vision Transformer 工作显示,在大规模数据预训练并迁移时,纯 Transformer 能取得有竞争力的图像识别结果。这一结论绑定训练数据、算力、正则化和迁移设置。把论文架构直接用于较小数据集,表现可能不同。
卷积的局部性和权重共享提供较强归纳偏置,数据有限时往往有助于学习。视觉 Transformer 的约束更弱,能从大数据学习更灵活关系,也更依赖优化、增强和预训练。比较架构时应控制参数量、预训练来源、输入分辨率和计算预算。
迁移学习还要关注领域差异。自然图像预训练得到的颜色、纹理和对象关系,未必适合医学影像、遥感多光谱或工业成像。额外波段、不同动态范围和超大场景需要重新设计输入投影、切片与评价。
下游任务需要恢复空间层级
分类只需要整图表示,检测和分割则需要多尺度位置特征。单一块网格对小对象不利,也难以同时表达纹理与大结构。实际系统会从多层提取特征,构建金字塔,或采用逐级合并标记的层次化结构。
密集预测还要把块级结果映射回像素。上采样、跳跃连接和边界细化都可能影响输出。视觉 Transformer 是编码器家族,不是完整检测或分割系统;最终能力取决于解码器、损失、标注和后处理。
对视频或时序图像,时间维度会进一步增加标记数。空间和时间注意力是联合计算还是分解计算,会改变成本和可表达关系。系统设计应从输出需求与容量边界出发,而不是只追求统一模型形式。
评价应覆盖准确性之外的条件
除了任务指标,还应测量吞吐、峰值显存、启动时间和不同分辨率下的延迟。训练侧的参数量相近不表示推理成本相近,硬件对矩阵形状和批量大小的利用率也会改变结果。
鲁棒性检查应包含裁剪、缩放、压缩、遮挡和背景变化,并按类别与对象尺寸分层。若模型依赖位置或背景捷径,总体准确率可能很高,却在部署相机或地理区域变化后失效。
视觉 Transformer 的核心价值,是提供一种让图像块通过注意力形成表示的途径。理解块大小、位置、计算规模、预训练条件和空间恢复机制,比记住某个模型名称更能判断它是否适合具体任务。
资料依据
- Original authors:An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale,https://arxiv.org/abs/2010.11929
- Original authors:Attention Is All You Need,https://arxiv.org/abs/1706.03762