Invention Publication
- Patent Title: APPARATUS, METHOD, DEVICE AND MEDIUM FOR LOSS BALANCING IN MULTI-TASK LEARNING
-
Application No.: US18571616Application Date: 2021-12-02
-
Publication No.: US20240303485A1Publication Date: 2024-09-12
- Inventor: Wenjing KANG , Xiaochuan LUO , Xianchao XU
- Applicant: Intel Corporation
- Applicant Address: US CA Santa Clara
- Assignee: Intel Corporation
- Current Assignee: Intel Corporation
- Current Assignee Address: US CA Santa Clara
- International Application: PCT/CN2021/135129 2021.12.02
- Date entered country: 2023-12-18
- Main IPC: G06N3/08
- IPC: G06N3/08

Abstract:
The disclosure provides an apparatus, method, device, and medium for loss balancing in MTL. The apparatus includes interface circuitry and processor circuitry. The processor circuitry is configured to initialize parameters of shared layers of a deep neural network for MTL using a pre-trained neural network; determine a custom interval consisting of a designated number of mini-batch training steps and a designated window of N custom intervals (N>2); for each task, calculate a loss change rate between each pair of N−1 pairs of neighboring custom intervals within a designated window prior to a present custom interval and a gradient magnitude with respect to selected shared weights within the designated window prior to the present custom interval, and adjust, a weight of the task, based on the calculated loss change rate and gradient magnitude with respect to selected shared weights.
Information query