跳到主要导航 跳到搜索 跳到主要内容

Defect Detection in Deep Learning Model Compilers for LLM-Generated Computation Graphs

投稿的翻译标题: LLM 生成计算图的深度学习模型编译器缺陷检测
  • Limin Pan
  • , Zhiyang Zhao
  • , Siyuan Shao
  • , Senlin Luo*
  • , Haoran Zhang
  • *此作品的通讯作者

科研成果: 期刊稿件文章同行评审

摘要

Defects in deep learning model compilers risk model inference crashes, compromising deployment security and usability. Current defect detection methods suffer from inadequate code-line coverage and limited di-versity in detectable defect types. Existing approaches rely on local operator constraints for detection, failing to trigger defects caused by multi-operator interactions, while semantic-preserving mutation strategies restrict the operator types in computation graph nodes, resulting in insufficient code-line coverage and significantly redu-cing defect detection rates. In this paper, a defect detection method was proposed, which employs multi-round prompting of LLMs to construct test cases. Prompts were created to guide LLMs in generating computation graphs, after which common operators were masked and substituted with rare ones. The graphs were iteratively updated to produce diverse test cases. Experimental results on multiple deep learning model compilers demon-strate that the proposed method significantly improves code coverage and defect detection rates compared to baseline approaches, exhibiting high reliability and practical value.

投稿的翻译标题LLM 生成计算图的深度学习模型编译器缺陷检测
源语言英语
页(从-至)1204-1212
页数9
期刊Beijing Ligong Daxue Xuebao/Transaction of Beijing Institute of Technology
45
11
DOI
出版状态已出版 - 2025

指纹

探究 'LLM 生成计算图的深度学习模型编译器缺陷检测' 的科研主题。它们共同构成独一无二的指纹。

引用此