欢迎访问皮皮网官网
皮皮网

【app直播源码架设】【空降带彩票系统源码】【源码超市图片背景视频】torchsoftmax源码

时间:2024-11-23 11:47:37 分类:时尚 来源:人才招聘网 源码

1.nn.Softmax(dim) 的理解
2.pytorch算子的源码cuda实现——expand,where,softmax
3.基于pytorch的二分类、多分类、源码多标签等问题总结
4.请问用gumbel-softmax的源码时候,怎么让softmax输出的概率分布
5.一文看懂如何实现softmax分类器(内附完全代码)
6.DisenKGAT论文代码精读

torchsoftmax源码

nn.Softmax(dim) 的理解

           使用pytorch框架进行神经网络训练时,涉及到分类问题,就需要使用softmax函数,这里以二分类为例,介绍nn.Softmax()函数中,参数的含义。

        1. 新建一个2x2大小的张量,一行理解成一个样本经过前面网络计算后的输出(1x2),则batch_size是2。

                        import numpy as np

                        import torch

                        import torch.nn as nn

                        a = np.array([[1.5, 6.7],[6.8, 3.4]])

                        b = torch.from_numpy(a)

        2. 下面调用nn.Softmax(dim),dim分别为0,1,看看结果是什么样子

                        f = nn.Softmax(dim = 0)

                        c = f(b)

            ç»“果: tensor([[0., 0.], [0., 0.]], dtype=torch.float)

            可以发现,是每一列和为1.

                        f = nn.Softmax(dim = 1)

            结果:tensor([[0., 0.],  [0., 0.]], dtype=torch.float)

            可以发现是每一行和为1

        所以,当nn.Softmax的输入是一个二维张量时,其参数dim = 0,是让列之和为1;dim = 1,是让行之和为1。

        若nn.Softmax的输入是三维张量时,dim的取值就变成了0,1,2,那又是代表什么意思呢,看下面的例子。

        a = np.array([[[1.5, 6.7, 2.4],

                      [6.8, 3.4, 9.3]],

                      [[3.1, 6.5, 1.9],

                      [8.9, 1.2, 2.5]]])

        我们把a换成一个三维数组,大小是2x2x3,可以看成是2个2x3大小的输入。

        这时,我们定义Softmax函数的dim为0,则结果是:

        tensor([[[0., 0., 0.],

                [0., 0., 0.]],

                [[0., 0., 0.],

                [0., 0., 0.]]], dtype=torch.float)

        可以发现,0.+0. = 1, 0.+0. = 1,即dim = 0,是让两个2x3数据的对应位置和为1.

        使dim=1,结果是:

        tensor([[[0., 0., 0.],

                [0., 0., 0.]],

                [[0., 0., 0.],

                [0., 0., 0.]]], dtype=torch.float)

        可以发现,0.+0. = 1,0.+0. = 1,即dim = 1,是让张量每个2x3数据自己的列之和为1.

        使dim=2,就是让张量每个2x3数据自己的行之和为1.

