- 专利标题: Mixed precision capable hardware for tuning a machine learning model
-
申请号: US16558536申请日: 2019-09-03
-
公开(公告)号: US11604647B2公开(公告)日: 2023-03-14
- 发明人: Xiao Sun , Chia-Yu Chen , Naigang Wang , Jungwook Choi , Kailash Gopalakrishnan
- 申请人: International Business Machines Corporation
- 申请人地址: US NY Armonk
- 专利权人: International Business Machines Corporation
- 当前专利权人: International Business Machines Corporation
- 当前专利权人地址: US NY Armonk
- 代理机构: Ryan, Mason & Lewis, LLP
- 代理商 Daniel Yeates
- 主分类号: G06F17/18
- IPC分类号: G06F17/18 ; G06N20/00 ; G06N5/04 ; G06F9/30 ; G06F7/483
摘要:
An apparatus includes a memory and a processor coupled to the memory. The processor includes first and second sets of arithmetic units having first and second precision for floating-point computations, the second precision being lower than the first precision. The processor is configured to obtain a machine learning model trained in the first precision, to utilize the second set of arithmetic units to perform inference on input data, to utilize the first set of arithmetic units to generate feedback for updating parameters of the second set of arithmetic units based on the inference performed on the input data by the second set of arithmetic units, to tune parameters of the second set of arithmetic units based at least in part on the feedback generated by the first set of arithmetic units, and to utilize the second set of arithmetic units with the tuned parameters to generate inference results.
公开/授权文献
信息查询