基于Hessian引导梯度的机器遗忘框架(HGU):融合曲率感知更新与后处理的可扩展深度模型数据移除

《Neurocomputing》:Hessian-guided gradient unlearning

【字体: 大 中 小 】 时间:2026年09月17日 来源:Neurocomputing 6.7

编辑推荐:

  研究人员提出了一种名为Hessian引导梯度遗忘(Hessian-Guided Gradient Unlearning, HGU)的框架,用于在深度模型中实现可扩展的数据移除。该方法将梯度的高效性与基于Hessian的精确性相结合,以提升遗忘性能。研究人员引入

  
研究人员提出了一种名为Hessian引导梯度遗忘(Hessian-Guided Gradient Unlearning, HGU)的框架,用于在深度模型中实现可扩展的数据移除。该方法将梯度的高效性与基于Hessian的精确性相结合,以提升遗忘性能。研究人员引入了曲率感知更新机制,该机制在不进行完整模型重训练的情况下近似实现重训练的效果。实验结果表明,该方法达到了89.8%的保留准确率(Retained Accuracy, RA)和48.4%的被遗忘类别准确率(Forgotten-Class Accuracy, FA)。研究人员通过在CIFAR-10、ImageNet子集(ImageNet-Subset)以及网络规模数据集上的验证,确认了该方法具有良好的可扩展性。

论文解读文章

一、研究背景与问题

随着机器学习系统在医疗、自动驾驶和内容生成等领域的广泛应用,严格的隐私与监管约束随之而来。诸如《通用数据保护条例》(General Data Protection Regulation, GDPR)和《加州消费者隐私法案》(California Consumer Privacy Act, CCPA)等法规赋予用户要求删除个人数据的权利。然而,标准的训练过程会将数据模式嵌入到模型参数中,导致训练后的模型难以轻易移除特定数据点的影响。这一矛盾催生了“机器遗忘”(Machine Unlearning)领域,其目标是消除目标数据对模型的影响,同时保持模型在保留数据上的性能。
现有的机器遗忘方法面临着一个明确的权衡:基于重训练的方法保真度高,但在大规模系统中成本高昂;而近似方法虽然修改参数以移除特定数据影响,但可能残留痕迹,削弱隐私保障。具体而言,基于梯度的方法利用一阶近似,可扩展性强,但仅能局部调整参数,在复杂决策边界下会留下残余影响。基于影响函数(Influence Function)的方法通过Hessian矩阵引入二阶信息,能够更准确地估计移除数据带来的影响,但其计算和求逆过程对于大规模模型而言代价高昂,限制了实际应用。因此,核心问题在于如何在无需完全重训练的前提下,实现既精确又高效的遗忘。

二、研究内容与结论

为解决上述问题,研究人员提出了Hessian引导梯度遗忘(HGU)框架。该框架保留了梯度更新的高效性,同时集成基于Hessian的近似来指导参数调整,从而在保持可扩展性的同时提升了移除精度。此外,研究人员还引入了后处理步骤,包括激活掩码(Activation Masking)和潜在空间修补(Latent-Space Inpainting),以进一步抑制残余记忆效应。实验在多个基准数据集上进行,包括CIFAR-10、Fashion-MNIST、SVHN、CelebA、ImageNet子集以及网络域数据集(Twitter100k、Amazon Products),结果表明HGU在遗忘效果、保留精度和运行时间开销方面均优于现有基线方法。该研究的重要意义在于提供了一个兼顾效率与精度的实用遗忘方案,支持在需要持续更新且受隐私约束的系统中部署。

三、关键技术方法