pytorch算子的cuda实现——expand,where,softmax

       算子的简单介绍

       orch.Tensor.expand() 是 PyTorch 中 Tensor 类的一个方法,用于扩展张量的源码维度。输入:input 是源码要扩展的张量,size 是源码app直播源码架设一个元组,指定了要扩展的源码每个维度的大小。输出:返回一个新的源码张量,形状是源码 input 张量的形状扩展后的形状。例如,源码x 的源码形状假设为(2,1,5,1),x.expand(2,源码4,5,6) 表示希望将 x 扩展为(2,4,5,6)形状的数组,注意 expand 只能在原始形状扩张,源码并且是源码原始维度的长度为1的地方。

       torch.where() 是源码 PyTorch 中的一个函数,用于根据给定的条件从两个张量中选择元素。输入:condition, inputx, inputy。注意 condition, inputx, inputy 不要求形状保持一致。例如,condition=[2,1,1,1],inputx=[1,3,4,1],inputy=[1,2,4,2],这都符合要求,此时输出的形状应该是[2,3,4,2],换句话说,where 应该支持广播运算。

       一维情况的 expand 和 where

       对于一维向量来说,使用 expand 和 where 实现非常简单。对于 expand,我们可以写出伪代码。对于 where,我们也可以写出伪代码。

       高维数组的 expand

       对于高维数组而言,expand 需要考虑在哪个维度扩张。假设输入向量 input 的形状为 [公式],其中至少存在一个 [公式](这是因为 expand 只能在某个维度的长度为1的地方扩张)。假设经过 expand 输出以后 output 的形状为 [公式]。由于计算机存储高维数组仍然需要将数组拉平,弄成一维数组存储,因此我们需要获得高维数组拉平后的长度以及全局索引。例如,input[inputIdx]: inputIdx =[公式]。output[outputIdx]: outputIdx =[公式]。由此我们就可以直接获得 expand 的公式。假设 [公式],即扩张的维度 dim=1,此时对于 output 的任意一个元素,假设索引为 [公式],那么这个元素对应的取值就应该是 input 中索引为 [公式] 的元素。

       高维数组的 where 编程

       有了 expand 的经验之后,where 的空降带彩票系统源码实现就相对容易理解了。在编写 expand 算子的过程中,核心内容是根据 outputIdx 反推得到 inputIdx。同样地,我们可以根据 outputIdx 计算出 inputxIdx, inputyIdx, conditionIdx,然后直接计算即可。

       一维向量的 softmax 介绍

       softmax 的实现相对复杂。这里本人暂时只能实现一维向量的 softmax。假设输入是 x,输出是 y,其中 [公式]。在处理过程中,为了避免指数变化导致溢出,首先需要计算出向量 x 的最大值 [公式],然后做一次变换 [公式],将变换以后的数组记录为 [公式],从而得到 [公式]。

       softmax 的实现可以分解为三步:

       遍历向量 x,计算得到 M,也就是一个 max 规约操作。

       更新向量 x,其中 [公式],计算 [公式],也就是一个 sum 规约。

       得到向量 y,其中 [公式]。

       向量的 sum 规约

       softmax 涉及到两次规约,这里我们先介绍使用的规约策略。我们先用 sum 规约介绍,其实 max 规约几乎如出一辙。

       交叉配对规约:reduce=0

       向量长度是 size(假设 size 是 2 的幂次方),交叉配对的原理是不断将奇数位置的元素加到偶数位置,最终将所有元素的和存放在起始位置。

       以代码举例说明,这里重点说明第 行的代码。定义了一个数组 tmp,其中 __shared__ 关键字表示这个数组放在共享内存中,以及 tmp 数组面对当前线程块中的所有线程都是共享的。对于每个线程块来说,tmp 存储的是该线程块每个线程对应的数据。__syncthreads() 在 GPU 端做一个同步,确保所有线程的数据都赋值到了 tmp。重点来看 行以后的代码。

       以图为例,blockDim=(8,1,1), gridDim=(4,1,1),一共有 个线程,blockDim.x=8。考虑第 行这个循环。当 strip=1 时,threadIdx.x%2 == 0,则 threadIdx.x 有 0,源码超市图片背景视频2,4,6 满足条件。当 threadIdx.x=0 时, 行代码就是 tmp[0] += tmp[1],类似地,成功将数据从 8 个位置转移到了 4 个位置,然后继续修改 strip=2,4,最后将所有元素的和转移到 tmp[0] 即可。最后将 tmp[0] 的数据放到 result[blockIdx.x] 中,也就是说 result 的每个元素存储的都是对应某个线程块内所有元素的和。最终结果全在 result 数组中,那么我们就可以推断出 result 的长度应该是 ceil(num_rows / BLOCK_DIM),其中 ceil 表示向上取整,num_rows 是向量的长度。

       交叉配对规约的特点是规约过程类似于一个二叉树,每次都把奇数位置的元素添加到偶数位置,架构很简单,但具体每一次规约的数据访问却是跳跃的。

       交错配对规约:reduce=1

       分析后面的规约代码,当 strip=4 时,threadIdx.x < 4,就有 threadIdx.x=0,1,2,3 满足条件。当 threadIdx.x=0 时,就有 tmp[0] += tmp[4],tmp[1] += tmp[5],tmp[2] += tmp[6],tmp[3] += tmp[7]。这种做法的好处是每次规约数组的访问都是连续的,可以减少访存开销。后面的数值实验也证明这种并行策略确实加速效果更好。

       shuffle warp:reduce=2

       这种做法的原理比较复杂,这里本人解释的不一定准确。首先声明了一个长度为 的共享内存数组 tmp,这个 实际上是线程的 warp,GPU 线程的设计是每 个线程为一组,称为一个线程簇 warp,我们的一维线程块的 BLOCK_DIM 每隔 个线程做一个切分。

       现在我们进一步考虑线程块内部的线程,对于某个线程块里面的某一个 warp 而言,这里面的 data 其实是线程私有变量,但是 data += __shfl_down_sync(0xffffffff, data, ) 的意思表示不同线程之间的 data 其实可以通过某种特殊方式相互读取累加的。该 warp 有 个线程,第一次累加的原则是,第 0 个线程和第 个线程相加,...,第 个线程和第 个线程数据相加,这样就把 warp 内 个线程的数据全部累加到了前 个线程。后面几句代码也是类似的,最后把该 warp 的源码分享故事大全搞笑 个线程数据累加到第一个线程。所以当 threadIdx.x%==0 时,表示 threadIdx.x 这个线程存储了该 warp 中 个线程数据的累加和。所以 tmp[threadIdx.x/] 位置存储的就是每一个 warp 所有线程数据的和。

       现在需要把 tmp 这 个数据类似原则累加,因此借助于 threadIdx.x 可以直接读取 tmp 的数据,但是 threadIdx.x 的范围太大,我们不要借助那么多所以需要做一个判断以免访问 tmp 越界。最后将 tmp 的数据做完规约以后得到的 data 就保存了当前线程块的所有线程数据的累加和。而我们线程块的总数是 gridDim.x,blockIdx.x 的范围就是 0 到 gridDim.x-1,所以 result 的长度还是 gridDim.x,每个元素保存了对应线程块所有数据的累加和。

       对于某个 warp,为什么第一次规约是以 为距离累加呢?实际上,从线程的角度来看,比如我们考虑最前面的 个线程,也就是 block 中第一个 warp,0 号线程一直到 号线程做累加的时候读取的数据正好是相邻的 个元素,累加的时候读取的是后面 个元素。不管怎么累加,不同线程之间读取的数据总数相邻,这样就保证了数据的局部性,所以访问时间进一步缩短。

       把这些并行策略封装在 pre_dot 以后,后面仍然是写一个 cuda_dot 函数将 result 的结果传回 CPU 的 host_z,然后再串行求 host_z 的元素和。特别注意的是此时调用需要使用 pre_dot<<>>(cuda_x, cuda_y, cuda_z, result),share_size 表示申请的共享内存大小,这里我一般使用 BLOCK_DIM×sizeof(double)。

       向量的 max 规约

       前面两种规约方式对于 max 来说是一样的,稍微有一点区别的是 reduce=2 这种抽牌规约策略。有兴趣的同学可以查看一下 __shfl_down_sync 函数的用法就知道这个是怎么回事了。

       初级 softmax

       有了上面的 sum 和 max 规约策略,我们可以给一个初级的 softmax 算法,步骤如下:

       1:先对数组 input 做一个 max 规约,此时会得到一个 result 数组,这个数组的长度是 ceil(N/BLOCK_DIM),其中 N 是数组的长度,BLOCK_DIM 是线程块的大小。观察向量的 max 规约可以发现这个规约其实只是对向量做了切割,切割成了 ceil(N/BLOCK_DIM),每一份长度是 BLOCK_DIM。而这个向量 result 的长度正好是 ceil(N/BLOCK_DIM),其中 result 的每个元素存储的是对应的那份长度为 BLOCK_DIM 的数据的 max,但是这个不是全局 max,因此我们需要把向量 result 从 GPU 搬运到 CPU,然后通过一个串行 for 循环来遍历 result,由此来获得全局 max,假设是 M。

       2:然后我们做一个更新向量 x,源码下载小说封面app其中 [公式],计算 [公式],此时这个做 sum 规约,仍然是把向量切割成了 ceil(N/BLOCK_DIM),每一份长度是 BLOCK_DIM,并且每份长度为 BLOCK_DIM 的数据的 sum 被存储到了 result 的对应位置(result 可以多次利用,不需要重新定义新的数组)。然后还是需要把向量 result 从 GPU 搬运到 CPU,然后通过一个串行 for 循环来遍历 result,由此来获得全局 sum,假设是 S。

       3:得到向量 y,其中 [公式]。这个过程比较无聊,不赘述。

       综合上面信息,我们发现这个初级的 softmax 的数据来来回回从 CPU 和 GPU 移动了好几次,效率不太高。

       高级 softmax

       我们回顾一下上面的步骤 1 和 2,我们发现生成的 result 需要从 GPU 移动到 CPU,然后还要串行遍历 result 计算全局结果,但我们可以类似在 GPU 端规约 result,依次来减少数据移动,增大速度。经过本人测试,这个想法完全可行,可以称之为二次规约:

       观察上面这段代码,本人定义了一个 girdmax 函数来实现 result 的规约,这个过程和 reduce=1 的规约策略一模一样,经过这样规约以后,全局的 max 将会被存储到 result[0],也就是 result 向量的开头元素。

       类似地,我们继续这么做,这个 gridsum 会将 result 的所有元素都加载到 index=0,也就是开头位置,这样就得到了我们的高级 softmax。

       高维数组的 softmax 目前还没调节好代码,有兴趣的同学可以自己试一试,完整代码参考本人博客。

       本人在下一篇博客给出了一个 softmax 算子融合的介绍,有兴趣可以参考。

