paperbot · PL 论文追踪

全部论文 待读 收藏

P2 ReMoS

ReMoS:减少迁移学习中缺陷继承的相关模型切片技术
ICSE Proceedings of the 44th International Conference on Software Engineering2022匹配度 3/10 · 论文研究迁移学习中的模型切片与缺陷继承,虽涉及“切片”和ML系统安全,但与用户核心方向(面向ML/Agentic程序的静态分析、LLM与程序分析/形式化验证交叉、浮点或贝叶斯程序分析)关联较弱,且不涉及LLM/Agent。引用 55 inbox
Ziqi Zhang, Yuanchun Li, Jindong Wang, Bingyan Liu, Ding Li, Yao Guo, Xiangqun Chen, Yunxin Liu
crossref: 2026-09-12dblp: 2026-09-12

AI 速览

一句话:提出 ReMoS,通过基于神经元覆盖的相关模型切片,在迁移学习微调中减少教师模型缺陷继承。

迁移学习允许开发者基于预训练教师模型构建自定义学生模型,但教师模型中的对抗脆弱性、后门等缺陷也可能被学生继承。减少此类缺陷很困难,因为学生模型通常不知道教师模型如何训练或是否被攻击。本文提出 ReMoS,一种相关模型切片技术,用于在迁移学习中减少缺陷继承并保留教师模型中有用知识。ReMoS 通过在学生任务上剖析教师模型获得神经元覆盖信息,计算与任务相关的模型切片(模型权重的子集),仅用该切片微调学生模型,而对无关权重从头重训练,以降低继承缺陷的风险。实验覆盖 7 种 DNN 缺陷、4 个 DNN 模型和 8 个数据集,结果表明 ReMoS 能有效减少继承缺陷:CV 任务降低 63% 至 86%,NLP 任务降低 40% 至 61%,同时平均仅损失约 3% 的准确率。

问题
迁移学习复用预训练教师模型时,教师模型的对抗脆弱性和后门等缺陷可能被学生模型继承,而学生通常无法获知教师的训练或攻击细节。
方法
ReMoS 在学生任务上剖析教师模型以获取神经元覆盖信息,据此计算与任务相关的模型权重切片,仅微调该切片,并对无关权重从头重训练。
评估
在 7 种 DNN 缺陷、4 个 DNN 模型和 8 个数据集上实验,评估缺陷继承减少效果与准确率代价。
原文摘要(Abstract)

Transfer learning is a popular software reuse technique in the deep learning community that enables developers to build custom models (students) based on sophisticated pretrained models (teachers). However, like vulnerability inheritance in traditional software reuse, some defects in the teacher model may also be inherited by students, such as well-known adversarial vulnerabilities and backdoors. Reducing such defects is challenging since the student is unaware of how the teacher is trained and/or attacked. In this paper, we propose ReMoS, a relevant model slicing technique to reduce defect inheritance during transfer learning while retaining useful knowledge from the teacher model. Specifically, ReMoS computes a model slice (a subset of model weights) that is relevant to the student task based on the neuron coverage information obtained by profiling the teacher model on the student task. Only the relevant slice is used to finetune the student model, while the irrelevant weights are retrained from scratch to minimize the risk of inheriting defects. Our experiments on seven DNN defects, four DNN models, and eight datasets demonstrate that ReMoS can reduce inherited defects effectively (by 63% to 86% for CV tasks and by 40% to 61% for NLP tasks) and efficiently with minimal sacrifice of accuracy (3% on average).

链接与引用

DOI 原文 · DBLP

BibTeX
@inproceedings{ZhangLWLL0CL22,
  title = {ReMoS},
  author = {Ziqi Zhang and Yuanchun Li and Jindong Wang and Bingyan Liu and Ding Li and Yao Guo and Xiangqun Chen and Yunxin Liu},
  booktitle = {ICSE},
  year = {2022},
  doi = {10.1145/3510003.3510191}
}