@inproceedings{a810b1084bbe47589b32159b4bde1401,
title = "Plum: Exploration and Prioritization of Model Repair Strategies for Fixing Deep Learning Models",
abstract = "The accuracy of DL models may not meet the user's expectations. To tackle this problem, existing work proposed diverse approaches, such as using more optimized training processes and training samples to evolve the model structure or parameters of such faulty DL models. In this paper, we present Plum, a novel hyperheuristic approach to fixing deep learning models. Plum generates a set of DL model candidates by applying low-level repair strategies. It then evaluates and prioritizes repair strategies based on their overall fixing effects exhibited by these model candidates and outputs a fixed DL model by applying the top-ranked repair strategy. We also formulate a novel repair strategy to show the compatibility of Plum in incorporating new repair strategies. The experiment on five DL models showed that Plum achieved improvements in test accuracy by 2.49\% and 3.11\% on the CIFAR-10 and CIFAR-100 datasets over the baselines and outperformed Apricot and MODE, two previous state-of-the-art deep learning repair techniques.",
keywords = "Debugging, Deep neural networks, hyperheuristic, Model evolution, Model repair, Strategy prioritization",
author = "Hao Zhang and W.K. Chan",
year = "2021",
doi = "10.1109/DSA52907.2021.00023",
language = "English",
isbn = "978-1-6654-4392-0",
series = "Proceedings - International Conference on Dependable Systems and Their Applications",
publisher = "IEEE",
pages = "140--151",
booktitle = "Proceedings - 2021 8th International Conference on Dependable Systems and Their Applications, DSA 2021",
address = "United States",
note = "8th International Conference on Dependable Systems and Their Applications, DSA 2021 ; Conference date: 11-09-2021 Through 12-09-2021",
}