提出可调损失的医学影像模型删减方法,实现精准去除非关键数据。
Targeted Unlearning Using Perturbed Sign Gradient Methods With Applications On Medical Images
- 基于梯度扰动的双层优化框架,支持迭代式删减训练样本。
- 在临床影像数据上,遗忘与保留指标均优于基线方法。
- 适合设备更新、数据偏移等真实医疗场景下的模型维护。
机器删减旨在不重新训练模型的前提下,消除特定训练样本的影响。以往工作多聚焦于隐私保护场景,本文将其扩展为部署后模型修订的通用工具,尤其针对临床环境中常见的数据漂移、设备淘汰和政策变更。为此,我们提出了基于边界的删减双层优化框架,可通过迭代算法求解,并在使用一阶算法时提供收敛性保证。方法引入可调损失设计,用于控制遗忘与保留之间的权衡,并支持新颖的模型组合策略,融合多次删减运行的优势。在基准数据集和真实临床影像数据集上的实验表明,该方法在遗忘与保留指标上均优于基线,包括涉及成像设备差异和解剖异常值的场景。本工作确立了机器删减作为临床应用中重训练的模块化、实用替代方案。
原文摘要 · Abstract (English)
Machine unlearning aims to remove the influence of specific training samples from a trained model without full retraining. While prior work has largely focused on privacy-motivated settings, we recast unlearning as a general-purpose tool for post-deployment model revision. Specifically, we focus on utilizing unlearning in clinical contexts where data shifts, device deprecation, and policy changes are common. To this end, we propose a bilevel optimization formulation of boundary-based unlearning that can be solved using iterative algorithms. We provide convergence guarantees when first-order algorithms are used to unlearn. Our method introduces tunable loss design for controlling the forgetting-retention tradeoff and supports novel model composition strategies that merge the strengths of distinct unlearning runs. Across benchmark and real-world clinical imaging datasets, our approach outperforms baselines on both forgetting and retention metrics, including scenarios involving imaging devices and anatomical outliers. This work establishes machine unlearning as a modular, practical alternative to retraining for real-world model maintenance in clinical applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。