研究人员采用的主要技术方法包括:1)结合梯度与Hessian的统一更新规则,该规则利用Hessian矩阵作为预条件子(Preconditioner),对梯度方向进行曲率感知的缩放,以提升遗忘精度;2)后处理中的激活掩码,通过设定阈值识别并抑制与被遗忘数据高度相关的神经元激活;3)潜在空间修补,使用插值技术修复因遗忘导致的特征表示间隙,维持下游任务性能;4)可扩展的二阶近似策略,包括对角近似(Diagonal Approximation)、低秩分解(Low-Rank Decomposition)和分块近似(Block-wise Approximation),以避免完整Hessian矩阵的计算与存储开销。所有方法均在ResNet-18和VGG-16架构上进行了评估。

四、研究结果

1. 与基线方法的比较:在CIFAR-10上,HGU实现了48.4%的被遗忘类准确率(Forget Accuracy, FA)和89.8%的保留类准确率(Retain Accuracy, RA),显著优于纯梯度方法(FA 51.12%, RA 84%)和纯Hessian方法(FA 47%, RA 86%)。在ImageNet子集上,HGU的FA为46%,RA为87%,同样表现最佳。在所有数据集上,HGU的KL散度(衡量与重训练模型输出分布的相似度)最低,表明其行为最接近理想的重训练模型。
2. 消融研究:通过对CIFAR-10进行消融实验,研究人员量化了各组件的贡献。仅使用梯度更新时,FA为55%,RA为84%;仅使用Hessian更新时,FA为48%,RA为86%;两者结合后,FA降至45%,RA升至89%。在此基础上添加掩码后处理,FA进一步降至42%,RA提升至91%;最终完整的HGU(结合+掩码+修补)达到了最低的FA(40%)和最高的RA(92%),证实了每个组件的增量贡献。
3. 后处理效果评估:掩码和修补后处理显著提升了遗忘效果。在CIFAR-10上,完整的HGU实现了98.7%的数据移除率(Data Removal Rate),峰值信噪比(Peak Signal-to-Noise Ratio, PSNR)为31.2 dB,结构相似性指数(Structural Similarity Index, SSIM)为0.95,均优于仅使用梯度或Hessian的方法。
4. 可扩展性验证:通过对Hessian进行对角近似,ResNet-18上的计算时间从完整Hessian的412.7秒降至5.1秒,加速约81倍;即使在拥有1.38亿参数的VGG-16上,对角近似也仅需34.8秒,而完整Hessian耗时超过一小时。这表明近似策略使二阶信息在大型模型中的应用变得可行。
5. 顺序遗忘稳定性:在多次连续遗忘操作的场景下,HGU保持了稳定的性能,RA波动小于1%,FA保持在较低水平,证明了其在重复删除请求下的鲁棒性。

五、总结与结论

研究人员提出的Hessian引导梯度遗忘(HGU)框架,通过融合一阶梯度的计算效率与二阶Hessian的曲率信息,并辅以后处理掩码与修补技术,有效解决了现有遗忘方法在精度与效率之间的权衡问题。该方法避免了完全重训练的高昂成本,同时显著抑制了被遗忘数据的残余影响。实验证明,HGU在多种视觉分类任务和大规模网络数据集上均表现出色,具备良好的可扩展性和实用性。
研究结论:本研究提出了一种结合梯度效率与Hessian精度的可扩展机器遗忘方法。通过引入曲率感知的参数更新以及掩码与修补后处理步骤,该方法在不进行完整模型重训练的情况下,实现了接近重训练水平的遗忘效果。实验结果验证了该方法在多个基准数据集上的有效性,并展示了其在满足GDPR和CCPA等隐私法规要求的系统中的部署潜力。
相关新闻
生物通微信公众号
微信
新浪微博
  • 搜索
  • 国际
  • 国内
  • 人物
  • 产业
  • 热点
  • 科普

热点排行

    今日动态 | 人才市场 | 新技术专栏 | 中国科学人 | 云展台 | BioHot | 云讲堂直播 | 会展中心 | 特价专栏 | 技术快讯 | 免费试用

    版权所有 生物通

    Copyright© eBiotrade.com, All Rights Reserved

    联系信箱:

    粤ICP备09063491号