Shapley-Guided Neural Repair Approach via Derivative-Free Optimization

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Sun, Xinyu, Liu, Wanwei, Chi, Haoang, Chen, Tingyu, Mao, Xiaoguang, Wang, Shangwen, Bu, Lei, Wang, Jingyi, Tan, Yang, Qi, Zhenyi
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866908928569245696
author Sun, Xinyu
Liu, Wanwei
Chi, Haoang
Chen, Tingyu
Mao, Xiaoguang
Wang, Shangwen
Bu, Lei
Wang, Jingyi
Tan, Yang
Qi, Zhenyi
author_facet Sun, Xinyu
Liu, Wanwei
Chi, Haoang
Chen, Tingyu
Mao, Xiaoguang
Wang, Shangwen
Bu, Lei
Wang, Jingyi
Tan, Yang
Qi, Zhenyi
contents DNNs are susceptible to defects like backdoors, adversarial attacks, and unfairness, undermining their reliability. Existing approaches mainly involve retraining, optimization, constraint-solving, or search algorithms. However, most methods rely on gradient calculations, restricting applicability to specific activation functions (e.g., ReLU), or use search algorithms with uninterpretable localization and repair. Furthermore, they often lack generalizability across multiple properties. We propose SHARPEN, integrating interpretable fault localization with a derivative-free optimization strategy. First, SHARPEN introduces a Deep SHAP-based localization strategy quantifying each layer's and neuron's marginal contribution to erroneous outputs. Specifically, a hierarchical coarse-to-fine approach reranks layers by aggregated impact, then locates faulty neurons/filters by analyzing activation divergences between property-violating and benign states. Subsequently, SHARPEN incorporates CMA-ES to repair identified neurons. CMA-ES leverages a covariance matrix to capture variable dependencies, enabling gradient-free search and coordinated adjustments across coupled neurons. By combining interpretable localization with evolutionary optimization, SHARPEN enables derivative-free repair across architectures, being less sensitive to gradient anomalies and hyperparameters. We demonstrate SHARPEN's effectiveness on three repair tasks. Balancing property repair and accuracy preservation, it outperforms baselines in backdoor removal (+10.56%), adversarial mitigation (+5.78%), and unfairness repair (+11.82%). Notably, SHARPEN handles diverse tasks, and its modular design is plug-and-play with different derivative-free optimizers, highlighting its flexibility.
format Preprint
id arxiv_https___arxiv_org_abs_2604_00422
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Shapley-Guided Neural Repair Approach via Derivative-Free Optimization
Sun, Xinyu
Liu, Wanwei
Chi, Haoang
Chen, Tingyu
Mao, Xiaoguang
Wang, Shangwen
Bu, Lei
Wang, Jingyi
Tan, Yang
Qi, Zhenyi
Software Engineering
Machine Learning
DNNs are susceptible to defects like backdoors, adversarial attacks, and unfairness, undermining their reliability. Existing approaches mainly involve retraining, optimization, constraint-solving, or search algorithms. However, most methods rely on gradient calculations, restricting applicability to specific activation functions (e.g., ReLU), or use search algorithms with uninterpretable localization and repair. Furthermore, they often lack generalizability across multiple properties. We propose SHARPEN, integrating interpretable fault localization with a derivative-free optimization strategy. First, SHARPEN introduces a Deep SHAP-based localization strategy quantifying each layer's and neuron's marginal contribution to erroneous outputs. Specifically, a hierarchical coarse-to-fine approach reranks layers by aggregated impact, then locates faulty neurons/filters by analyzing activation divergences between property-violating and benign states. Subsequently, SHARPEN incorporates CMA-ES to repair identified neurons. CMA-ES leverages a covariance matrix to capture variable dependencies, enabling gradient-free search and coordinated adjustments across coupled neurons. By combining interpretable localization with evolutionary optimization, SHARPEN enables derivative-free repair across architectures, being less sensitive to gradient anomalies and hyperparameters. We demonstrate SHARPEN's effectiveness on three repair tasks. Balancing property repair and accuracy preservation, it outperforms baselines in backdoor removal (+10.56%), adversarial mitigation (+5.78%), and unfairness repair (+11.82%). Notably, SHARPEN handles diverse tasks, and its modular design is plug-and-play with different derivative-free optimizers, highlighting its flexibility.
title Shapley-Guided Neural Repair Approach via Derivative-Free Optimization
topic Software Engineering
Machine Learning
url https://arxiv.org/abs/2604.00422