A06 · 第 5 章 · 第三编 视觉表示与综合复习
视觉 Transformer 与混合架构
从图像分块、线性嵌入和二维位置编码出发,推导多头自注意力的形状与二次复杂度,比较分类标记、池化、卷积归纳偏置、局部窗口和混合骨干,并建立分辨率、数据增强与计算预算之间的可核查选择方法。
报告页面错误本章目标
- 把二维图像转换为分块标记序列,并逐轴检查线性投影和多头注意力形状。
- 解释位置编码、分类标记和平均池化各自解决的问题及适用边界。
- 计算分辨率、块尺寸与注意力矩阵的二次成本,避免只按参数量判断显存。
- 比较卷积、全局注意力、局部窗口和混合骨干的归纳偏置与信息路径。
- 为分类或密集预测设计可复现的数据增强、分辨率迁移和失败诊断。
本页目录
从二维网格到标记序列
视觉 Transformer 的第一步不是“对像素直接做语言模型”,而是把图像转换成长度可控、语义明确的标记序列。设一批图像张量为 ,非重叠块高宽为 。当两个空间尺寸都能整除块尺寸时,块网格为
每个块展平为长度 的向量,再乘共享投影矩阵 ,得到维数为 的标记。工程上可用卷积核和步幅均等于块尺寸的卷积实现同一非重叠投影;这只是高效重排与共享线性映射,并不自动赋予后续层完整的卷积层级结构。
若 或 不能整除块尺寸,必须固定裁剪、补边或可变块规则。补边标记要配合有效性掩码,避免池化和注意力把人工边界当成真实内容。训练、验证和部署若采用不同补边方向,即使张量尺寸一致,也会改变对象相对位置。
输入批量为 ,块尺寸为 ,嵌入维数 。块网格为 ,所以每幅图有 个块;每块展平维数为 ,共享投影把 维映到 维。
若在序列首部加入一个分类标记,长度为 。取六个注意力头,每头维数 ,则查询、键和值均可整理为 。一层每个样本每个头产生 的分数矩阵;整批共有
个分数。若仅以单精度保存这些分数,约占 MB;反向传播、归一化概率、查询键值、残差和其他层还会增加显存,因此该数不能当作整层峰值显存。
线性嵌入保留什么
分块会把局部像素压入一个向量。投影矩阵可学习颜色、边缘和纹理组合,但块内的空间信息只有通过投影权重隐式保留;块与块之间的边界在这一刻已被离散化。块越大,序列越短、计算越省,却更容易把小目标、细线和锐利边界压缩到同一标记中。块越小,细节入口增多,但标记数按块面积的倒数增长。
一个 灰度块按行展平为 。取投影
则标记 。若该块位于网格第二行第一列,并给它加入位置向量 ,送入编码器的是 。投影共享表示各块用同一特征抽取规则;位置向量不同表示同样的局部图案出现在不同坐标。这个小例子也说明:嵌入不是可逆压缩的保证,四个输入被压成两个数后一般无法唯一恢复。
位置编码恢复二维几何
不加入位置时,自注意力对标记排列是置换等变的:若以同一排列重排输入,输出也相应重排;再做对称池化后,模型无法仅凭机制区分“左上角有对象”和“右下角有对象”。因此需注入绝对、相对或条件式位置。视觉中最好保留行列结构,例如把行向量和列向量相加,或在注意力分数中加入由二维相对位移决定的偏置。
学习得到的绝对位置表通常绑定训练块网格。推理网格改变时可在二维上插值,但插值不是纯粹的文件格式转换:位置频率、长宽比和边界距离都可能改变,应当把插值规则写入模型版本并在目标分辨率验证。若只把二维表拉平成一维再插值,可能混淆行末到下一行开头的几何邻接。
多头自注意力让所有标记交换信息
对加入位置后的序列 ,一层先计算
将通道拆成 个头且 ,第 个头为
可表示位置偏置, 可屏蔽补边或不允许的连接。各头输出拼接后再做线性投影。多头不是把图像机械分成多个区域,而是允许不同子空间学习不同的相似性与聚合方式。编码块还含残差、归一化和逐标记前馈网络;注意力负责标记间混合,前馈网络负责通道内非线性变换。
取两个二维标记 ,并令查询、键、值投影均为单位矩阵。对第一个查询,缩放分数为 。其 softmax 权重约为 ,输出约为
第二个查询对称地得到 。可见输出既保留较相似的自身标记,也聚合另一个标记。若交换两个标记但不交换位置编码,内容和位置的配对已经改变;若内容与位置一起交换,则输出相应交换。这正是位置必须与块坐标稳定绑定的原因。
注意力权重描述一次前向传播中的加权通道,不能自动等同于因果解释或像素重要性。残差、前馈层、多个头和后续层都会改变信息。若要解释预测,应结合遮挡、反事实、梯度或任务特定诊断,并检查方法本身的稳定性。
分类标记与池化是不同读出合同
分类任务常在序列前加入可学习的分类标记。它不对应任何图像块,而是通过多层注意力收集信息,最终向量送入分类头。另一方案是对有效块标记做平均池化。分类标记提供专门的信息汇聚位置;平均池化结构简单,却必须排除补边标记,并可能稀释只占少数块的小对象信号。
两者没有脱离数据与训练协议的固定胜负。应保持骨干、增强、训练预算和分类头容量可比,再检查总体指标、类别分组、校准和失败样本。对于检测、分割等密集任务,仅保留分类标记显然不够:需要保留块标记,将其重排为二维特征网格,并通过上采样、多尺度融合或任务解码器产生空间输出。
分辨率使全局注意力产生二次成本
标准全局注意力构造 分数矩阵,主要注意力计算近似为 ,分数或概率存储近似为 。查询键值投影和前馈网络还有 量级,故在短序列、大通道时,二次项未必是唯一瓶颈;在高分辨率和小块下,它会迅速占主导。
方形图像使用 块并加入一个分类标记。分辨率 时,块数为 ,长度 ,每头分数元素数为 。分辨率改为 时,块数为 ,长度 ,每头有 个分数,约为前者的 倍。
像素数只增加到约 倍,但全局注意力分数接近其平方增长。若图像高宽同时加倍,块数约变四倍,分数矩阵约变十六倍。选择输入分辨率时必须同时记录批量、精度格式、层数和峰值显存,不能只比较参数量。
局部窗口把每个标记限制在含 个标记的窗口内,单层注意力成本可近似为 ,当 固定时随图像面积近线性增长。代价是单层不能跨窗口直接通信,需要移位窗口、跨窗口层、下采样层级或少量全局标记逐步建立长程联系。窗口边界也是一种结构假设,不应把“更省”误写成“与全局注意力等价”。
数据规模、增强与归纳偏置
卷积将局部连接、权重共享和平移等变性直接写入网络;层级下采样又逐步扩大感受野。朴素视觉 Transformer 允许全局内容相关混合,但内置的局部性和多尺度结构较弱。较弱的先验可以提供更灵活的拟合空间,也意味着模型往往更依赖数据覆盖、预训练、正则化和增强。不存在脱离模型规模、任务、噪声与训练预算的统一“最低数据量”。
增强必须保持标签语义。分类中的随机裁剪若把目标完全裁掉,原标签可能失真;检测和分割还要同步变换框与掩码。混合图像类增强需要明确定义软标签或区域标签。训练增强不应悄悄进入验证集;测试时若使用多裁剪或集成,应计入延迟并与单次推理分开报告。位置插值、输入归一化与补边方式也属于可复现协议。
卷积与注意力的混合边界
混合架构常用卷积干或重叠投影先提取局部特征并降低空间尺寸,再在较粗网格上使用全局或窗口注意力。它保留局部纹理和多尺度入口,同时控制标记数。另一种做法是在注意力层之间加入深度卷积或相对位置偏置。是否采用混合设计,应由输出空间精度、对象尺度、延迟和显存共同决定,而不是由模型家族名称决定。
一项质检任务中的缺陷宽度多为六到十个像素。若输入上直接使用 非重叠块,单个缺陷可能与大量背景一起压进一个标记;把块改为 会把每边标记数翻倍,使全局注意力分数约增至十六倍。
可比较三个候选:第一, 小块全局注意力,细节入口多但成本高;第二,保留高分辨率卷积干,在降采样后使用窗口注意力,并用多尺度解码器恢复边界;第三,维持 块的较轻分类模型。若任务要求像素级定位,第三个候选不满足输出合同;前两个仍需在同一数据划分上测量小缺陷召回、边界误差、峰值显存和端到端延迟。第二个候选是合理起点,但没有测量前不能宣称必胜。
实际诊断可沿信息路径展开:先确认对象在输入中有多少像素,再检查分块或特征步幅是否过粗,随后检查注意力掩码、位置编码和多尺度融合,最后查看阈值、后处理与分布偏移。这样能区分“模型没看见细节”“表示看见但头部没输出”和“输出被后处理删除”。
练习
查看提示
查看解答
查看提示
查看解答
查看提示
查看解答
查看提示
查看解答
查看提示
查看解答
查看提示
查看解答
关系与资源
- 注意力机制 给出查询、键、值与缩放点积的基础。
- 多头注意力与位置编码 解释多头子空间和序列位置的通用形式。
- 卷积网络架构与残差连接 提供局部层级与混合骨干的比较基线。
- 离散卷积、感受野与参数共享 连接块投影、局部性和特征步幅。
- 语义分割、实例分割与密集预测 展示块标记如何进入空间解码器。
- 几何、数据偏差与计算机视觉综合复习 将模型选择放回任务、标注和评估合同。
CS231n: Deep Learning for Computer Vision
Aditesh Kumar, Wenlong Huang, Cristobal Eyzaguirre
用于核对 A06 的离散卷积、网络结构、目标检测、密集预测、视觉 Transformer 及计算机视觉实验边界。
打开官方来源