Invention Grant
- Patent Title: Auto weight scaling for RPUs
-
Application No.: US16427598Application Date: 2019-05-31
-
Publication No.: US11556770B2Publication Date: 2023-01-17
- Inventor: Malte Rasch , Tayfun Gokmen
- Applicant: International Business Machines Corporation
- Applicant Address: US NY Armonk
- Assignee: International Business Machines Corporation
- Current Assignee: International Business Machines Corporation
- Current Assignee Address: US NY Armonk
- Agency: Michael J. Chang, LLC
- Agent James Olsen
- Main IPC: G06N3/063
- IPC: G06N3/063 ; G06N3/08 ; G06F7/52 ; G06F7/02 ; G06N3/04

Abstract:
Techniques for auto weight scaling a bounded weight range of RPU devices with the size of the array during ANN training are provided. In one aspect, a method of ANN training includes: initializing weight values winit in the array to a random value, wherein the array represents a weight matrix W with m rows and n columns; calculating a scaling factor β based on a size of the weight matrix W; providing digital inputs x to the array; dividing the digital inputs x by a noise and bound management factor α to obtain adjusted digital inputs x′; performing a matrix-vector multiplication of the adjusted digital inputs x′ with the array to obtain digital outputs y′; multiplying the digital outputs y′ by the noise and bound management factor α; and multiplying the digital outputs y′ by the scaling factor β to provide digital outputs y of the array.
Public/Granted literature
- US20200380349A1 Auto Weight Scaling for RPUs Public/Granted day:2020-12-03
Information query