Mar, 2024

基于聚类的神经网络标准化层

TL;DR深度学习面临的挑战包括内部协变量偏移、标签偏移、梯度消失/爆炸、过拟合和计算复杂性。本文引入基于聚类的规范化方法 (CB-Norm) 的两个变种——有监督聚类规范化 (SCB-Norm) 和无监督聚类规范化 (UCB-Norm),提出了一种突破性的一步规范化方法。CB-Norm利用高斯混合模型来解决与梯度稳定性和学习加速相关的挑战。SCB-Norm引入了预定义的数据分区,通过分配的聚类来规范化激活。UCB-Norm动态地对神经元激活进行聚类,适应任务特定的挑战,而不依赖预定义的数据分区。这种双重方法确保了对多样化学习场景的灵活应对。CB-Norm创新地采用一步规范化的方法,其中每个混合成分的参数(激活空间中的聚类)作为深度神经网络的权重。这种自适应聚类过程在训练过程中同时解决了聚类和深度神经网络任务的分辨率,标志着该领域的一个显著进步。