仅靠结构无法重建高性能神经网络权重,加入行为一致性损失可显著提升效果。
Structure Is Not Enough: Leveraging Behavior for Neural Network Weight Reconstruction
- 在权重空间训练自编码器时引入输出行为对比损失
- 重建模型在下游任务中性能大幅提升,尤其在权重生成上表现更好
- 适合关注模型权重压缩、生成与迁移的开发者
神经网络权重近年来作为新型数据模态受到关注,可用于预测精度、超参数优化或表示学习等任务。现有方法通常使用自编码器(AE)结合对比和重构损失进行训练,在多个下游任务中表现良好且重构误差低。然而,尽管重构误差小,重建模型的性能仍显著低于原模型,限制了其在权重生成中的应用。本文指出,仅依赖权重间欧氏距离的结构损失无法捕捉对高性能重建至关重要的特征。我们提出在权重空间训练中引入行为损失,即在相同输入下比较原模型与重建模型的输出差异。实验表明,结构信号与行为信号具有强协同效应,显著提升了所有评估任务的表现,特别是在神经网络权重重构与生成方面。
原文摘要 · Abstract (English)
The weights of neural networks (NNs) have recently gained prominence as a new data modality in machine learning, with applications ranging from accuracy and hyperparameter prediction to representation learning or weight generation. One approach to leverage NN weights involves training autoencoders (AEs), using contrastive and reconstruction losses. This allows such models to be applied to a wide variety of downstream tasks, and they demonstrate strong predictive performance and low reconstruction error. However, despite the low reconstruction error, these AEs reconstruct NN models with deteriorated performance compared to the original ones, limiting their usability with regard to model weight generation. In this paper, we identify a limitation of weight-space AEs, specifically highlighting that a structural loss, that uses the Euclidean distance between original and reconstructed weights, fails to capture some features critical for reconstructing high-performing models. We analyze the addition of a behavioral loss for training AEs in weight space, where we compare the output of the reconstructed model with that of the original one, given some common input. We show a strong synergy between structural and behavioral signals, leading to increased performance in all downstream tasks evaluated, in particular NN weights reconstruction and generation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。