基于pytorch的二分类、多分类、多标签等问题总结

       基本照搬 Honda:基于pytorch的二分类、多分类、多标签等问题总结;其中加了一些自己看的时候的思考。

       1. Sigmoid函数和Softmax函数

       1.1 Sigmoid

       Sigmoid函数常用于多标签分类问题,即存在多个正确答案的情况,如胸部X光检查、住院等。在构建分类器时,使用Sigmoid函数处理各个原始输出值,以解决存在多个正确答案的问题。

       1.2 Softmax

       Softmax函数适用于多类别分类问题,即只有一个正确答案的情况,如手写数字、鸢尾花等。在构建分类器时,使用Softmax函数处理各个原始输出值,以解决只有唯一正确答案的问题。

       2. Pytorch中的nn.sigmoid和nn.softmax

       2.1 nn.Softmax

       Softmax函数的表达式为:[公式]

       nn.Softmax可以用于激活函数或多分类,其中分类模型如下:

       2.2 nn.LogSoftmax

       相对于nn.Softmax,nn.LogSoftmax对输出进行了log操作,公式如下:[公式]

       2.3 nn.CrossEntropyLoss

       nn.CrossEntropyLoss用于计算交叉熵,适用于二分类和多分类任务。

       2.4 torch.nn.Sigmoid 和 BCELoss()

       Sigmoid公式为:[公式]

       BCELoss()用于二分类,计算方式如下:[公式]

       对于多标签分类,将Sigmoid和BCELoss整合,即BCEWithLogitsLoss()。

       3. 二分类中的nn.sigmoid和nn.softmax

       3.1 区别

       Sigmoid函数和Softmax函数在二分类问题上是等价的。Sigmoid函数将数值压缩到(0,1)之间,得到的结果可以理解为分类成目标类别的概率。Softmax函数本身针对多项分布,当类别数是2时,它退化为二项分布。

       3.2 使用场景

       在二分类问题时,二者可以互换使用。对于多类别分类任务,通常使用Softmax。对于多标签分类任务,通常使用Sigmoid。

       4. 小思考

       在二分类问题中,使用softmax和sigmoid的结果是相同的。对于多类别分类,sigmoid可以用于多分类,但使用softmax更为合理。在多标签分类中,将问题拆解成多个二分类问题,使用sigmoid进行分类。

