用生成对抗样本增强小数据模型,提升骨骼识别系统抗攻击能力。
Attack Assessment and Augmented Identity Recognition for Human Skeleton Data
- 结合真实与生成数据,评估并加固小样本骨骼识别模型
- 在多种未见攻击下实现更强鲁棒性,准确率不下降
- 适合安全场景中数据稀缺的生物识别系统
针对安全应用中基于小规模LiDAR骨骼数据训练的机器学习模型易受对抗攻击的问题,本文提出Attack-AAIRS,在原有AAIRS框架基础上增强模型鲁棒性。该方法利用少量真实数据与GAN生成的合成数据,学习并生成针对HCN-ID模型弱点的对抗样本,通过这些攻击样本扩充训练集以实现模型“免疫”。十折交叉验证显示,Attack-AAIRS显著提升了对FGSM、PGD、加性高斯噪声、MI-FGSM和BIM等未见攻击的防御能力。同时,生成攻击样本的质量评分与原始良性合成样本相当,且接种后模型在真实测试集上保持与原始模型一致的准确率,证明该方法可在不牺牲性能前提下有效增强模型安全性。
原文摘要 · Abstract (English)
Machine learning models trained on small data sets for security applications are especially vulnerable to adversarial attacks. Person identification from LiDAR based skeleton data requires time consuming and expensive data acquisition for each subject identity. Recently, Assessment and Augmented Identity Recognition for Skeletons (AAIRS) has been used to train Hierarchical Co-occurrence Networks for Person Identification (HCN-ID) with small LiDAR based skeleton data sets. However, AAIRS does not evaluate robustness of HCN-ID to adversarial attacks or inoculate the model to defend against such attacks. Popular perturbation-based approaches to generating adversarial attacks are constrained to targeted perturbations added to real training samples, which is not ideal for inoculating models with small training sets. Thus, we propose Attack-AAIRS, a novel addition to the AAIRS framework. Attack-AAIRS leverages a small real data set and a GAN generated synthetic data set to assess and improve model robustness against unseen adversarial attacks. Rather than being constrained to perturbations of limited real training samples, the GAN learns the distribution of adversarial attack samples that exploit weaknesses in HCN-ID. Attack samples drawn from this distribution augment training for inoculation of the HCN-ID to improve robustness. Ten-fold cross validation of Attack-AAIRS yields increased robustness to unseen attacks- including FGSM, PGD, Additive Gaussian Noise, MI-FGSM, and BIM. The HCN-ID Synthetic Data Quality Score for Attack-AAIRS indicates that generated attack samples are of similar quality to the original benign synthetic samples generated by AAIRS. Furthermore, inoculated models show consistent final test accuracy with the original model trained on real data, demonstrating that our method improves robustness to adversarial attacks without reducing test performance on real data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。