一句话:提出 FreeFuzz,通过挖掘开源代码追踪动态信息来模糊测试深度学习库,发现 49 个缺陷。
深度学习库是构建和运行模型的基础,但其公共 API 主要暴露于 Python,动态类型使自动确定输入参数类型困难,底层库测试工作有限。论文提出 FreeFuzz,首个通过挖掘开源资源模糊测试 DL 库的方法。它从库文档、开发者测试和野外模型中收集代码,插桩运行并追踪每个 API 的参数类型与值、输入/输出张量形状等动态信息,据此进行模糊测试。在 PyTorch 和 TensorFlow 上,FreeFuzz 追踪并测试了 1158 个热门 API,比 LEMON 多 9 倍、开销低 3.5 倍;发现 49 个 bug,其中 38 个被开发者确认为此前未知。
Deep learning (DL) systems can make our life much easier, and thus are gaining more and more attention from both academia and industry. Meanwhile, bugs in DL systems can be disastrous, and can even threaten human lives in safety-critical applications. To date, a huge body of research efforts have been dedicated to testing DL models. However, interestingly, there is still limited work for testing the underlying DL libraries, which are the foundation for building, optimizing, and running DL models. One potential reason is that test generation for the underlying DL libraries can be rather challenging since their public APIs are mainly exposed in Python, making it even hard to automatically determine the API input parameter types due to dynamic typing. In this paper, we propose FreeFuzz, the first approach to fuzzing DL libraries via mining from open source. More specifically, FreeFuzz obtains code/models from three different sources: 1) code snippets from the library documentation, 2) library developer tests, and 3) DL models in the wild. Then, FreeFuzz automatically runs all the collected code/models with instrumentation to trace the dynamic information for each covered API, including the types and values of each parameter during invocation, and shapes of input/output tensors. Lastly, FreeFuzz will leverage the traced dynamic information to perform fuzz testing for each covered API. The extensive study of FreeFuzz on PyTorch and TensorFlow, two of the most popular DL libraries, shows that FreeFuzz is able to automatically trace valid dynamic information for fuzzing 1158 popular APIs, 9X more than state-of-the-art LEMON with 3.5X lower overhead than LEMON. To date, FreeFuzz has detected 49 bugs for PyTorch and TensorFlow (with 38 already confirmed by developers as previously unknown).