请问用gumbel-softmax的时候,怎么让softmax输出的概率分布

       在探讨使用gumbel-softmax时如何让softmax输出概率分布,可以参考pytorch中torch.nn.functional.gumbel_softmax的实现。

       具体操作中,当参数hard设置为True时,返回的结果是one-hot向量。其中,y_soft表示采样得到的概率分布,y_hard是基于此概率分布计算得出的one-hot向量。这里的detach()方法实际上将一个张量从计算图中移除,使其成为常量,以确保在反向传播时不会计算它的梯度。

       综上所述,gumbel-softmax通过构造一个数值上等同于one-hot向量的张量,但在反向传播过程中梯度沿y_soft返回,实现了一种在概率分布与one-hot向量间平滑过渡的方式。这种方式在处理离散概率分布时具有优势,能够提供更好的灵活性和稳定性。

一文看懂如何实现softmax分类器(内附完全代码)

       首先导入需要的包。

       设置小批量数目为,转换类型并生成迭代器,与线性回归的读取数据方法相似。

       初始化模型参数:每个图像有个特征值,网络输出为个类别,因此存在*个权重参数和个偏差参数。

       实现softmax运算,将其转换为非负数,每行元素之和为1。

       回顾tensor的按维度操作,定义softmax运算公式。

       将权重和偏差参数与输入数据相乘,得到输出结果。

       定义损失函数,使用torch.gather函数提取真实标签对应的预测概率。

       根据交叉熵函数公式计算损失值。

       计算分类准确率,从预测概率中找出最大概率对应的索引,与真实标签比较,计算准确率。

       优化算法选择小批量随机梯度下降。

       训练模型,调节迭代周期数和学习率获得分类更准确的模型。

       预测模型效果,训练五次后,损失值从0.减少到0.,准确率从0.提升到0.。

       完整程序包含以上步骤,实现softmax分类器。

