论文标题

培训可再现的深度学习模型

Towards Training Reproducible Deep Learning Models

论文作者

Chen, Boyuan, Wen, Mingzhi, Shi, Yong, Lin, Dayi, Rajbahadur, Gopi Krishnan, Ming, Zhen, Jiang

论文摘要

可重复性是人工智能(AI)越来越关注的问题,尤其是在深度学习领域(DL)。能够重现DL模型对于基于AI的系统至关重要,因为它与培训,测试,调试和审计等各种任务密切相关。但是,由于软件中的随机性(例如DL算法)和硬件中的非确定性(例如GPU),DL模型要挑战要复制。有各种各样的做法来减轻上述问题。但是,其中许多要么太干扰性,要么只能在特定的用法上下文中工作。在本文中,我们提出了一种系统的方法来培训可再现的DL模型。 Our approach includes three main parts: (1) a set of general criteria to thoroughly evaluate the reproducibility of DL models for two different domains, (2) a unified framework which leverages a record-and-replay technique to mitigate software-related randomness and a profile-and-patch technique to control hardware-related non-determinism, and (3) a reproducibility guideline which explains the rationales and the mitigation strategies on conducting DL模型可重现的培训过程。案例研究结果表明,我们的方法可以成功再现六个开源量和1个商业DL模型。

Reproducibility is an increasing concern in Artificial Intelligence (AI), particularly in the area of Deep Learning (DL). Being able to reproduce DL models is crucial for AI-based systems, as it is closely tied to various tasks like training, testing, debugging, and auditing. However, DL models are challenging to be reproduced due to issues like randomness in the software (e.g., DL algorithms) and non-determinism in the hardware (e.g., GPU). There are various practices to mitigate some of the aforementioned issues. However, many of them are either too intrusive or can only work for a specific usage context. In this paper, we propose a systematic approach to training reproducible DL models. Our approach includes three main parts: (1) a set of general criteria to thoroughly evaluate the reproducibility of DL models for two different domains, (2) a unified framework which leverages a record-and-replay technique to mitigate software-related randomness and a profile-and-patch technique to control hardware-related non-determinism, and (3) a reproducibility guideline which explains the rationales and the mitigation strategies on conducting a reproducible training process for DL models. Case study results show our approach can successfully reproduce six open source and one commercial DL models.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源