公开(公告)号：US20250077861A1

公开(公告)日：2025-03-06

申请号：US18574995

申请日：2021-11-03

Applicant: Intel Corporation

Inventor： Haihao SHEN ,  Feng TIAN ,  Xi CHEN ,  Huma ABIDI ,  Yuwen ZHOU

IPC: G06N3/08 ,  G06N3/0495 ,  G06V10/774

Abstract: The disclosure provides an apparatus, method, device and medium for label-balanced calibration in post-training quantization of DNNs. An apparatus includes interface circuitry configured to receive a training dataset and processor circuitry coupled to the interface circuitry. The processor circuitry is configured to generate a small ground truth dataset by selecting images with a ground truth number of 1 from the training dataset; generate a calibration dataset randomly from the training dataset; if any image in the calibration dataset has the ground truth number of 1, remove the image from the small ground truth dataset; generate a label balanced calibration dataset by replacing an image with a ground truth number greater than a preset threshold in the calibration dataset with a replacing image selected randomly from the small ground truth dataset; and perform calibration using the label balanced calibration dataset in post-training quantization. Other embodiments are disclosed and claimed.