Samples Dispatching Mechanism for Accelerating Recommendation Model Training in Edge Intelligent Computing System
-
摘要:
在边缘智能计算系统中使用边缘工作节点训练深度学习推荐模型（deep learning recommendation model，DLRM）具有诸多优势，尤其是在数据隐私保护、低延迟和个性化推荐等方面. 然而，由于嵌入表的规模庞大，在训练DLRM时通常采用1个或多个参数服务器来维护全局嵌入表，同时利用多个边缘节点缓存嵌入表的一部分. 在此架构下，需要在边缘节点和参数服务器间传输嵌入以保证嵌入数据一致性，嵌入传输代价通常主导了训练周期. 目标旨在研究边缘智能计算系统中，当面对异构网络和资源受限等挑战时，如何将嵌入样本调度到合适的边缘节点上进行训练，以最小化总嵌入传输代价. 为此，提出了一个基于预期嵌入传输代价的嵌入样本调度机制ESD. 在ESD中，设计了一个结合资源密集型最优解法和启发式解法的调度决策方法HybridDis，以实现决策质量和资源消耗之间的平衡. 使用C++和Python实现了ESD的原型系统，并在真实工作负载下将其与现有最先进的机制进行比较. 大量实验结果表明，ESD可将嵌入传输代价至多降低36.76%，并且在端到端DLRM训练速度上实现了最高1.74倍的加速.
Abstract:Using edge workers to train deep learning recommendation model (DLRM) in edge intelligent computing system brings several benefits, particularly in terms of data privacy protection, low latency and personalization. However, due to the huge size of embedding tables, typical DLRM training frameworks adopt one or more parameter servers to maintain global embedding tables, while leveraging several edge workers to cache part of them. This incurs significant transmission cost for embedding transmissions between workers and parameter servers, which can dominate the training cycle. In this paper, we investigate how to dispatch input embedding samples to appropriate edge workers to minimize the total embedding transmission cost when facing edge-specific challenges such as heterogeneous networks and limited resources. We develop ESD, a novel mechanism that optimizes the dispatching of input embedding samples to edge workers based on expected embedding transmission cost. We propose HybridDis as the dispatch decision method within ESD, which combines a resource-intensive optimal algorithm and a heuristic algorithm to balance decision quality and resource consumption. We implement a prototype of ESD using C++ and Python and compare it with state-of-the-art mechanisms on real-world workloads. Extensive experimental results show that ESD reduces the embedding transmission cost by up to 36.76% and achieves up to 1.74 times speedup in end-to-end DLRM training.
-
Keywords:
- distributed training /
- edge intelligence /
- deep learning /
- recommendation model /
- dispatch mechanism
-
随着人工智能技术的迅猛发展，出现了许多优秀的深度学习模型，如转换器（Transformer）[1]. 最初，Transformer被广泛应用于自然语言处理领域并取得了巨大成功. 随后，视觉转换器（vision Transformer，ViT）[2]在计算机视觉领域也逐渐取代卷积神经网络（convolutional neural network，CNN），在图像分类[3]、语义分割[4-5]、目标检测[6-7]、图像检索[8]任务中，ViT展现出了优异的效果. 然而，由于这些模型具有高度复杂的线性关系以及超大规模的参数量，人类无法理解其内部决策过程，因此通常被视为黑盒模型，这成为许多需要安全性和社会认可的现实应用中的主要缺陷[9-10]. 尽管在现实世界已经部署了一些人工智能系统，但在医疗诊断领域，由于其固有且不可否认的风险，Transformer衍生模型难以真正投入使用[11].
传统的深度学习模型解释方法通常是对输入进行遮挡扰动，当遮挡区域为模型认为的重要区域时，往往会导致模型的输出得分较低[12]. 但这种基于遮挡的解释方法通常对遮挡策略有着较高的要求，随机遮挡往往难以达到较好的解释效果.Ding等人[13]提出一种基于随机多粒度随机游走算法的遮挡策略，从粗粒度到细粒度对图像进行遮挡，并以注意力作为指引，大大提升了扰动的效率，并取得了较好的解释效果.Petsiuk等人[14]提出利用蒙特卡洛采样生成一定数量的掩码，通过掩码与输入相乘对输入进行扰动，利用模型输出的类别分数与掩码加权求和得到最终的解释结果. 这一过程无需访问模型内部，因此这是一种黑盒解释方法. 不同的是，Xie等人[15]提出利用ViT模型的最终层输出生成掩码，而无需利用蒙特卡洛采样，解决了解释效率低、伪影导致的分数误差以及掩码像素覆盖偏差的问题.
然而，简单利用ViT编码器最终层输出生成掩码忽略了2个关键方面：一是掩码中存在大量的不确定信息，如果不采取策略将会导致不完整，不可靠的物体定位信息；二是掩码在上下采样过程中造成的信息丢失问题，会导致解释结果中存在大量的空间噪声及细节缺失. 正如文献[16-17]所指出的，ViT存在过度平滑问题，即随着堆叠的层数的增加，各令牌的语义信息趋于一致，导致最终层输出的掩码高度相似，我们遵循了文献[15]中的层次聚类方法，减少冗余掩码数量. 图1展示了积极掩码、消极掩码以及一些不确定性掩码. 为获得更优的显著性解释结果，我们应尽可能充分利用积极掩码. 图1（b）显示了积极掩码，它能够基本完整地反映出原图对预测做出重大贡献的物体位置. 相反，图1（c）展示了消极掩码，这些掩码无意义或与原图完全相反，对可视化结产生消极影响. 此外，图1（d）中的不确定掩码通常会对整体产生一定程度的掩盖，无法确定其对可视化结果的影响，需要进一步的细粒度信息来确定其是否为积极掩码或消极掩码.
Yao[18]提出的三支决策方法能够有效应对上述不确定性问题. 与传统两支决策不同，当目前证据不能做出接受或拒绝的决策时，三支决策采用延迟决策的方式处理信息，将信息置于边界域，以降低错误决策带来的损失. 基于多粒度结构，序贯三支决策应运而生，序贯三支决策对进入边界域的信息通过可靠的细粒度的信息再次进行决策，形成了从粗粒度到细粒度的顺序过程. Savchenk[19]将序贯三支决策应用于图像识别，提高了图像识别的速度，Ju等人[20]开发的具有合理子空间的序贯三支分类器有效提升了分类性能，Li等人[21]提出了代价敏感的序贯三支决策用于人脸识别. 通过序贯三支策略，我们能够从多粒度的角度获取积极掩码，生成更有益的可视化结果[22-23].
自注意力机制是Transformer架构的核心模块，可视化类别令牌的注意力矩阵是ViT一种天然解释方法[24]. 然而，需要注意的是，可视化注意力矩阵的结果与类别无关，即无论输入何种类别，模型的解释结果均相同，这对于多类别图像显然是不合理的. 因此，我们不使用类别令牌，而是通过注意力矩阵分析各令牌之间的信息交互，获取所有令牌包含的前景、背景等细节语义信息. 我们提出一个注意力融合（attention fusion，AF）模块，旨在生成更为准确的可视化解释图.
针对ViT架构的独有特性，本文提出了一种基于序贯三支掩码和注意力融合的Transformer解释方法（SAF-Explainer）. 首先，将图像输入Transformer编码器，并上采样最终层输出作为掩码. 然后，通过序贯三支掩码模块，将掩码划分为积极掩码，消极掩码和不确定掩码. 并通过多粒度层级顺序分析，选出所有的积极掩码. 接着，利用Transformer计算置信度分数，并与积极掩码加权求和，得到初步解释结果. 接下来，使用提出的注意力融合模块对结果进行细节修正. 首先，保存每一层Transformer编码器中的注意力矩阵，并跨层聚合生成关系矩阵. 关系矩阵能够反映图像块之间的关系密切程度，例如前景图像块会与其他前景图像块具有较高的关系，与背景图像块的关系则较低. 因此，通过计算初步解释结果与关系矩阵之间的余弦相似度，得出每一个图像块的块重要性分数，利用重要性分数对初步解释结果进行融合，即可得到最终解释结果. 通过热图的方式，可以清晰地呈现最终的可视化结果，这些结果能够有效反映ViT模型在决策时的依据，从而体现模型的可解释性.
本文主要贡献如下：
1）本文提出序贯三支掩码模块（sequential three-way mask，S3WM），以解决ViT最终层特征图生成的掩码质量不确定性问题，通过设置特定的阈值及条件，可以有效地将掩码划分为积极掩码、消极掩码与不确定掩码3种类型，使用积极掩码可以明显提高模型解释效果；
2）本文提出注意力融合模块（attention fusion，AF），它通过聚合ViT中每一层的注意力矩阵，生成可以反映图像块前景、背景与边缘区域关系的关系矩阵，通过余弦相似度度量获取图像块重要性分数，对解释结果进行加权融合，这一过程有效地解决了初步解释结果中存在的噪声问题，提高模型解释效果；
3）本文在3个自然图像数据集以及1个脑部肿瘤医学图像数据集对SAF-Explainer进行定量定性分析以及可视化结果验证.
1. 相关工作
ViT由于其黑盒特性，本身不具备可解释性，通常需要应用事后解释方法来为训练后的模型生成解释. ViT的可解释性方法可以分为2大类：一类是最初为CNN设计的解释方法，尽管这些方法最初是为CNN设计的，但它们可以迁移并应用于ViT；另一类是专门针对ViT特性设计的解释方法.
1.1 基于反向传播的方法
早期基于反向传播的方法通过梯度近似像素扰动前后模型输出的差值来衡量像素的重要性，梯度的绝对值越高，该像素对预测的影响越显著. 然而，基于梯度的灵敏度图通常包含大量噪声，Smilkov等人[25]进行多次采样并添加高斯噪声，取多次反向传播结果的平均值来获取平滑的灵敏性图. Sundararajan等人[26]也是计算梯度的平均值，其样本通过输入图像与基线图片之间的插值获得，将梯度积分累计获取灵敏性图. 相关性逐层传播方法（layer-wise relevance propagation，LRP）从模型输出开始，反向传播计算各层间的相关性系数[27]，像素的相关性反应了该像素对模型决策的贡献. Voita等人[28]通过LRP挑选出相关性较高的Transformer注意力头. 这些注意力头被视为对预测贡献较大的部分. 然后，剪枝掉绝大多数相关性较低的注意力头. 即使在剪枝后，模型仍能保持较好的性能. Chefer等人[29]提出了一个Transformer归因方法（Transformer attribution，T-Attribution），它采用LRP计算Transformer模型中每一层、每一个注意力头的相关性，然后通过梯度加权，聚合生成新的注意力矩阵，最终生成特定于类的可视化解释.
1.2 基于类激活图的方法
类激活图（class activation map，CAM）的概念最早出现在文献[30]中.CAM将网络中的全连接层替换为全局平均池化层，每个激活图与最终预测输出之间的权重，用于衡量该激活图对预测的贡献程度. 接着，通过权重与激活图之间的线性加权求和，聚合生成重要性分数. 最后，通过热图方式可视化重要区域. 作为CAM的推广，梯度加权的类激活图（gradient-weighted class activation map，Grad-CAM[31]）及其变体方法[32]使用反向传播获取的梯度均值作为每个激活图对最终预测的贡献程度，对激活图进行线性加权求和，使其不依赖于特定架构. Wang等人[33]指出由于梯度饱和或梯度消失问题，往往会导致产生大量视觉噪声和虚假置信度问题，即权重较高的激活图输入到网络中获取的分数却很低，从而提出一种分数加权的激活图方法（score-weighted class activation map，Score-CAM）. 尽管以上方法可以有效定位物体，但由于使用的激活图均来自最终层，生成的可视化结果通常较为粗糙. Jiang等人[34]利用梯度信息为激活图中的每个位置生成单独的权重，来自浅层的激活图可以捕获细粒度的对象定位信息，将不同层的类激活图进行融合后，即可生成精确完整的类激活图. 这种方法生成的解释图可作为伪标签，有效应用于弱监督对象定位和分割任务[35-36].
1.3 基于扰动的方法
Zeiler等人[12]通过滑动一个灰度正方形块来遮挡图像扰乱输入，从而扰乱输入. 然后，将被遮挡的图像输入分类模型中，观察深层特征图和输出结果. 当遮挡区域为图像重要区域时，通常会导致模型输出的分类分数降低. Ding等人[13]使用随机游走算法，从粗到细2个信息粒度对图像进行扰动，观察对模型预测的影响进而解释模型. 随机输入采样解释（randomized input sampling for explanation，RISE）通过蒙特卡洛采样生成一定数量的掩码. 然后，将被掩盖的图像输入模型，以获取特定于类的置信度得分作为权重. 接着，将权重与掩码进行线性加权求和，即可得到图像的重要性分数[14]. 由于掩码是随机生成且与模型无关，RISE是一种无需访问模型内部结果的通用方法. 与RISE类似，因果解释视觉Transformer（causal explanation of vision Transformer，ViT-CX）通过将掩码输入到模型中获取权重以生成可视化解释[15]. 但与RISE不同的是，ViT-CX的掩码不是随机生成的，而是通过最终层补丁嵌入上采样获取，并通过层次聚类算法减少掩码的数量.
1.4 基于注意力的方法
ViT通过自注意力机制对图像上下文进行全局建模，因此可视化注意力分数是了解其工作机理的直觉想法. 然而，简单地使用最后一层编码器的原始注意力矩阵，通常在定位预测目标物体上效果较差. 因为ViT中还涉及前馈网络及大量的跳跃连接，且中间层信息也被忽略. Abnar等人[24]假设自注意力是线性组合的，通过使用有向无环图对网络中的信息流进行建模，将上下编码器层中的令牌视作结点，注意力分数为边的权重，递归相乘注意力矩阵以获取更为准确的注意力分数. 以上工作均使用注意力矩阵中的类别令牌进行可视化. 然而，我们的工作不依赖于注意力矩阵中的类别令牌，而是分析了其他令牌之间的交互作用，通过聚合跨层注意力矩阵生成令牌关系矩阵，融合生成准确合理的可视化解释结果.
在针对ViT架构的方法中，基于反向传播的方法仅考虑梯度信息，导致解释结果比较保守，通常只能识别部分前景物体信息，且重要性得分不高；由于缺乏针对ViT特性定制，基于扰动的方法通常会产生包含大量背景区域的解释结果，难以令人信服；而基于注意力的方法通常只能提供类别无关的解释结果，并且由于专注于注意力组件，缺乏对ViT架构的整体探索，解释结果往往包含大量空间噪声. 为了解决上述缺点，我们的工作充分利用了ViT架构的特性，使用S3WM模块来选择积极掩码并生成初步的解释结果，准确定位到前景物体；接着，本研究深入探讨了ViT架构的自注意力机制，通过所提出的AF模块，消除空间中的噪声信息，优化前景物体边缘细节，最终生成定位精准、噪声低的可视化解释结果.
2. Transformer解释方法
本文提出了基于序贯三支掩码和注意力融合的Transformer解释方法，具体流程如图2所示.
图2（a）为标准ViT模块，图2（b）为S3WM模块，图2（c）为AF模块. 首先，将图像输入ViT模型，获取最后一层编码器的输出，通过重塑和上采样生成掩码用于扰动图像输入. 由于最后一层编码器的输出具有高度的语义相似性，我们采用了层次凝聚聚类算法来聚合相似的掩码，以减少冗余性. 由于掩码质量的不确定性，将掩码集放入S3WM模块，通过多粒度层级顺序分析选出积极掩码集MPOS，并生成初步解释结果S，S可以准确定位物体位置，但往往包含背景噪声信息. 接着，本文分析了自注意力机制，探索了图像块之间的信息交互，并提出了AF模块. 该模块利用跨层的注意力信息生成关系矩阵R，用于进一步优化解释结果中的细节信息，最终生成最优的解释结果V，有效提高模型可解释性.
2.1 预备知识
2.1.1 标准ViT模型
本文主要考虑用于分类任务的标准ViT模型. 令f(⋅)为标准ViT模型， {\boldsymbol{I}} \in {\mathbb{R}^{H \times W \times C}} 为输入图像，其中H，W，C分别是图像的高、宽和通道，将其切割分成大小 P \times P \times C 的共N块小图像块，接着展平后线性嵌入成D维，在头部拼接上类别令牌并加入位置信息编码嵌入，最终产生 N + 1 = HW/{P^2} + 1 个令牌作为输入 {{\boldsymbol{X}}_0} \in {\mathbb{R}^{(N + 1) \times D}} 馈送到Transformer编码器重复L遍，每一层的输出为 {{\boldsymbol{X}}_i} \in {\mathbb{R}^{(N + 1) \times D}} ，i=1, 2, …, L. 在每一个Transformer编码器中， {{\boldsymbol{X}}_i} 经过多头自注意力模块和多层感知机模块处理，并在每个模块后应用跳跃连接，以对全局进行建模. 最后，从 {{\boldsymbol{X}}_L} 中取出类别令牌，输送到多层感知机中获取分类结果.
2.1.2 多头自注意力机制
受人类在复杂场景中能够有效注意到重点区域的启发，注意力机制被引入计算机视觉领域. 以第l层Transformer编码器中自注意力模块为例，将序列 X 作为输入，通过投影变换为查询矩阵Q，键矩阵K和值矩阵V. 接着，将Q与 {K^{\text{T}}} 进行矩阵乘法，并整体除以 \sqrt {{d_{\boldsymbol{K}}}} ，以防止经过softmax后出现梯度消失的问题，然后将矩阵放入softmax函数进行归一化，即可得到原始注意力矩阵 {A^{(l)}} ，其中每一行表示当前令牌对其他图像块的关注程度，最后将 {A^{(l)}} 与V进行矩阵乘法即可得到一次自注意力层的输出结果：
{{\boldsymbol{A}}^{(l)}} = softmax\left(\frac{{{\boldsymbol{Q}}{{\boldsymbol{K}}^{\text{T}}}}}{{\sqrt {{d_{\boldsymbol{K}}}} }}\right), (1) Attention({\boldsymbol{Q}},{\boldsymbol{K}},{\boldsymbol{V}}) = {{\boldsymbol{A}}^{(l)}}{\boldsymbol{V}}, (2) 其中 {d_{\boldsymbol{K}}} 是K的通道维度. 以上为单个自注意力头的工作过程，此时 {{\boldsymbol{A}}^{(l)}} \in {\mathbb{R}^{(N + 1) \times (N + 1)}} .
受卷积神经网络中使用多个卷积核提取不同特征的启发，多头自注意力使用多个自注意头学习多种不同类型的上下文交互信息，增强模型的表达能力，然后，将每个头学习到的特征进行拼接，获得多头自注意力的输出结果：
{{\boldsymbol{head}}_i} = Attention({\boldsymbol{QW}}_i^{\boldsymbol{Q}},{\boldsymbol{KW}}_i^{\boldsymbol{K}},{\boldsymbol{VW}}_i^{\boldsymbol{V}}), (3) MultiHead({\boldsymbol{Q}},{\boldsymbol{K}},{\boldsymbol{V}}) = Concat({{\boldsymbol{head}}_1},…,{{\boldsymbol{head}}_h}){\boldsymbol{W}}, (4) 其中 {\boldsymbol{W}}_i^{\boldsymbol{Q}} ， {\boldsymbol{W}}_i^{\boldsymbol{K}} ， {\boldsymbol{W}}_i^{\boldsymbol{V}} ， {\boldsymbol{W}} 是投影变换参数矩阵， {{\boldsymbol{head}}_i} 为每个头的计算输出结果， h 为头的个数， Concat 表示拼接操作，经过多头自注意力操作后的注意力矩阵为 {{\boldsymbol{A}}^l} \in {\mathbb{R}^{h \times (N + 1) \times (N + 1)}} .
2.2 序贯三支掩码模块
2.2.1 掩码生成
与Score-CAM通过上采样CNN的最后一层激活图获取掩码类似，我们使用ViT最后一层编码器输出重塑并上采样生成掩码. 设初始输入为 {{\boldsymbol{X}}_0} \in {\mathbb{R}^{(N + 1) \times D}} ，经过L层编码器后的输出为 {{\boldsymbol{X}}_L} \in {\mathbb{R}^{(N + 1) \times D}} ，舍去类别令牌后重塑为 {\boldsymbol{m}} \in {\mathbb{R}^{\sqrt N \times \sqrt N \times D}} ，其中每个位置的值代表图像块在空间位置的信息，上采样后获取D个掩码 {{\boldsymbol{M}}_i} \in {\mathbb{R}^{H \times W}} ，构成掩码集 M = \{ {{\boldsymbol{M}}_1},{{\boldsymbol{M}}_2},…,{{\boldsymbol{M}}_D}\} . 在Transformer架构中，图像被分割并转换为补丁令牌，然后与类令牌拼接后，馈送到Transformer编码器中进行L次重复操作，每次输出的补丁令牌的形状保持不变. 由于多次执行自注意力模块，补丁令牌逐渐趋于一致. 随着编码器层的加深，补丁令牌之间的成对余弦相似度逐步升高，这意味着令牌的语义信息逐渐丢失[17]. 因此，掩码之间存在冗余，并包含高质量的积极掩码，低质量的消极掩码，以及大量不确定性掩码. 我们采用层次凝聚聚类算法减少掩码冗余. 首先将每个掩码视作1个单独的簇，使用余弦相似度计算相似度度量矩阵，并通过对相似性度量矩阵求反来得到距离矩阵. 假设掩码 {{\boldsymbol{M}}_i} 与掩码 {{\boldsymbol{M}}_j} 已经被展平为一维张量，则它们之间的距离可以表示为
Distance({{\boldsymbol{M}}_i},{{\boldsymbol{M}}_j}) = 1 - Cosine\_sim({{\boldsymbol{M}}_i},{{\boldsymbol{M}}_j}), (5) Cosine\_sim({{\boldsymbol{M}}_i},{{\boldsymbol{M}}_j}) = \frac{{{{\boldsymbol{M}}_i} \cdot {{\boldsymbol{M}}_j}}}{{\left\| {{{\boldsymbol{M}}_i}} \right\|\left\| {{{\boldsymbol{M}}_j}} \right\|}}, (6) 其中“ \cdot ”表示点乘操作， \left\| \cdot \right\| 表示欧几里得范数.
通过式（5）计算可以得到距离矩阵，距离矩阵中各元素值表示掩码之间的成对距离，通过设定阈值d可以控制生成的聚类数量. 计算簇间距离时采用完全连接策略，即使用2簇中掩码之间的最大距离作为2簇的距离，确保聚合生成的掩码的差异性. 假设经过聚类后共产生c个簇，每个簇包含不同数量的掩码，我们通过对簇内掩码取平均，生成最终经过层次聚类后的掩码集 {M_{\text{C}}} = \{ {{\boldsymbol{M}}_1},{{\boldsymbol{M}}_2},…,{{\boldsymbol{M}}_c}\} .
2.2.2 序贯三支决策
层次聚类在一定程度上减少了掩码的冗余性，但掩码质量仍具有不确定性. 信息量不足是造成不确定性问题的主要原因，这里，我们采用序贯三支决策策略处理不确定性掩码. 首先，将掩码放入ViT模型 f( \cdot ) ，输出的概率可以提供粗粒度的掩码质量信息. 利用这些信息，我们可以将掩码分为积极掩码、消极掩码和不确定性掩码.
对 {{\boldsymbol{M}}_i} \in {M_{\text{C}}} ，如果放入模型后输出与输入图像I相当的概率，即体现了图像核心区域信息，而翻转掩码 {{\textit{1}}} - {{\boldsymbol{M}}_i} 输出较低概率，即仅捕捉到一些次要信息时，则该掩码属于积极掩码，放入接受域POS：
\begin{split} PO{S_{(\alpha ,\beta )}}(f) =\;& \{ {{\boldsymbol{M}}_i} \in {M_{\text{C}}}|(\beta < f(({{\textit{1}}} - {{\boldsymbol{M}}_i}) \odot {\boldsymbol{I}}) < \\ &\alpha ) \wedge (f({{\boldsymbol{M}}_i} \odot {\boldsymbol{I}}) > \alpha )\} , \end{split} (7) 其中 {{\boldsymbol{M}}_i} \in {M_{\text{C}}} 表示当前处理的是属于候选掩码集 {M_{\text{C}}} 中的掩码， {{\boldsymbol{M}}_i} \odot {\boldsymbol{I}} 可以得到经过掩码 {{\boldsymbol{M}}_i} 覆盖后的图像，将其放入ViT模型 f( \cdot ) 中即可得到输出概率，“ \wedge ”为逻辑且运算符，“ \odot ”为逐元素乘法， \alpha 与 \beta 为控制阈值.
在某些情况下，掩码 {{\textit{1}}} - {{\boldsymbol{M}}_i} 可能才是我们所需要的积极掩码，此时应将翻转后的掩码归入接受域POS. 如图3所示，第1行为消极掩码 {{\boldsymbol{M}}_i} 及 {{\boldsymbol{M}}_i} \odot {\boldsymbol{I}} 的可视化结果， {{\boldsymbol{M}}_i} 没有捕捉到图像中的核心信息，但将其取反后，第2行显示的翻转掩码 {{\textit{1}}} - {{\boldsymbol{M}}_i} 是我们需要的积极掩码，因此我们将其放入POS.
当掩码 {{\boldsymbol{M}}_i} 或其翻转掩码{{\textit{1}}} - {{\boldsymbol{M}}_i} 放入模型输出极低的概率，说明掩码 {{\boldsymbol{M}}_i} 捕捉到的是杂乱无用的信息，该掩码属于消极掩码，放入拒绝域NEG：
\begin{split} NE{G_{(\gamma )}}(f) =\;& \{ {{\boldsymbol{M}}_i} \in {M_{\text{C}}}|(f(({{\textit{1}}} - {{\boldsymbol{M}}_i}) \odot {\boldsymbol{I}}) < \\ &\gamma ) \vee (f({{\boldsymbol{M}}_i} \odot {\boldsymbol{I}}) < \gamma )\} , \end{split} (8) 其中“ \vee ”为逻辑或运算符， \gamma 为控制阈值.
剩余掩码在未有进一步的细粒度信息时，无法判断为积极掩码或消极掩码，属于不确定性掩码，不确定性掩码放入边界域BND等待延迟决策：
BN{D_{(\alpha ,\beta ,\gamma )}}(f) = {M_{\text{C}}} - PO{S_{(\alpha ,\beta )}}(f) - NE{G_{(\gamma )}}(f), (9) 其中“ - ”表示差集运算符，表示集合A中的元素但不在集合B中的元素. 在这里，公式表示取出候选掩码集 {M_{\text{C}}} 中所有的剩余掩码，将其放入BND.
接着，我们建议使用KL散度（Kullback-Leibler divergence）来进一步处理BND中的不确定性掩码. 具体来说，我们的模型通过衡量原始图像I与BND中掩码的概率分布差异来筛选积极掩码. 其中， f(I) 为目标分布 {\boldsymbol{P}} ， f({{\boldsymbol{M}}_i} \odot {\boldsymbol{I}}) 为近似分布 {{\boldsymbol{Q}}_i} ，我们定义目标分布 {\boldsymbol{P}} 和 {{\boldsymbol{M}}_i} 对应的近似分布 {{\boldsymbol{Q}}_i} 之间的KL散度：
KL({\boldsymbol{P}}||{{\boldsymbol{Q}}_i}) = \sum {{\boldsymbol{P}}{l} {\text{b}}\frac{{\boldsymbol{P}}}{{{{\boldsymbol{Q}}_i}}}} . (10) 通过式（10）计算出所有 {{\boldsymbol{M}}_i} \in BND 与I的KL散度值，并将其按从小到大顺序排序. 取其中KL散度值较大与较小的掩码 {{\boldsymbol{M}}_i} 放入POS. KL散度大的掩码与输入图像I的概率分布差异大，说明该掩码具备特异性，有利于生成有意义的解释结果，而KL散度小的掩码则可以较好的体现原始图像I的特征. 对于KL散度处于中间范围的掩码，它们的表现较为模糊和不确定. 为了避免这种不确定性对解释结果产生损害，我们将其放入NEG中：
\begin{split} PO{S_{(\delta )}}(KL) =\;& \{ {{\boldsymbol{M}}_i} \in BND|(KL(f(I)||f({{\boldsymbol{M}}_i} \odot \\ &{\boldsymbol{I}})) < \delta \vee (KL(f({\boldsymbol{I}})||f({{\boldsymbol{M}}_i} \odot {\boldsymbol{I}}) > \\ &1 - \delta )\} , \end{split} (11) \begin{split} NE{G_{(\delta )}}(KL) =\;& \{ {{\boldsymbol{M}}_i} \in BND|\delta < (KL(f({\boldsymbol{I}})|| \\ &f({{\boldsymbol{M}}_i} \odot {\boldsymbol{I}})) < 1 - \delta \} , \end{split} (12) 其中 \delta 为控制阈值， {{\boldsymbol{M}}_i} \in BND 指在第1轮三支决策中放入边界域中的掩码.
通过序贯三支决策筛选后，获得最终的积极掩码集 {M_{POS}} = \{ {{\boldsymbol{M}}_1},{{\boldsymbol{M}}_2},…,{{\boldsymbol{M}}_p}\} ，其中p为积极掩码集中掩码的个数. 对于输入图像I中每个像素x的重要性值计算通过对掩码 {{\boldsymbol{M}}_i} 与其放入模型后对应类别c的输出 {f_c}({{\boldsymbol{M}}_i} \odot {\boldsymbol{I}}) ，加权求和，并除以掩码集的期望值 E({M_{POS}}) 获取：
{\boldsymbol{S}}(x) = \dfrac{{\displaystyle\sum\limits_{i = 1}^p {{f_c}({{\boldsymbol{M}}_i} \odot {\boldsymbol{I}}) \cdot {{\boldsymbol{M}}_i}(x)} }}{{E({M_{POS}})}}, (13) 其中 {f_c}( \cdot ) 表示特定于类别c的模型， E( \cdot ) 为取均值函数.
接着，为了便于利用注意力融合模块对解释结果进行优化，对 {\boldsymbol{S}} 进行归一化处理：
{\boldsymbol{S}} = \frac{{{\boldsymbol{S}} - {{\boldsymbol{S}}_{\min }}}}{{{{\boldsymbol{S}}_{\max }} - {{\boldsymbol{S}}_{\min }}}}. (14) 序贯三支掩码模块算法如算法1所示：
算法1. 序贯三支掩码模块算法（S3WM）.
输入：掩码集 {M_{\text{C}}} ，输入图像I，标准ViT模型 f( \cdot ) ，阈值集 \{ (\alpha ,\beta ,\gamma ,\delta )\} ；
输出：初步解释结果S.
① POS = NEG = BND = \emptyset ；/*初始化3个域*/
② \begin{array}{l}PO{S}_{(\alpha ,\beta )}(f)=\{{{\boldsymbol{M}}}_{i}\in {M}_{\text{C}}|(\beta < f(({{\textit{1}}}-{{\boldsymbol{M}}}_{i})\odot \end{array} {\boldsymbol{I}}) < \alpha )\wedge (f({{\boldsymbol{M}}}_{i}\odot {\boldsymbol{I}})> \alpha )\} ；
③ \begin{array}{l}NE{G}_{(\gamma )}(f)=\{{M}_{i}\in {M}_{\text{C}}|(f((1-{M}_{i})\odot I) < \end{array} \gamma )\vee (f({M}_{i}\odot I) < \gamma )\} ；
④ BN{D}_{(\alpha ,\beta ,\gamma )}(f)={M}_{\text{C}}-PO{S}_{(\alpha ,\beta )}(f)-NE{G}_{(\gamma )}(f) ；
⑤ POS=POS\cup PO{S}_{(\alpha ,\beta )}(f) ；
⑥ BND=BND\cup BN{D}_{=(\alpha ,\beta ,\gamma )}(f) ；
⑦ \begin{array}{l}PO{S}_{(\delta )}(KL)=\{{{\boldsymbol{M}}}_{i}\in BND|(KL(f({\boldsymbol{I}})\left|\right|f({{\boldsymbol{M}}}_{i}\odot \end{array} {\boldsymbol{I}}))) < \delta \vee (KL(f(I)\left|\right|f({{\boldsymbol{M}}}_{i}\odot {\boldsymbol{I}}) > 1-\delta)\} ；
⑧ \begin{array}{l}NE{G}_{(\delta )}(KL)=\{{{\boldsymbol{M}}}_{i}\in BND|\delta < (KL(f({\boldsymbol{I}})\left|\right|\end{array} f({{\boldsymbol{M}}}_{i}\odot {\boldsymbol{I}})) < 1-\delta \} ；
⑨ POS=POS\cup PO{S}_{（\delta ）}（KL） ；
⑩ S=0 ；
⑪ for Mi in POS do/*利用积极掩码加权求和*/
⑫ {\boldsymbol{S}}={\boldsymbol{S}}+{f}_{c}({{\boldsymbol{M}}}_{i}\odot {\boldsymbol{I}})\cdot {{\boldsymbol{M}}}_{i} ；
⑬ end for
⑭ {\boldsymbol{S}}={\boldsymbol{S}}/E(POS) ；
⑮ {\boldsymbol{S}}=({\boldsymbol{S}}-{{\boldsymbol{S}}}_{\mathrm{min}})/({{\boldsymbol{S}}}_{\mathrm{max}}-{{\boldsymbol{S}}}_{\mathrm{min}}) ；/*归一化*/
⑯ return {\boldsymbol{S}} . /*返回初步解释结果*/
2.3 注意力融合模块
2.3.1 自注意力信息交互
自注意力机制的核心功能是实现全局信息的交互. 然而，在可解释人工智能领域，以往对自注意力机制的研究大多仅限于类别令牌[24,29]，忽略了占据多数的其余补丁令牌，从而缺乏对自注意力机制整体效益的探索. 我们通过实验分析了自注意力机制中其他补丁令牌的关注区域，进一步探索了跨补丁之间的信息交互[37]. 如图4所示，我们将图像划分为N个部分，并将它们标记为前景块、背景块和边缘块，以便进行详细分析.
![]()
图 4 自注意力机制信息交互分析结果Figure 4. Results of information interaction analysis of self-attention mechanism由2.1.2节分析可知，第l层Transformer编码器层的注意力矩阵为 {{\boldsymbol{A}}^{(l)}} \in {\mathbb{R}^{h \times (N + 1) \times (N + 1)}} ，为了便于分析，我们将注意力矩阵在头部方向上取平均，并舍去类别令牌后，其维度为 N \times N ，每行表示当前图像块对其余图像块的关注程度. 如图4所示，我们分别使用猫与蝴蝶作为前景块对自注意力信息进行分析.图4中第1列为为原始图像，图像被切块处理并按号划分为前景、背景和边缘图像块，图4中第2~4列分别可视化了每一层Transformer编码器所有前景、背景、边缘图像块对其余图像块的平均注意力分数，其中第1个表示对前景图像块的注意力分数，第2个表示对背景图像块的注意力分数，第3个表示对边缘图像块的注意力分数. 观察结果表明，前景区域的图像块始终对前景区域保持较高的关注度. 随着编码器层数的加深，对边缘区域的关注有所提升，但对背景区域的关注始终较低；背景区域的图像块对背景区域也保持较高关注度，但随着层数的加深，对前景和边缘区域的关注度也有所提升；边缘图像块对边缘区域始终保持的很高的关注度，对背景区域关注度低，呈明显的区分性.
这些观察驱使我们利用这些注意力信息生成关系矩阵R，并与初步解释结果S融合，从而生成逐块重要性分数. 通过对图像块逐块加权融合，进一步细化可视化解释结果，生成最终可视化解释结果V.
2.3.2 关系矩阵
考虑一个标准ViT模型，包括l层编码器，每层包含h个自注意力头. 我们提取所有l层中的h个头的注意力矩阵，并舍弃类别令牌得 {\boldsymbol{A}} \in {\mathbb{R}^{l \times h \times N \times N}} . 为了考虑不同编码器层中语义信息，我们首先从层的维度上进行聚合. 由于第1层编码器层尚未获取明确的语义信息，而最后一层编码器层的语义信息已经趋于平滑，我们舍弃该2层，聚合剩下的 l - 2 层：
{\boldsymbol{r}} = \dfrac{{\displaystyle\sum\limits_{i = 2}^{l - 1} {{{({{\boldsymbol{A}}_i} - E({\boldsymbol{A}}))}^2}} }}{{l - 2}}, (15) E({\boldsymbol{A}}) = \dfrac{{\displaystyle\sum\limits_{i = 2}^{l - 1} {{{\boldsymbol{A}}_i}} }}{{l - 2}}, (16) 其中 E({\boldsymbol{A}}) 可以理解为考虑了不同编码器层的一般性关系的注意力矩阵.
利用 {{\boldsymbol{A}}_i} - E({\boldsymbol{A}}) 计算可以得到前景、背景、边缘图像区域的关系差异， {\boldsymbol{r}} \in {\mathbb{R}^{h \times N \times N}} 表示为考虑了跨编码器层的关系矩阵. 根据文献[28]，在多头自注意力中，不同的头部关注的重点不同，我们采取类似策略对r进行聚合得到最终的关系矩阵R：
{\boldsymbol{R}} = \dfrac{{\displaystyle\sum\limits_{i = 1}^h {{{({{\boldsymbol{r}}_i} - E({\boldsymbol{r}}))}^2}} }}{h}, (17) E({\boldsymbol{r}}) = \dfrac{{\displaystyle\sum\limits_{i = 1}^h {{{\boldsymbol{r}}_i}} }}{h}, (18) 其中 E(r) 可以理解为考虑了不同注意力头部的一般性关系的注意力矩阵.
利用 {r_i} - E(r) 计算可以得到前景、背景、边缘图像区域的关系差异， R \in {\mathbb{R}^{N \times N}} 表示图像全局的关系矩阵，每行表示当前图像块与其他图像块之间的关系，列中的值越大，表明2个图像块之间的关系越密切.
我们随机采样了部分前景、背景和边缘图像块，并选取图像块序号在关系矩阵R中对应的行进行重塑操作，如图5第1列所示，正方形块（红色）为采样的前景图像块，圆形块（蓝色）为采样的边缘图像块，三角形块（绿色）为采样的背景图像块. 图5中第2到第4列分别是前景、边缘、背景图像块的可视化结果. 观察结果显示，3种类型的图像块与自身周围的图像块关系更为密切，同时也与具有相同属性的远处图像块存在一定的联系. 这进一步验证了我们在2.3.1节中的结论.
2.3.3 注意力融合
由于初步解释结果S能够有效定位物体位置，我们将关系矩阵R的注意力信息与S融合. 具体而言，我们首先将 {\boldsymbol{S}} \in {\mathbb{R}^{H \times W}} 下采样至 \sqrt N \times \sqrt N 后展平为 1 \times N ，接着计算 {{\boldsymbol{R}}_i} \in {\mathbb{R}^{i \times N}},i = 1,2,…,N 与S的余弦相似度作为第i个图像块的重要性分数：
{{\boldsymbol{P}}_i} = \frac{{{\boldsymbol{S}} \cdot {{\boldsymbol{R}}_i}}}{{\left\| {\boldsymbol{S}} \right\|\left\| {{{\boldsymbol{R}}_i}} \right\|}},i = 1,2,…,N, (19) 直觉上 {{\boldsymbol{P}}_i} 可以理解为第i个图像块关联区域与物体位置信息的相关程度. 当第i个图像块为前景图像块时，会获取较高的重要性得分；当第i个图像块为边缘图像块时，得分会处于中等水平；当第i个图像块为背景图像块时，会获取较低的重要性得分.
通过上述方法计算得到重要性得分 {\boldsymbol{P}} \in {\mathbb{R}^{1 \times N}} ，我们将其重塑为 \sqrt N \times \sqrt N 并上采样至 H \times W 的尺寸，接着与S进行逐元素相乘，最终得到重要性图V：
{\boldsymbol{V}}(x) = {\boldsymbol{P}}(x) \odot {\boldsymbol{S}}(x). (20) 可视化实验及消融实验结果表明，通过注意力融合产生的重要性分数可以有效提升可视化解释结果的定位能力.
SAF-Explainer的完整过程如算法2所示.
算法2. 基于序贯三支掩码和注意力融合的Transformer解释方法.
输入：输入图像I；
输出：解释结果V.
① X=Get_Output（I）；
② Mask=Upsample（Reshape（X））；
③ MC=Agglomerative_Clustering(Mask）；
④ S=S3WM(MC)；/*获取初步解释结果*/
⑤ A=Get_Attention（I）；
⑥ r=0；
⑦ for Ai in A do /*聚合注意力矩阵*/
⑧ r=r+square（Ai − E（A））；
⑨ end for
⑩ r=r/（l − 2）；/*除以层数取均值*/
⑪ R=0；
⑫ for head in r do /*聚合生成关系矩阵*/
⑬ R=R+square（head − E（r））；
⑭ end for
⑮ R=R/h；
⑯ P=dot（S,R）/（norm_2（S） \cdot norm_2（R））；
⑰ V=P \odot S；
⑱ return V. /*获取最终解释结果*/
3. 实验设置及结果分析
我们首先将所提出的SAF-Explainer架构与其他基线架构在定性评估和分割任务上的进行比较，以验证其优越性；再去掉SAF-Explainer中重要模块进行消融实验，验证架构中所提出模块的有效性.
3.1 实验设置
本文实验使用如下数据集进行定性定量评估.
1）ImageNet 2012[38]的验证集，由来自1 000个类别的5万张图像组成. 主要用于验证ViT模型在自然图像上的可解释性能力.
2）COCO 2017（Microsoft common objects in context 2017）[39]验证集，包含来自80个不同类别的
5000 个带注释的分割图像.3）BraTS 2023（brain tumor segmentation 2023）[40-42]数据集，BraTS 2023训练数据集包含1 251名受试者，每个受试者均有来自4种不同磁共振成像模态的3D体积：1）原生T1；2）对比后T1加权；3）T2加权；4）T2流体衰减反转恢复，这些模态严格对齐，并以1 mm × 1 mm × 1 mm的各向同性分辨率重新采样，输入3D体积的大小为240 × 240 × 155.我们遵循文献[43]中的处理方法，通过沿z轴切片每个模态体积以形成总共177 175个2D图像来预处理数据. 由于官方提供的验证数据集中没有真实分割标签，我们将BraTS训练数据集按9:1划分为训练集与验证集，训练集用于ViT分类模型的训练，验证集用于验证SAF-Explainer模型在医学图像上的可解释性.
4）VOC 2012（pascal visual object classes 2012[44]）数据集，包含来自20个类别的1 449个图像的带注释分割的验证集.
分割实验将每个解释结果通过阈值进行二值化形成分割结果，通过图像分割指标来衡量模型性能：1）像素准确率PA（pixel accuracy）；2）平均交并比mIoU（mean intersection over union）；3）戴斯相似性系数DSC（dice similarity coefficient）；设 {n_{ij}} 为类别i被预测成类别j的像素个数， {n_{{\text{cls}}}} 为目标类别个数（包括背景）.
像素准确率为所有预测正确的像素个数与总预测像素个数之比，它易于理解和计算，可以简单直观的反应分割结果的效果，PA的计算公式如下：
PA = \dfrac{{\displaystyle\sum\limits_i {{n_{ii}}} }}{{\displaystyle\sum\limits_i {\displaystyle\sum\limits_j {{n_{ij}}} } }}. (21) 预测区域和真实区域交集除以预测区域和真实区域的并集可以得到单个类别下的IoU，通过计算出每个类别下的IoU取平均即可得到mIoU：
mIoU = \dfrac{1}{{{n_{{\text{cls}}}}}}\dfrac{{{n_{ii}}}}{{\displaystyle\sum\limits_j {{n_{ij}} + \displaystyle\sum\limits_j {{n_{ji}}} - {n_{ii}}} }}. (22) 戴斯相似性系数是衡量2个集合相似度的指标，它通过计算模型预测结果与真实标签之间的相似程度来度量模型的准确性，取值范围是[0,1]，值越大说明模型预测结果与真实标签相似度越高，分割效果越好DSC计算公式如下：
DSC = \dfrac{{2 \times {n_{ii}}}}{{2 \times {n_{ii}} + \displaystyle\sum\limits_{j \ne i} {{n_{ji}}} + \displaystyle\sum\limits_{j \ne i} {{n_{ij}}} }}. (23) 本文实验选用超参 \alpha = 0.98 ， \beta = 0.1 ， \gamma = 0.01 ， \delta = 0.15 ， d = 0.1 ，本文采用的实验平台为PC（13th Gen Intel(R) Core(TM) i9-13900K @ 3.00 GHz），显卡为NVIDIA GeForce RTX
4090 ，内存容量64 GB，Windows10 专业版操作系统，开发工具为JetBrains PyCharm2022.2.1 专业版，使用Python语言实现实验中相关算法.在3.2节和3.3节中，我们将SAF-Explainer与4个基线进行比较：1）Raw Attention[2]；2）T-Attribution[29]；3）Grad-CAM[31]；4）ViT-CX[15].
3.2 定性评估
本文使用ImageNet 2012自然图像数据集和BraTS医学图像数据集上进行定性评估. 图6展示了SAF-Explainer与其他基线方法在单类别自然图像上的解释结果. Grad-CAM和Raw Attention几乎无法定位到图像中的主体部分，T-Attribution虽然可以定位到主体，但其热图覆盖区域往往只能占到主体的50%，例如，它仅关注熊与狼狗的脸部或鸽子的头部，而对它们的身体部分关注较少，无法全面展现模型的解释能力. 我们的方法能够实现最为准确的定位效果，这得益于序贯三支掩码模块中积极掩码的关键作用. 尽管ViT-CX能够定位大部分主体对象，但背景区域往往包含大量噪声信息. 在描绘主体边缘的细节时，其效果也不够精确，无法准确描述物体轮廓. 由于注意力融合模块对边缘图像块的独特设计，SAF-Explainer在一些细节处理方面表现更好，如蝴蝶的边缘更为平滑，熊和狼狗的腿部轮廓描绘的更为准确. 同时，注意力融合也对背景图像块信息进行处理，大大减少了背景中包含的噪声信息.
SAF-Explainer和基线方法对在多类别自然图像的解释结果如图7所示.Raw Attention获得的结果与类别无关，1张图片上2个不同的类别会生成相同的解释，无法体现模型工作机理.Grad-CAM虽然能生成不同类别的解释，但结果通常只包含零散的点，并不令人信服.ViT-CX对2个类别的结果是互补的，不能反映类别特定的特征. 尽管T-Attribution方法可以为2个类别生成不同的解释图，但它仍然缺乏信心. 相反，所提出的方法可以为2类提供准确的解释结果.
本文采用在ImageNet 2012预训练的ViT模型基础上，对磁共振成像脑肿瘤数据集进行2分类迁移学习，将图像分为正常与含脑肿瘤图像，并对迁移学习后的ViT模型进行可解释性评估. 图8展示了SAF-Explainer与其他基线方法在医学图像上的解释结果，其中第7列为肿瘤分割标准答案. 由于医学图像数据集较小，大多数方法难以产生良好的解释结果，Raw Attention未能捕捉肿瘤位置，T-Attribution和ViT-CX虽然在某些情况下可以定位肿瘤，但其对肿瘤位置的信心不足，通常只能定位的少部分肿瘤位置，且其噪声问题依旧存在. Grad-CAM仅在少数情况下能够定位肿瘤位置. 相比之下，本文方法不仅能够准确定位肿瘤位置，还能够有效减少空间噪声，表现出更好的解释效果.
3.3 分割实验
我们使用VOC 2012数据集和COCO 2017数据集进行分割结果评估. 由于数据集中部分图像包含多个标签注释，我们在评估时将图像中所有带注释的对象均视为真实标签，因此所有方法的整体性能指标较低. 各方法在VOC 2012数据集、COCO 2017数据集上的分割指标如表1和表2所示，表中加粗数字表示最优值.
表 1 各方法在VOC 2012数据集上分割性能对比Table 1. Comparison of Segmentation Performance of Each Method on VOC 2012 Dataset% 方法 PA mIoU DSC Raw Attention 62.15 39.61 33.49 T-Attribution 74.01 50.64 46.89 Grad-CAM 65.32 41.25 30.59 ViT-CX 63.39 44.92 48.60 SAF-Explainer（本文） 74.12 52.72 50.30 表 2 各方法在COCO 2017数据集上分割性能对比Table 2. Comparison of Segmentation Performance of Each Method on COCO 2017 Dataset% 方法 PA mIoU DSC Raw Attention 62.02 40.10 33.38 T-Attribution 68.95 46.36 39.95 Grad-CAM 65.20 41.55 30.56 ViT-CX 56.30 37.73 38.98 SAF-Explainer（本文） 69.73 46.59 40.25 从表1中可以看出，Raw Attention，Grad-CAM，ViT-CX在VOC 2012数据集上的PA和mIoU表现相对较差. 原因在于Raw Attention只使用了模型注意力信息，Grad-CAM仅依赖模型梯度信息，而ViT-CX仅使用模型输出信息，这些单一信息源不足以提供全面的解释. T-Attribution结合了注意力信息和梯度信息，取得了更好的效果，但在图像细节处理方面仍然存在不足. SAF-Explainer通过融合模型输出和注意力信息，PA，mIoU，DSC指标均达到了最优值，分别为74.12%，52.72%，50.30%，超过次优方法0.11%，2.08%，1.7%.
由表2可知，COCO 2017数据集由于类别更多，导致各解释方法的分割性能相比VOC 2012数据集有所下降. 只使用单一模型信息的方法表现依旧较差，我们的方法在PA，mIoU，DSC指标上分别达到69.73%，46.59%，40.25%，超过次优值0.78%，0.23%，0.3%.T-Attribution在各指标上可以达到次优值，这主要是由于它综合利用了梯度信息与注意力信息. 结合图6可以看出，由于T-Attribution对解释细节的处理不足，生成的解释结果过于保守，ViT-CX生成的解释图虽然可以定位到主体对象，但由于缺乏针对ViT特性的处理，存在大量的空间噪声. 我们的方法能够更精确地定位图像中的物体位置，同时在主体边缘细节的描绘上表现出色，且包含较少的空间噪声.
3.4 消融实验
为了验证所提模块的有效性，我们设计了3种方法的变体，并在VOC和COCO数据集进行分割实验，各模块的消融实验结果如表3所示. 1）w/o S3WM，即剔除序贯三支掩码模块；2）w/o AF，即剔除注意力融合模块；3）w/o S3WM，AF，即同时剔除序贯三支掩码模块和注意力融合模块.
表 3 VOC 2012与COCO 2017数据集的消融实验结果Table 3. Results of Ablation Experiments on VOC 2012 and COCO 2017 Datasets% 方法 PA mIoU DSC VOC
2012COCO
2017VOC
2012COCO
2017VOC
2012COCO
2017w/o S3WM 72.84 67.64 51.02 45.54 47.89 39.93 w/o AF 71.09 69.06 50.91 45.78 49.51 39.01 w/o S3WM，AF 69.50 66.86 49.17 44.79 47.63 38.97 SAF-Explainer
（本文）74.12 69.73 52.72 46.59 50.30 40.25 移除序贯三支掩码模块（即w/o S3WM）后，VOC 2012和COCO 2017数据集的PA分别下降1.28%和2.09%，mIoU下降1.7%和1.05%，DSC下降2.41%和0.32%. 由此可以看出，S3WM模块有效提高了模型对物体位置的定位能力. 通过S3WM模块处理掩码质量不确定性问题，筛选积极掩码，并利用这些掩码对图像进行扰动，可以显著提高模型的解释效果.
此外，我们也尝试同时剔除序贯三支掩码模块与注意力融合模块（即w/o S3WM，AF）. 结果显示，3项指标在VOC 2012和COCO 2017数据集上的分割性能相较于单独剔除S3WM或AF模块时进一步下降. 这验证了2个模块叠加的有效性，并进一步说明了序贯三支掩码模块在处理掩码质量不确定性问题和筛选积极掩码方面的重要性. 同时，跨层注意力信息聚合生成的关系矩阵能够有效优化解释结果的细节信息.
4. 结论与展望
Transformer模型在计算机视觉领域重要性日益增加，对模型完整与准确的可解释性需求也大大提高. 然而，目前关于Transformer可解释性的研究有限. 本文提出SAF-Explainer，它基于掩码解释方法设计，主要提出了S3WM模块，通过特定的阈值条件设定解决了掩码质量不确定性问题，提出了AF模块，通过聚合注意力矩阵生成关系矩阵，解决了解释结果包含大量噪声的问题，以优化解释结果中的细节信息. 在ImageNet 2012，VOC 2012，COCO 2017，BraTS 2023数据集上的实验表明，我们的方法无论在自然图像还是在医学图像上表现均优于其他基线方法. 由于SAF-Explainer是针对视觉领域的Transformer设计，在未来的研究中，我们会尝试延申SAF-Explainer到其他领域如自然语言处理和多模态领域. 例如，在自然语言处理领域，Transformer模型被训练用于文本情感分类. 我们可以通过生成重要性热图，展示文本中各个词语对模型预测的贡献程度. 这些热图有助于理解模型做出特定预测的原因.
作者贡献声明：成晓天提出了算法核心思想，设计了实验方案，完成实验并撰写论文；丁卫平提出了整个算法的框架并对整个算法思想进行完善，指导了论文修改；耿宇、黄嘉爽、鞠恒荣完善了算法的思路，指导了论文写作并修改论文；郭静协助完成部分实验及完善论文内容.
-
![]()
图 2 深度学习推荐模型训练过程中的未命中拉取、更新推送和逐出推送操作
Figure 2. Miss Pull, Update Push, and Evict Push transmission operations in DLRM training
![]()
图 3 ESD中的嵌入样本调度过程概览
Figure 3. Overview of embedding samples dispatching process in ESD
表 1 符号列表
Table 1 List of Symbols
符号 描述 \mathcal{W} 边缘工作节点集合 {\mathcal{E}}_{i} 迭代 {I}_{i} 的输入嵌入样本， {\mathcal{E}}_{i}=\{{E}_{1},{E}_{2},… ,{E}_{m\times n}\} {I}_{i} 第 i 个训练迭代 m 每个工作节点的批量大小 {E}_{i} 一个嵌入样本， {E}_{i}=\{{x}_{1},{x}_{2},… \} {x}_{i} 一个嵌入样本的ID \boldsymbol{E}\boldsymbol{m}\boldsymbol{b}({{x}}_{{i}}) 嵌入样本ID为 {x}_{i} 对应的嵌入值（向量） {D}_{\rm tran} 一个嵌入的数据量 {B}_{w}^{j} 工作节点 {w}_{j} 和参数服务器间的网络带宽 {T}_{\rm tran}^{j} 工作节点 {w}_{j} 和参数服务器间传输一个嵌入的代价， {T}_{\rm tran}^{j}=\dfrac{{D}_{\rm tran}}{{B}_{w}^{j}} 
下载: 导出CSV
表 2 在使用8个工作节点时，不同批量大小下串行与并行实现的匈牙利算法执行时间
Table 2 Execution Time Between Serial and Parallel Implementations of Hungarian Algorithm for Different Batch Size when Using 8 Workers
ms 每节点
批量大小32 64 128 256 512 1024 CPU串行 9 62 528 3360 50976 134986 GPU并行 21 28 82 186 811 1385
下载: 导出CSV
-
[1] Gu Yulong, Bao Wentian, Ou Dan, et al. Self-supervised learning on users’ spontaneous behaviors for multi-scenario ranking in e-commerce[C]//Proc of the 30th ACM Int Conf on Information & Knowledge Management. New York: ACM, 2021: 3828–3837
[2] Wang Jizhe, Huang Pipei, Zhao Huan, et al. Billion-scale commodity embedding for e-commerce recommendation in Alibaba[C]//Proc of the 24th ACM SIGKDD Int Conf on Knowledge Discovery & Data Mining. New York: ACM, 2018: 839–848
[3] Smith B, Linden G. Two decades of recommender systems at amazon. Com[J]. IEEE Internet Computing, 2017, 21(3): 12−18 doi: 10.1109/MIC.2017.72
[4] Gomez-Uribe C, Hunt N. The netflix recommender system: Algorithms, business value, and innovation[J]. ACM Transactions on Management Information System, 2015, 6(4): 1−19
[5] Covington P, Adams J, Sargin E. Deep neural networks for Youtube recommendations[C]//Proc of the 10th ACM Conf on Recommender Systems. New York: ACM, 2016: 191−198
[6] Schedl M, Knees P, Gouyon F. New paths in music recommender systems research[C]//Proc of the 11th ACM Conf on Recommender Systems. New York: ACM, 2017: 392−393
[7] Sharma A, Jiang J, Bommannavar P, et al. Graphjet: Real-time content recommendations at Twitter[J]. Proceedings of the VLDB Endowment, 2016, 9(13): 1281−1292 doi: 10.14778/3007263.3007267
[8] Boeker M, Urman A. An empirical investigation of personalization factors on Tiktok[C]//Proc of the ACM Web Conf 2022. New York: ACM, 2022: 2298−2309
[9] Ying R, He Ruining, Chen Kaifeng, et al. Graph convolutional neural networks for web-scale recommender systems[C]//Proc of the 24th ACM SIGKDD Int Conf on Knowledge Discovery & Data Mining. New York: ACM, 2018: 974−983
[10] 彭迎涛,孟小峰,杜治娟. 多样化推荐综述[J]. 计算机研究与发展,2025,62(2):285−313 doi: 10.7544/issn1000-1239.202330600 Peng Yingtao, Meng Xiaofeng, Du Zhijuan. Survey on diversified recommendation[J]. Journal of Computer Research and Development, 2025, 62(2): 285−313 (in Chinese) doi: 10.7544/issn1000-1239.202330600
[11] Wang Siqi, Feng Tianyu, Yang Hailong, et al. Atrec: Accelerating recommendation model training on CPUs[J]. IEEE Transactions on Parallel and Distributed Systems, 2024, 35(6): 905−918 doi: 10.1109/TPDS.2024.3381186
[12] Sayed A, Himeur Y, Alsalemi A, et al. Intelligent edge-based recommender system for Internet of energy applications[J]. IEEE Systems Journal, 2021, 16(3): 5001−5010
[13] Himeur Y, Alsalemi A, Al-Kababji A, et al. A survey of recommender systems for energy efficiency in buildings: Principles, challenges and prospects[J]. Information Fusion, 2021, 72: 1−21 doi: 10.1016/j.inffus.2021.02.002
[14] Pourpanah F, Etemad A. Exploring the landscape of ubiquitous in-home health monitoring: A comprehensive survey[J]. ACM Transactions on Computing for Healthcare, 2024, 5(4): 1−43
[15] Su Xin, Giancarlo S, Vincenzo M, et al. An edge intelligence empowered recommender system enabling cultural heritage applications[J]. IEEE Transactions on Industrial Informatics, 2019, 15(7): 4266−4275 doi: 10.1109/TII.2019.2908056
[16] Yin Hongzhi, Chen Tong, Qu Liang, et al. On-device recommender systems: A tutorial on the new-generation recommendation paradigm[C]//Proc of the ACM Web Conf 2024. New York: ACM, 2024: 1280−1283
[17] Cai Qiqi, Cao Jian, Xu Guandong, et al. Distributed recommendation systems: Survey and research directions[J]. ACM Transactions on Information Systems, 2024, 43(1): 1−38
[18] Long Jing, Ye Guanhua, Chen Tong, et al. Diffusion-based cloud-edge-device collaborative learning for next POI recommendations[C]//Proc of the 30th ACM SIGKDD Int Conf on Knowledge Discovery & Data Mining. New York: ACM, 2024: 1324−1337
[19] Yuan Wei, Qu Liang, Cui Lizhen, et al. Hetefedrec: Federated recommender systems with model heterogeneity[C]//Proc of the 40th Int Conf on Data Engineering. Piscataway, NJ: IEEE, 2024: 2976−2987
[20] Yongbo Yu, Fuxun Yu, Xiang Sheng, et al. Eaglerec: Edge-scale recommendation system acceleration with inter-stage parallelism optimization on GPUs[C]//Proc of the 60th Design Automation Conf. Piscataway, NJ: IEEE, 2023: 1−6
[21] Gong Yu, Jiang Ziwen, Feng Yufei, et al. EdgeRec: Recommender system on edge in mobile Taobao[C]//Proc of the 29th ACM Int Conf on Information & Knowledge Management. New York: ACM, 2020: 2477−2484
[22] Himeur Y, Sohail S, Bensaali F, et al. Latest trends of security and privacy in recommender systems: A comprehensive review and future perspectives[J]. Computers & Security, 2022, 118: 102746
[23] Guo Yeting, Liu Fang, Cai Zhiping, et al. PREFER: Point-of-interest recommendation with efficiency and privacy-preservation via federated edge learning[J]. Proceedings of the ACM on Interactive, Mobile, Wearable and Ubiquitous Technologies, 2021, 5(1): 1−25
[24] Li Youhuizi, Yu Haitao, Zeng Yan, et al. HFSA: A semi-asynchronous hierarchical federated recommendation system in smart city[J]. IEEE Internet of Things Journal, 2023, 10(21): 18808−18820 doi: 10.1109/JIOT.2023.3281909
[25] Wu Jiang, Yang Yunchao, Hu Miao, et al. FCER: A federated cloud-edge recommendation framework with cluster-based edge selection[J]. IEEE Transactions on Mobile Computing, 2025, 24(3): 1731−1743 doi: 10.1109/TMC.2024.3484493
[26] Cheng H T, Koc L, Harmsen J, et al. Wide & deep learning for recommender systems[C]//Proc of the 1st Workshop on Deep Learning for Recommender Systems. New York: ACM, 2016: 7−10
[27] Guo Huifeng, Tang Ruiming, Ye Yunming, et al. DeepFM: A factorization-machine based neural network for CTR prediction[J]. arXiv preprint, arXiv: 1703.04247, 2017
[28] Jiang Jiazhi, Tian Rui, Du Jiangsu, et al. Mixrec: Orchestrating concurrent recommendation model training on CPU-GPU platform[C]//Proc of the 41st Int Conf on Computer Design. Piscataway, NJ: IEEE, 2023: 366−374
[29] Guo Huifeng, Guo Wei, Gao Yong, et al. ScaleFreeCTR: Mixcache-based distributed training system for CTR models with huge embedding table[C]//Proc of the 44th Int Conf on Research and Development in Information Retrieval. New York: ACM, 2021: 129−1278
[30] Zhao Xiangyu, Wang Maolin, Zhao Xinjian, et al. Embedding in recommender systems: A survey[J]. arXiv preprint arXiv: 2310.18608, 2023.
[31] Zhang Hailin, Liu Zirui, Chen Boxuan, et al. CAFE: Towards compact, adaptive, and fast embedding for large-scale recommendation models[J]. Proceedings of the ACM on Management of Data, 2024, 2(1): 1−28
[32] 苗旭鹏,张敏旭,邵蓥侠,等. PS-Hybrid:面向大规模推荐模型训练的混合通信框架[J]. 清华大学学报(自然科学版),2022,62(9):1417−1425 Miao Xupeng, Zhang Minxu, Shao Yingxia, et al. PS-Hybrid: Hybrid communication framework for large recommendation model training[J]. Journal of Tsinghua University (Science and Technology), 2022, 62(9): 1417−1425 (in Chinese)
[33] Zhang Yuanxing, Chen Langshi, Yang Siran, et al. Picasso: Unleashing the potential of GPU-centric training for wide-and-deep recommender systems[C]//Proc of the 38th Int Conf on Data Engineering. Piscataway, NJ: IEEE, 2022: 3453−3466
[34] Acun B, Murphy M, Wang Xiaodong, et al. Understanding training efficiency of deep learning recommendation models at scale[C]//Proc of the 27th IEEE Int Symp on High-Performance Computer Architecture. Piscataway, NJ: IEEE, 2021: 802−814
[35] Song Xiaoniu, Zhang Yiwen, Chen Rong, et al. Ugache: A unified GPU cache for embedding-based deep learning[C]//Proc of the 29th Symp on Operating Systems Principles. New York: ACM, 2023: 627−641
[36] Kaggle. Click-through rate prediction[EB/OL]. [2025-02-24]. https://www.kaggle.com/c/avazu-ctr-prediction.
[37] Zeng Chaoliang, Liao Xudong, Cheng Xiaodian, et al. Accelerating neural recommendation training with embedding scheduling[C]//Proc of the 21st USENIX Symp on Networked Systems Design and Implementation. Berkeley, CA: Association, 2024: 1141−1156
[38] Agarwal S, Yan Chengpo, Zhang Ziyi, et al. Bagpipe: Accelerating deep recommendation model training[C]//Proc of the 29th Symp on Operating Systems Principles. New York: ACM, 2023: 348−363
[39] Youngeun K, Minsoo R. Training personalized recommendation systems from GPU scratch: Look forward not backwards[C]//Proc of the 49th Annual Int Symp on Computer Architecture. New York: ACM, 2022: 860−873
[40] Adam P, Sam G, Francisco M, et al. Pytorch: An imperative style, high-performance deep learning library[C]//Proc of the 33rd Inter Conf on Neural Information Processing Systems. Red Hook: Curran Associates Inc, 2019: 8026−8037
[41] Ma Kaihao, Yan Xiao, Cai Zhenkun, et al. Fec: Efficient deep recommendation model training with flexible embedding communication[J]. Proceedings of the ACM on Management of Data, 2023, 1(2): 1−21,
[42] Saeed G, Lan Guanghui, Zhang Hongchao. Mini-batch stochastic approximation methods for nonconvex stochastic composite optimization[J]. Mathematical Programming, 2016, 155(1): 267−305
[43] Chanwon P, Jemin L. Mobile edge computing-enabled heterogeneous networks[J]. IEEE Transactions on Wireless Communications, 2020, 20(2): 1038−1051
[44] Li Yun, Ma Hui, Wang Lei, et al. Optimized content caching and user association for edge computing in densely deployed heterogeneous networks[J]. IEEE Transactions on Mobile Computing, 2020, 21(6): 2130−2142
[45] Taegeon U, Byungsoo O, Minyoung K, et al. Metis: Fast automatic distributed training on heterogeneous GPU[C]//Proc of the 2024 USENIX Annual Technical Conf. Berkeley, CA: USENIX Association, 2024: 563−578
[46] Ling Neiwen, Wang Kai, He Yuze, et al. Rt-MDL: Supporting real-time mixed deep learning tasks on edge platforms[C]//Proc of the 19th ACM Conf on Embedded Networked Sensor Systems. New York: ACM, 2021: 1−14
[47] Kong Z, Xu Qiang, Meng Jiayi, et al. Accumo: Accuracy-centric multitask offloading in edge-assisted mobile augmented reality[C]//Proc of the 29th Annual Int Conf on Mobile Computing and Networking. New York: ACM, 2023: 1−16
[48] Zhao M, Choudhary D, Tyagi D, et al. Recd: Deduplication for end-to-end deep learning recommendation model training infrastructure[J]. arXiv preprint, arXiv: 2211.05239, 2022
[49] Miao Xupeng, Zhang Hailin, Shi Yining, et al. HET: scaling out huge embedding model training via cache-enabled distributed framework[J]. Proceedings of the VLDB Endowment, 2021, 15(2): 312−320 doi: 10.14778/3489496.3489511
[50] Adnan M, Maboud Y, Mahajan D, et al. Accelerating recommendation system training by leveraging popular choices[J]. Proceedings of the VLDB Endowment, 2021, 15(1): 127−140 doi: 10.14778/3485450.3485462
[51] Wang Chunnan, Wang Hongzhi, Wang Junzhe, et al. Autosr: Automatic sequential recommendation system design[J]. IEEE Transactions on Knowledge and Data Engineering, 2024, 36(11): 5647−5660 doi: 10.1109/TKDE.2024.3400031
[52] Li Jiayu, He Zhiyu, Cui Yumeng, et al. Towards ubiquitous personalized music recommendation with smart bracelets[J]. Proc of the ACM on Interactive, Mobile, Wearable and Ubiquitous Technologies, 2022, 6(3): 1−34
[53] Wang Qinyong, Yin Hongzhi, Chen Tong, et al. Next point-of-interest recommendation on resource-constrained mobile devices[C]//Proc of the ACM Web Conf 2020. New York: ACM, 2020: 906−916
[54] Long Jing, Chen Tong, Nguyen Q, et al. Decentralized collaborative learning framework for next POI recommendation[J]. ACM Transactions on Information Systems, 2023, 41(3): 1−25
[55] Muhammad K, Wang Q, O'Reilly-Morgan D, et al. Fedfast: Going beyond average for faster training of federated recommender systems[C]//Proc of the 26th ACM SIGKDD Int Conf on Knowledge Discovery & Data Mining. New York: ACM, 2020: 1234−1242
[56] Sun Zehua, Xu Yonghui, Liu Yong, et al. A survey on federated recommendation systems[J]. IEEE Transactions on Neural Networks and Learning Systems, 2024, 36(1): 6−20
[57] Zhang Chunxu, Long Guodong, Zhou Tianyi, et al. Gpfedrec: Graph-guided personalization for federated recommendation[C]//Proc of the 30th ACM SIGKDD Int Conf on Knowledge Discovery & Data Mining. New York: ACM, 2024: 4134−4142
[58] Ding Yuchen, Zhang Siqing, Fan Boyu, et al. Fedloca: Low-rank coordinated adaptation with knowledge decoupling for federated recommendations[C]//Proc of the 18th ACM Conf on Recommender Systems. New York: ACM, 2024: 690−700
[59] Belal Y, Bellet A, Mokhtar S B, et al. PEPPER: Empowering user-centric recommender systems over gossip learning[J]. Proceedings of the ACM on Interactive, Mobile, Wearable and Ubiquitous Technologies, 2022, 6(3): 1−27
[60] Xia S, Wei P, Liu Yanchen, et al. Reca: A multi-task deep reinforcement learning-based recommender system for co-optimizing energy, comfort and air quality in commercial building[C]//Proc of the 10th ACM Int Conf on Systems for Energy-Efficient Buildings, Cities, and Transportation. New York: ACM, 2023: 99−109
[61] Gao Ye, Ma Meiyi, Gordon K, et al. A monitoring, modeling, and interactive recommendation system for in-home caregivers: Demo abstract[C]//Proc of the 18th ACM Conf on Embedded Networked Sensor Systems. New York: ACM, 2020: 587−588
[62] Matam K, Ramezani H, Wang Fan, et al. Quickupdate: A real-time personalization system for large-scale recommendation model[C]//Proc of the 21st USENIX Symp on Networked Systems Design and Implementation. Berkeley, CA: USENIX Association, 2024: 731−744
[63] Wang Zheng, Wang Yuke, Deng Jiaqi, et al. Rap: Resource-aware automated GPU sharing for multi-GPU recommendation model training and input preprocessing[C]//Proc of the 29th ACM Int Conf on Architectural Support for Programming Languages and Operating Systems. New York: ACM, 2024: 964−979
[64] Yang Chen, Chen Jin, Yu Qian, et al. An incremental update framework for online recommenders with data-driven prior[C]//Proc of the 32th ACM Int Conf on Information & Knowledge Management. New York: ACM, 2023: 4894−4900
[65] Sima C, Fu Y, Sit M K, et al. Ekko: A large-scale deep learning recommender system with low-latency model update[C]//Proc of the 16th USENIX Symp on Operating Systems Design and Implementation. Berkeley, CA: USENIX Association, 2022: 821−839
[66] Yu Keping, Guo Zhiwei, Shen Yu, et al. Secure artificial intelligence of things for implicit group recommendations[J]. IEEE Internet of Things Journal, 2021, 9(4): 2698−2707
[67] Deng Yongheng, Wang Guanbo, Yu Sheng, et al. Relayrec: Empowering privacy-preserving CTR prediction via cloud-device relay learning[C]//Proc of the 23rd ACM/IEEE Int Conf on Information Processing in Sensor Networks. Piscataway, NJ: IEEE, 2024: 188−199
[68] Wang Ruoxi, Fu Bin, Fu Gang, et al. Deep & cross network for ad click predictions[C]//Proc of the 23rd ACM SIGKDD Int Conf on Knowledge Discovery and Data Mining. New York: ACM, 2017: 1−7
[69] Chen Wenqiang, Zhan Lizhang, Ci Yuanlong, et al. FLEN: Leveraging field for scalable CTR prediction[J]. arXiv preprint, arXiv: 1911.04690, 2019
[70] Adnan M, Maboud Y E, Mahajan D, et al. Heterogeneous acceleration pipeline for recommendation system training[C]//Proc of the 51st Annual Int Symp on Computer Architecture. Piscataway, NJ: IEEE, 2024: 1063−1079
[71] 贺巩山,赵传磊,蒋金虎,等. 面向深度学习的数据存储技术综述[J/OL]. 计算机学报,2025[2025-03-01].http://cjc.ict.ac.cn/online/onlinepaper/hgs-202551994241.pdf He Gongshan, Zhao Chuanlei, Jiang Jinhu, etal. A survey of data storage technologies for deep learning[J/OL]. Chinese Journal of Computers, 2025[2025-03-01]. http://cjc.ict.ac.cn/online/onlinepaper/hgs-202551994241.pdf (in Chinese)
[72] Xie Minhui, Lu Youyou, Wang Qing, et al. Petps: Supporting huge embedding models with persistent memory[J]. Proceedings of the VLDB Endowment, 2023, 16(5): 1013−1022 doi: 10.14778/3579075.3579077
[73] Wei Yingcan, Langer M, Yu Fan, et al. A GPU-specialized inference parameter server for large-scale deep recommendation models[C]//Proc of the 16th ACM Conf on Recommender Systems. New York: ACM, 2022: 408-419
[74] Goyal P, Dollár P, Girshick R, et al. Accurate, large minibatch SGD: Training ImageNet in one hour[J]. arXiv preprint, arXiv: 1706.02677, 2017
[75] Kuhn H. The hungarian method for the assignment problem[J]. Naval Research Logistics Quarterly, 1955, 2(1/2): 83−97 doi: 10.1002/nav.3800020109
[76] Lopes P, Yadav S, Ilic A, et al. Fast block distributed CUDA implementation of the hungarian algorithm[J]. Journal of Parallel and Distributed Computing, 2019, 130: 50−62 doi: 10.1016/j.jpdc.2019.03.014
[77] Lawler E. Combinatorial Optimization: Networks and Matroids[M]. New York: Olt, Rinehart and Winston, 2001
[78] Munkres J. Algorithms for the assignment and transportation problems[J]. Journal of the Society for Industrial and Applied Mathematics, 1957, 5(1): 32−38 doi: 10.1137/0105003
[79] Kaggle. Display advertising challenge[EB/OL]. [2025-02-23]. https://www.kaggle.com/c/criteo-display-ad-challenge
[80] Tallis M, Yadav P. Reacting to variations in product demand: An application for conversion rate (CR) prediction in sponsored search[J]. arXiv preprint, arXiv: 1896.08211, 2018
[81] Delestrac P, Battacharjee D, Yang Simei, et al. Multi-level analysis of gpu utilization in ml training workloads[C]//Proc of 2024 Design, Automation & Test in Europe Conf & Exhibition. Piscataway, NJ: IEEE, 2024: 1−6
[82] Shubha S, Shen Haiying, Iyer A. Usher: Holistic interference avoidance for resource optimized ML inference[C]//Proc of the 18th USENIX Symp on Operating Systems Design and Implementation. Berkeley, CA: USENIX Association, 2024: 947−964
[83] Yuan Wei, Yang Chaoqun, Qu Liang, et al. Hide your model: A parameter transmission-free federated recommender system[C]//Proc of the 40th Int Conf on Data Engineering. Piscataway, NJ: IEEE, 2024: 611−624
[84] Zhang Ye, Deng Yongheng, Yue Sheng, et al. DualRec: A collaborative training framework for device and cloud recommendation models[J]. IEEE Transactions on Mobile Computing, 2025. https://ieeexplore.ieee.org/abstract/document/10840283
[85] Lian Xiangru, Yuan Binhang, Zhu Xuefeng, et al. Persia: An open, hybrid system scaling deep learning-based recommenders up to 100 trillion parameters[C]//Proc of the 28th ACM SIGKDD Int Conf on Knowledge Discovery & Data Mining. New York: ACM, 2022: 3288−3298
[86] Lai Fan, Zhang Wei, Liu Rui, et al. AdaEmbed: Adaptive embedding for large-scale recommendation models[C]//Proc of the 17th USENIX Symp on Operating Systems Design and Implementation. Berkeley, CA: USENIX Association, 2023: 817−831
[87] Zhao Xiangyu, Liu Haochen, Fan Wenqi, et al. Autoemb: Automated embedding dimensionality search in streaming recommendations[C]//Proc of the 21st Int Conf on Data Mining. Piscataway, NJ: IEEE, 2021: 896−905
[88] Luo Qinyi, Wang Penghan, Zhang Wei, et al. Fine-grained embedding dimension optimization during training for recommender systems[J]. arXiv preprint, arXiv: 2401.04408, 2024
[89] Bahreini T, Badri H, Grosu D. Mechanisms for resource allocation and pricing in mobile edge computing systems[J]. IEEE Transactions on Parallel and Distributed Systems, 2021, 33(3): 667−682
[90] He Ying, Fang Jingcheng, Yu F R, et al. Large language models (LLMs) inference offloading and resource allocation in cloud-edge computing: An active inference approach[J]. IEEE Transactions on Mobile Computing, 2024, 23(12), 11253−11264
[91] Tan Haisheng, Wang Yi, Zhang Chi, et al. Asymptotically tight approximation for online file caching with delayed hits and bypassing[J]. IEEE Transactions on Networking, 2025. https://ieeexplore.ieee.org/abstract/document/10936289
计量
- 文章访问数: 153
- HTML全文浏览量: 18
- PDF下载量: 34
