LLM2D
FetaFix:深度学习模型转换的自动故障定位与修复
FetaFix: Automatic Fault Localization and Repair of Deep Learning Model Conversions
作者: Nikolaos Louloudakis, Perry Gibson, Jos\'e Cano, Ajitha Rajan
发布日期: 4/29/2025
arXiv ID: oai:arXiv.org:2312.15101v4

摘要

arXiv:2312.15101v4 宣告类型:替换交叉 摘要:在不同设备之间转换深度学习模型是最大化模型兼容性的常见步骤,同时还可以利用某个深度学习框架中特有的优化功能。然而,这个转换过程可能会存在bug,导致转换后的模型无法部署或存在问题,显著降低其预测准确性。 在本文中,我们提出了一种用于模型转换过程中故障定位和修复的自动化方法FetaFix。FetaFix能够检测和修复在模型输入、参数、超参数以及模型图转换过程中引入的故障。FetaFix利用一套从代码仓库和论坛中调研的常见转换问题中挖掘出的故障类型,在转换后的目标模型中定位潜在的转换故障,然后对其进行适当的修复,例如用源模型的参数替换目标模型的参数。这一过程是迭代的,针对数据集中的每一张图片,将源模型和转换后的目标模型的输出标签差异进行比较,直至所有差异都被解决。我们评估了FetaFix在四个不同深度学习框架中转换的三种广泛使用的图像识别模型中的故障修复效果。总体而言,FetaFix成功修复了检测到的755个转换故障中的462个,并且显著提高了14个出错的转换案例中的15个案例的性能。