【论文详解】Sparsity Invariant CNNs

Abstract

       本文主要研究了稀疏输入下的卷积神经网络,并用稀疏的激光雷达扫描数据进行实验。传统的CNN在输入稀疏数据时性能很差,即使提供了丢失数据的位置,效果也不理想。为了解决这个问题,本文提出了一个简单有效的稀疏卷积层,在卷积运算中明确考虑了丢失数据的位置。 

1.Introduction

       卷积神经网络CNN几乎影响了计算机视觉的所有领域。通常情况下,CNN的输入是一个图像或者视频,用稠密的矩阵或者张量表示。通过将卷积层与非线性层、池化层结合,CNN能够在第一层提取低层次特征,然后再后续层中依次学习更高层次的特征。然而,当网络的输入是稀疏或者不规则的(例如只有10%的像素携带有用信息),对于每个滤波器的位置该怎样定义不十分明确,输入的数量和位置会发生变化。

       为了解决这个问题,一个简单的方法是给所有无信息的位置分配一个默认值,但是这种方法只能得到次优的结果,因为过滤器必须对所有可能的激活模式保持不变,其数量随着过滤器的尺寸呈指数增长。

       本文提出了一个简单有效的解决方案:引入了一个稀疏的卷积层,它根据输入像素的有效性对卷积核的元素进行加权。此外,第二个stream将关于像素有效性的信息传送给网络的后续层。

       重要的是,实验证明,该网络对不同稀疏程度的输入数据都有很好的效果

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
以下是一个带有sparsity参数的稀疏Transformer的完整代码: ```python import torch import torch.nn as nn import torch.nn.functional as F from torch.autograd import Variable class SparseSelfAttention(nn.Module): def __init__(self, d_model, num_heads, sparsity, dropout=0.1): super(SparseSelfAttention, self).__init__() self.d_model = d_model self.num_heads = num_heads self.sparsity = sparsity self.dropout = nn.Dropout(dropout) assert d_model % num_heads == 0 self.depth = d_model // num_heads self.qkv = nn.Linear(d_model, 3 * d_model) self.proj = nn.Linear(d_model, d_model) self.softmax = nn.Softmax(dim=-1) def forward(self, x, mask=None): batch_size, seq_len, d_model = x.size() qkv = self.qkv(x).view(batch_size, seq_len, 3, self.num_heads, self.depth) qkv = qkv.permute(2, 0, 3, 1, 4) q, k, v = qkv[0], qkv[1], qkv[2] scores = torch.matmul(q, k.transpose(-2, -1)) scores = scores / (self.depth ** 0.5) if mask is not None: scores = scores.masked_fill(mask == 0, -1e9) if self.sparsity > 0: # Calculate sparsity pattern num_sparse_tokens = int(seq_len * self.sparsity) sparse_indices = torch.topk(scores.view(batch_size, self.num_heads, -1), num_sparse_tokens, dim=-1, largest=False)[1] # Create sparse mask mask = torch.ones_like(scores) mask.scatter_(-1, sparse_indices, 0) # Set sparse scores to -inf to exclude them from softmax scores = scores.masked_fill(mask == 0, -1e9) attn = self.softmax(scores) attn = self.dropout(attn) context = torch.matmul(attn, v) context = context.permute(1, 2, 0, 3).contiguous().view(batch_size, seq_len, -1) output = self.proj(context) return output class SparseTransformer(nn.Module): def __init__(self, d_model, num_heads, num_layers, sparsity, dropout=0.1): super(SparseTransformer, self).__init__() self.d_model = d_model self.num_heads = num_heads self.num_layers = num_layers self.sparsity = sparsity self.dropout = nn.Dropout(dropout) self.layers = nn.ModuleList([SparseSelfAttention(d_model, num_heads, sparsity, dropout) for _ in range(num_layers)]) def forward(self, x, mask=None): for layer in self.layers: x = x + layer(x, mask=mask) x = self.dropout(x) return x ``` 上述代码中,在SparseSelfAttention类的初始化函数中增加了一个sparsity参数,用于控制稀疏程度。在forward函数中,当sparsity大于0时,会根据scores计算出一个稀疏模式,然后创建一个稀疏掩码,并将稀疏位置的scores设置为负无穷以排除它们的影响。通过这种方式,可以实现带有稀疏控制的Transformer模型。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值