arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2104.04848cs.LG
收藏

基于群分解的自等变网络搜索

Autoequivariant Network Search via Group Decomposition

University of Illinois at Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);Rice University(莱斯大学)

Sourya Basu, Akshayaa Magesh, Harshit Yadav, Lav R. Varshney

摘要:

最近的工作表明,作为一种归纳偏差的组等方差提高了神经网络的分类和生成性能。然而,当感兴趣的组很大且未知时,设计组等变神经网络是具有挑战性的。此外,诱导等方差可以显著减少具有固定特征大小的网络中的独立参数的数量,从而影响其整体性能。我们解决这些问题,证明了一个新的群论结果的上下文中的等变神经网络,表明网络是等变的一个大的组,当且仅当它是等变的较小的群体,它的构造。利用这一结果,我们设计了一种新的快速组等变构造算法,以及一种基于深度Q学习的搜索算法,在缩减的搜索空间中,产生了我们称之为自等变网络(AEN)的网络。在新的基准数据集G-MNIST和G-Fashion-MNIST上进行测试时,AEN在等方差和网络大小之间找到了正确的平衡,这些数据集分别通过我们发布的MNIST和Fashion-MNIST上的组转换获得。将这些结果扩展到组卷积神经网络,在其中我们在等方差,增强和网络大小之间进行优化,我们发现组等方差是CIFAR 10,SVHN,RotMNIST,ASL,EMNIST和KMNIST等几个数据集上所有高性能GCNN中最主要的因素。

英文摘要:

Recent works show that group equivariance as an inductive bias improves neural network performance for both classification and generation. However, designing group-equivariant neural networks is challenging when the group of interest is large and is unknown. Moreover, inducing equivariance can significantly reduce the number of independent parameters in a network with fixed feature size, affecting its overall performance. We address these problems by proving a new group-theoretic result in the context of equivariant neural networks that shows that a network is equivariant to a large group if and only if it is equivariant to smaller groups from which it is constructed. Using this result, we design a novel fast group equivariant construction algorithm, and a deep Q-learning-based search algorithm in a reduced search space, yielding what we call autoequivariant networks (AENs). AENs find the right balance between equivariance and network size when tested on new benchmark datasets, G-MNIST and G-Fashion-MNIST, obtained via group transformations on MNIST and Fashion-MNIST respectively that we release. Extending these results to group convolutional neural networks, where we optimize between equivariances, augmentations, and network sizes, we find group equivariance to be the most dominating factor in all high-performing GCNNs on several datasets like CIFAR10, SVHN, RotMNIST, ASL, EMNIST, and KMNIST.

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

↑