一句话:提出 BGMD 利用特征偏斜与代表性不足特征高效诊断误预测,并用 MAPS 重加权采样修复模型。
解释和调试机器学习模型对确保鲁棒性至关重要。现有误预测解释技术通常从全部特征“盲目”推导规则,特征多时难以扩展。本文提出 BGMD,利用两类先验:数据特征分布高度偏斜,以及模型在代表性不足特征子集上表现较差,从而高效诊断误预测。进一步提出 MAPS,在重训练时提高易误预测子集权重,使模型关注代表性不足特征。实证显示,BGMD 优于 SOTA 误预测诊断器,诊断时间减少 92%;MAPS 在修复误预测数据性能上优于两种 SOTA,且不损害全量数据性能。研究工件已公开。
Interpreting and debugging machine learning models is necessary to ensure the robustness of the machine learning models. Explaining mispredictions can help significantly in doing so. While recent works on misprediction explanation have proven promising in generating interpretable explanations for mispredictions, the state-of-the-art techniques “blindly” deduce misprediction explanation rules from all data features, which may not be scalable depending on the number of features. To alleviate this problem, we propose an efficient misprediction explanation technique named Bias Guided Misprediction Diagnoser (BGMD), which leverages two prior knowledge about data: a) data often exhibit highly-skewed feature distributions and b) trained models in many cases perform poorly on subdataset with under-represented features. Next, we propose a technique named MAPS (Mispredicted Area UPweight Sampling). MAPS increases the weights of subdataset during model retraining that belong to the group that is prone to be mispredicted because of containing under-represented features. Thus, MAPS make retrained model pay more attention to the under-represented features. Our empirical study shows that our proposed BGMD outperformed the state-of-the-art misprediction diagnoser and reduces diagnosis time by 92%. Furthermore, MAPS outperformed two state-of-the-art techniques on fixing the machine learning model's performance on mispredicted data without compromising performance on all data. All the research artifacts (i.e., tools, scripts, and data) of this study are available in the accompanying website [1].
@inproceedings{GesiSGCA23,
title = {Leveraging Feature Bias for Scalable Misprediction Explanation of Machine Learning Models},
author = {Jiri Gesi and Xinyun Shen and Yunfan Geng and Qihong Chen and Iftekhar Ahmed},
booktitle = {ICSE},
year = {2023},
doi = {10.1109/icse48619.2023.00135}
}