DisenKGAT论文代码精读

       公式1:互信息

       解释:该公式定义了随机变量x和z之间的互信息。

       公式2:初始组件嵌入

       公式解释:- h^0_{ u,k}:实体u在第k个潜在空间中的初始嵌入。- σ:激活函数(例如ReLU或Sigmoid)。- W_k:用于第k个潜在空间的投影矩阵。- x_u:实体u的特征向量。

       对应代码文件:model.py文件:DisenLayer.py详细讲解总结

       公式3和4:组件级别交互

       公式解释:- m(v, k, r):从邻居v聚合的消息。- φ:组合操作符,可以是减法、乘法、交叉交互或循环相关。- h_{ v,k}:邻居v在第k个组件中的嵌入。- h_r:关系r的嵌入。- θ_r:关系r的投影矩阵。- W_r:关系r的投影矩阵。- diag(w_r):将向量w_r变成对角矩阵。

       对应代码文件:DisenLayer.py详细讲解总结

       如果您对某个具体部分有更详细的疑问或需要进一步的解释,请告诉我!

       公式5:注意力得分

       公式解释:- α_k(u, v, r):组件k中邻居v对于实体u的注意力得分,考虑关系r。- softmax:Softmax函数,用于归一化注意力得分。- (e^k_{ u,r})^T:实体u在关系r的第k个组件中的嵌入的转置。- e^k_{ v,r}:邻居v在关系r的第k个组件中的嵌入。

       对应代码文件:DisenLayer.py详细讲解总结

       公式6:组件聚合

       公式解释:- h^{ l+1}{ u,k}:实体u在第k个组件中的第l+1层嵌入。- σ:激活函数。- ∑:求和符号,用于聚合邻居信息。- N(u):实体u的邻居集合。- α_k(u, v, r):组件k中邻居v对于实体u的注意力得分。- φ:组合操作符。- h^l{ v,k}:邻居v在第k个组件中的第l层嵌入。- h^l_r:关系r的第l层嵌入。- θ_r:关系r的投影矩阵。

       对应代码文件:DisenLayer.py详细讲解

       ReLU激活函数,对应公式中的点积部分 (e^k_{ u,r})^T · e^k_{ v,r}。-alpha = softmax(alpha, edge_index_i, num_nodes=self.p.num_ent):对alpha进行softmax归一化,对应公式中的softmax。

       总结公式7:关系表示更新

       公式解释:- h^{ l+1}r:关系r的第l+1层嵌入。- W^l{ rel}:第l层的线性变换矩阵。- h^l_r:关系r的第l层嵌入。

       对应代码文件:DisenLayer.py详细讲解总结

       公式8:互信息正则化

       对应代码文件:model.py文件:model.py详细讲解

       .Tanh())

       总结公式9:KL散度

       公式解释:- L(h_{ u,i}, h_{ u,j}):KL散度损失。- D_KL:KL散度。- p(h_{ u,i} | h_{ u,j}):真实条件概率分布。- q_θ (h_{ u,i} | h_{ u,j}):变分条件概率分布,参数为θ。

       对应代码文件:model.py详细讲解总结

       公式:组件级别预测

       公式解释:- ψ_k(u, r, v):候选三元组(u, r, v)在第k个组件中的评分。- f:评分函数。- vec:向量化操作,将矩阵展平为向量。- h^L_{ u,k}:实体u在第k个组件中的第L层嵌入。- h^L_r:关系r的第L层嵌入。- star:卷积操作。- ω:卷积核。- W:线性变换矩阵。- h^L_{ v,k}:实体v在第k个组件中的第L层嵌入。

       对应代码文件:model.py详细讲解

       对应关系:-comb_emb:组合后的实体和关系嵌入,对应公式中的 f(h^L_{ u,k}; h^L_r)。- stack_inp:经过卷积操作的输入,对应公式中的 h^L_r star ω。- x:经过激活函数和线性变换后的嵌入,对应公式中的 f(vec(f(h^L_{ u,k}; h^L_r star ω)))W。

       if self.p.score_order == 'before': x = torch.einsum('bk,bkn->bn', [attention, x]) pred = torch.sigmoid(x) elif self.p.score_order == 'after': x = torch.sigmoid(x) pred = torch.einsum('bk,bkn->bn', [attention, x]) pred = torch.clamp(pred, min=0., max=1.0) return pred, corr

       总结公式:注意力评分

       公式解释:- β_k(u, r):组件k在给定关系r中的重要性。- softmax:Softmax函数,用于归一化注意力得分。- (h^L_{ u,k} circ θ_r)^T:实体u在关系r的第k个组件中的嵌入与关系权重的元素级乘积的转置。- h^L_r:关系r的第L层嵌入。

       对应代码文件:model.py详细讲解

       x = x.view(-1, self.p.num_factors, self.p.gcn_dim)

       总结

       如果您对某个具体部分有更详细的疑问或需要进一步的解释,请告诉我!

       公式:最终预测评分

       公式解释:

       对应代码文件:model.py详细讲解

       f(h^L_{ u,k}; h^L_r star ω))W)。

       if self.p.score_order == 'before': x = torch.einsum('bk,bkn->bn', [attention, x]) pred = torch.sigmoid(x) elif self.p.score_order == 'after': x = torch.sigmoid(x) pred = torch.einsum('bk,bkn->bn', [attention, x]) pred = torch.clamp(pred, min=0., max=1.0) return pred, corr

       总结公式:总体目标

       公式解释:- L:总损失函数。- B:批量大小。- N:知识图谱中的实体数量。- ∑_{ (u, r) ∈ batch}:对批次中的所有三元组(u, r)求和。- ∑_i:对所有候选实体i求和。- t_i:标签,表示三元组(u, r, v_i)是否为有效三元组。- log(ψ_{ final}(u, r, v_i)):最终预测评分的对数值。- λ:互信息正则化损失的权重。- L_mi:互信息正则化损失。

       对应代码文件:run.py详细讲解总结

       如果您对某个具体部分有更详细的疑问或需要进一步的解释,请告诉我!

