arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.07539cs.LGcs.CV

面向医学影像的机器遗忘

Machine Unlearning for Medical Imaging

Helmholtz Munich(亥姆霍兹慕尼黑中心);Technical University of Munich(慕尼黑工业大学);King’s College London(伦敦国王学院);Imperial College London(伦敦帝国学院)

Reza Nasirigerdeh, Nader Razmi, Julia A. Schnabel, Daniel Rueckert, Georgios Kaissis

AI总结:

本研究评估不同机器遗忘算法在医学影像领域的效果与效率,发现其在保留集、遗忘集表现良好且无性别偏见,但会损害模型泛化能力、存在样本难度偏见且有额外调参开销,为算法改进指明方向。

摘要:

机器遗忘是指从预训练模型中移除特定训练样本集影响的过程,其旨在落实“被遗忘权”——该权利赋予患者等个体重新考量自身对医学影像模型等模型所做贡献的权利。本研究评估了不同遗忘算法在医学影像领域的有效性(性能)与计算效率。评估结果显示,所考察的遗忘算法在保留集(允许保留其对模型影响的样本)和遗忘集(应消除其对模型贡献的样本)上表现良好,且对男性或女性样本无偏见。然而,这些算法会对模型的泛化能力产生负面影响,尤其是当遗忘集规模较大时。此外,它们可能对简单或困难样本存在偏见,且超参数调优需要额外的计算开销。综上,机器遗忘在医学影像领域颇具前景,但现有遗忘算法仍需进一步改进,才能更切实地应用于医疗场景。

英文摘要:

Machine unlearning is the process of removing the impact of a particular set of training samples from a pretrained model. It aims to fulfill the "right to be forgotten", which grants the individuals such as patients the right to reconsider their contribution in models including medical imaging models. In this study, we evaluate the effectiveness (performance) and computational efficiency of different unlearning algorithms in medical imaging domain. Our evaluations demonstrate that the considered unlearning algorithms perform well on the retain set (samples whose influence on the model is allowed to be retained) and forget set (samples whose contribution to the model should be eliminated), and show no bias against male or female samples. They, however, adversely impact the generalization of the model, especially for larger forget set sizes. Moreover, they might be biased against easy or hard samples, and need additional computational overhead for hyper-parameter tuning. In conclusion, machine unlearning seems promising for medical imaging, but the existing unlearning algorithms still needs further improvements to become more practical for medical applications.

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

↑