DREG通过聚焦关键层的梯度,显著提升模型精度与抗噪能力。
DREG: A Layer-Wise Jacobian Regularization as a General-Purpose Penalty
- 按激活导数大小动态分配正则化压力,聚焦高敏感层
- 在960组实验中整体准确率最高,数据稀缺时优势更明显
- 仅需固定超参即可通用,适合现代大模型部署
我们开展了大规模实证研究,剖析导数正则化(DREG)的贡献。在覆盖4种激活函数、6种正则化器、8个数据集和5个随机种子的960组实验中,结果表明:DREG在整体及干净场景下准确率最高(相比无正则、权衰减和IGPen,p ≤ 0.031);在噪声鲁棒性上仅次于谱归一化(SN),是唯一两类逐层正则化之一;尤其在现代Transformer常用激活函数GELU下表现突出,适用于视觉与自然语言任务中的复杂数据;其优势在数据稀缺条件下最为明显,契合其作为几何归纳偏置替代数据量正则化的定位。整个实验仅使用单一超参λ=10⁻²·⁵,无需针对数据集调优,验证其作为可即插即用正则化器的普适性。该效果源于其设计:将正则化集中在激活导数最大的层,而非均匀施加。
原文摘要 · Abstract (English)
We present a large-scale empirical study isolating the contributions of the Derivative Regularization penalty (DREG). Across a fully-crossed factorial sweep of 960 experiments spanning 4 activations, 6 regularizers, 8 datasets, and 5 random seeds, we ask: when, where, and why does DREG work? Our results establish three principal findings. First, DREG achieves the highest overall and clean-regime accuracy among all regularizers evaluated (significantly so against the unregularized baseline, Weight Decay, and IGPen; Wilcoxon $p \leq 0.031$). It ranks second in noise robustness behind Spectral Normalization (SN) - the only two layer-wise regularizers in the study. Second, DREG is globally the best-performing regularizer under GELU, the default activation in modern transformer architectures, particularly on both messy vision and messy NLP benchmarks, suggesting direct applicability to frontier deep learning settings. Third, DREG's advantage over competing regularizers is most pronounced under data scarcity, consistent with its role as a geometric inductive bias that substitutes for the regularizing effect of data volume. Throughout, DREG is applied with a single fixed hyperparameter $λ= 10^{-2.5}$ and no per-dataset tuning, supporting its characterization as a plug-and-play regularizer for neural networks with nontrivial Jacobian structure. These findings are consistent with DREG's design: concentrating regularization pressure on layers where the activation derivative is largest, rather than constraining the network uniformly.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。