NLP修炼系列之分类loss二分类、多分类与多标签分类loss损失函数

       本文将探讨二分类、多分类与多标签分类的区别与损失函数,详细分析softmax、log_softmax、CrossEntropyLoss 和 NLLLoss 四个函数的应用与对比,以深入理解它们之间的差异和用法。

       首先,我们需要了解基本概念:

       二分类:任务有两个类别,每个样本属于其中的一个,如新闻分为体育与非体育。

       多分类:任务有多个类别,每个样本属于一个类别,如新闻分类为体育、财经、其他。

       多标签分类:任务有多个类别,每个样本可以属于多个类别,如新闻可能同时属于体育、财经、教育。

       接下来,分析二分类任务:

       输出层仅有一个单元,使用sigmoid损失函数和二分类交叉熵损失。

       torch中提供两种二分类损失函数:BCELoss 和 BCEWithLogitsLoss。

       二者区别在于BCEWithLogitsLoss直接对最后的logits进行处理,不再需要sigmoid缩放。

       使用sigmoid+BCELoss或softmax+交叉熵实现。

       也可直接使用CrossEntropyLoss。

       多分类任务的处理方式:

       输出层单元数为类别数量,使用softmax损失函数和多分类交叉熵损失。

       标签转换为one-hot向量,表示每个类别的独热编码。

       实现代码包括softmax+nll_loss或直接使用CrossEntropyLoss。

       多标签分类任务:

       视为多个二分类任务,每个标签为独立的二分类问题。

       使用sigmoid+BCELoss进行实现,对每个标签进行单独评分。

       标签以one-hot编码形式表示,如[0,0,0,1,0,1,0,0,1]。

       实现代码包括sigmoid+BCELoss或直接使用BCEWithLogitsLoss。

       总结损失函数:

       激活函数将输出缩放到[0,1]。

       求对数操作计算输出的对数。

       累加求和计算损失。

       softmax、log_softmax、CrossEntropyLoss 和 NLLLoss 的对比:

       softmax/sigmoid:对应激活函数操作。

       log_softmax:完成激活函数与对数计算。

       nll_loss:对应损失计算的最后一步。

       CrossEntropyLoss:整合所有步骤。

       多分类交叉熵损失实现方法:

       三步实现:softmax+log+nll_loss。

       两步实现:log_softmax+nll_loss。

       一步实现:crossEntropyLoss。

       总结全文,本文详细介绍了二分类、多分类与多标签分类的区别,以及损失函数softmax、log_softmax、CrossEntropyLoss 和 NLLLoss 的对比,帮助读者理解这些函数在不同分类任务中的应用与区别。

pytorch多分类任务输出层使用softmax,该选择什么损失函数

       多分类任务中,通常在全连接层输出数据为batch_size*n_classes。此结构适配torch.nn.CrossEntropyLoss(),该函数内整合了softmax、log和NLLloss,故全连接层输出结果可直接接入此函数,实现交叉熵损失计算。函数中reduction参数有三种选择:'none'表示返回每个样本的交叉熵损失,'sum'表示返回每个样本损失的总和,'mean'表示返回平均交叉熵损失。

       若全连接层输出结果上先应用了torch.softmax(),则不宜再直接使用torch.nn.CrossEntropyLoss()。此时,需在应用softmax后,进行log操作并取对应位置结果计算损失。具体公式为:loss = torch.log(torch.softmax(results_fc, dim=1))[torch.arange(len(target), target].sum()。

       其中,results_fc为全连接层输出结果,target为真实标签。torch.softmax函数亦可用F.softmax替代。

copyright © 2016 powered by 皮皮网   sitemap