Invention Grant
- Patent Title: Neural network training in a distributed system
-
Application No.: US16588603Application Date: 2019-09-30
-
Publication No.: US11941528B2Publication Date: 2024-03-26
- Inventor: Vignesh Vivekraja , Thiam Khean Hah , Randy Renfu Huang , Ron Diamant , Richard John Heaton
- Applicant: Amazon Technologies, Inc.
- Applicant Address: US WA Seattle
- Assignee: Amazon Technologies, Inc.
- Current Assignee: Amazon Technologies, Inc.
- Current Assignee Address: US WA Seattle
- Agency: Kilpatrick Townsend & Stockton LLP
- Main IPC: G06N3/084
- IPC: G06N3/084 ; G06N3/063 ; G06N3/045 ; G06N3/10

Abstract:
Methods and systems for performing a training operation of a neural network are provided. In one example, a method comprises: performing backward propagation computations for a second layer of a neural network to generate second weight gradients; splitting the second weight gradients into portions; causing a hardware interface to exchange a first portion of the second weight gradients with the second computer system; performing backward propagation computations for a first layer of the neural network to generate first weight gradients when the exchange of the first portion of the second weight gradients is underway, the first layer being a lower layer than the second layer in the neural network; causing the hardware interface to transmit the first weight gradients to the second computer system; and causing the hardware interface to transmit the remaining portions of the second weight gradients to the second computer system.
Public/Granted literature
- US20210097396A1 NEURAL NETWORK TRAINING IN A DISTRIBUTED SYSTEM Public/Granted day:2021-04-01
Information query