REGION METRICS FOR CLASS BALANCING IN MACHINE LEARNING SYSTEMS

    公开(公告)号:US20220092366A1

    公开(公告)日:2022-03-24

    申请号:US17478177

    申请日:2021-09-17

    Abstract: Techniques are disclosed for an image understanding system comprising a machine learning system that applies a machine learning model to perform image understanding of each pixel of an image, the pixel labeled with a class, to determine an estimated class to which the pixel belongs. The machine learning system determines, based on the classes with which the pixels are labeled and the estimated classes, a cross entropy loss of each class. The machine learning system determines, based on one or more region metrics, a weight for each class and applies the weight to the cross entropy loss of each class to obtain a weighted cross entropy loss. The machine learning system updates the machine learning model with the weighted cross entropy loss to improve a performance metric of the machine learning model for each class.

Patent Agency Ranking