让卷积网络第一层卷积核近似正交,提升模型泛化能力。
Almost Right: Making First-Layer Kernels Nearly Orthogonal Improves Model Generalization
- 仅对第一层卷积核施加近似正交约束,轻量高效。
- 在三种开放集任务中显著提升模型泛化性能。
- 适合生物特征与医疗图像等开放域场景使用。
尽管近年来卷积神经网络(CNN)训练方法不断进步,其在开放集任务(如生物识别和医学领域)中的泛化能力仍不理想。人类对未知视觉刺激具有极强的泛化能力,这启发了基于早期视觉冗余最小化的信息编码假说。现有正交化方法依赖矩阵投影、结构修改或特定初始化,常过度限制网络学习并增加计算开销。本文提出一种灵活轻量的方法,仅对第一层卷积核施加成对正交约束,有效减少特征冗余,同时避免过度约束。在胸部X光异常检测、合成人脸检测和虹膜活体攻击检测三个开放集任务上验证,该方法优于当前先进正交化技术,显著提升模型泛化性。论文附开源代码。
原文摘要 · Abstract (English)
Despite several algorithmic advances in the training of convolutional neural networks (CNNs) over the years, their generalization capabilities are still subpar across several pertinent domains, particularly within open-set tasks often found in biometric and medical contexts. On the contrary, humans have an uncanny ability to generalize to unknown visual stimuli. The efficient coding hypothesis posits that early visual structures (retina, Lateral Geniculate Nucleus, and primary visual cortex) transform inputs to reduce redundancy and maximize information efficiency. This mechanism of redundancy minimization in early vision was the inspiration for CNN regularization techniques that force convolutional kernels to be orthogonal. However, the existing works rely upon matrix projections, architectural modifications, or specific weight initializations, which frequently overtly constrain the network's learning process and excessively increase the computational load during loss function calculation. In this paper, we introduce a flexible and lightweight approach that regularizes a subset of first-layer convolutional filters by making them pairwise-orthogonal, which reduces the redundancy of the extracted features but at the same time prevents putting excessive constraints on the network. We evaluate the proposed method on three open-set visual tasks (anomaly detection in chest X-ray images, synthetic face detection, and iris presentation attack detection) and observe an increase in the generalization capabilities of models trained with the proposed regularizer compared to state-of-the-art kernel orthogonalization approaches. We offer source codes along with the paper.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。