arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1612.04468cs.CVcs.AIstat.ML
收藏

有限监督神经网络的稀疏分解层

Sparse Factorization Layers for Neural Networks with Limited Supervision

University of Michigan(密歇根大学)

Parker Koch, Jason J. Corso

摘要:

虽然CNN在许多视觉问题上已经取得了巨大的进步,但它们依赖于大量的标记训练数据。另一方面,字典学习并不能扩展到CNN可以处理的问题的大小,尽管它在去噪和修复等低级视觉任务中非常有效。最近,人们越来越关注将字典学习方法用于监督任务,如分类和逆问题。我们提出了两个基于字典学习的新网络层:稀疏因子分解层和卷积稀疏因子分解层,分别类似于全连接层和卷积层。使用我们的推导,这些层可以被放入现有的CNN中,以端到端的方式与反向传播一起训练,并以经典CNN无法实现的方式利用半监督。我们通过实验将具有这两个新层的网络与基线CNN进行比较。我们的研究结果表明,当监督数据很少时,具有稀疏分解层的网络能够优于经典CNN。与没有稀疏因子分解层的CNN相比,它们在某些任务中的性能也有所提高。

英文摘要:

Whereas CNNs have demonstrated immense progress in many vision problems, they suffer from a dependence on monumental amounts of labeled training data. On the other hand, dictionary learning does not scale to the size of problems that CNNs can handle, despite being very effective at low-level vision tasks such as denoising and inpainting. Recently, interest has grown in adapting dictionary learning methods for supervised tasks such as classification and inverse problems. We propose two new network layers that are based on dictionary learning: a sparse factorization layer and a convolutional sparse factorization layer, analogous to fully-connected and convolutional layers, respectively. Using our derivations, these layers can be dropped in to existing CNNs, trained together in an end-to-end fashion with back-propagation, and leverage semisupervision in ways classical CNNs cannot. We experimentally compare networks with these two new layers against a baseline CNN. Our results demonstrate that networks with either of the sparse factorization layers are able to outperform classical CNNs when supervised data are few. They also show performance improvements in certain tasks when compared to the CNN with no sparse factorization layers with the same exact number of parameters.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