arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2505.08022cs.LGcs.NAmath.NA

神经网络在对抗攻击下具有鲁棒性的动态低秩压缩

Dynamical Low-Rank Compression of Neural Networks with Robustness under Adversarial Attacks

Computer Science and Mathematics Division(计算机科学与数学 division);Geospatial Science and Human Security Division(地理科学与人类安全 division)

Steffen Schotthöfer, H. Lexie Yang, Stefan Schnake

AI总结:

本文提出动态低秩训练与谱正则化方法,在实现94%以上压缩的同时保持或提升对抗鲁棒性,且支持秩自适应。

摘要:

在资源受限设备上部署神经网络需要模型既紧凑又能抵御对抗输入。然而,压缩与对抗鲁棒性往往相互冲突。本文提出了一种动态低秩训练方案,并引入一种新颖的谱正则化器来控制每层低秩核心的条件数。该方法在不牺牲干净数据准确率的前提下,缓解了压缩模型对对抗扰动的敏感性。该方法与模型和数据无关,计算高效,并支持秩自适应以自动压缩当前网络。在标准架构、数据集和对抗攻击上的大量实验表明,正则化网络可实现超过94%的压缩率,同时相对于未压缩基线恢复或提升对抗准确率。

英文摘要:

Deployment of neural networks on resource-constrained devices demands models that are both compact and robust to adversarial inputs. However, compression and adversarial robustness often conflict. In this work, we introduce a dynamical low-rank training scheme enhanced with a novel spectral regularizer that controls the condition number of the low-rank core in each layer. This approach mitigates the sensitivity of compressed models to adversarial perturbations without sacrificing accuracy on clean data. The method is model- and data-agnostic, computationally efficient, and supports rank adaptivity to automatically compress the network at hand. Extensive experiments across standard architectures, datasets, and adversarial attacks show the regularized networks can achieve over 94% compression while recovering or improving adversarial accuracy relative to uncompressed baselines.

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

↑