Machine learning device and method

    公开(公告)号:US12159411B2

    公开(公告)日:2024-12-03

    申请号:US16996871

    申请日:2020-08-18

    Abstract: Provided is a machine learning device and method that enables machine learning of labeling, in which a plurality of labels are attached to volume data at one effort with excellent accuracy, using training data having label attachment mixed therein.
    A probability calculation unit (14) calculates a value (soft label) indicating a likelihood of labeling of a class Ci for each voxel of a second slice image by means of a learned teacher model (13a). A detection unit (15) detects “bronchus” and “blood vessel” for the voxels of the second slice image using a known method, such as a region expansion method and performs labeling of “bronchus” and “blood vessel”. A correction probability setting unit (16) replaces the soft label with a hard label of “bronchus” or “blood vessel” detected by the detection unit (15). A distillation unit (17) performs distillation of a student model (18a) from the teacher model (13a) using the soft label after correction by means of the correction probability setting unit (16). With this, the learned student model (18a) is obtained.

    Machine learning device and method

    公开(公告)号:US11823375B2

    公开(公告)日:2023-11-21

    申请号:US17017647

    申请日:2020-09-10

    Inventor: Deepak Keshwani

    Abstract: Provided are a machine learning device and a method capable of performing machine learning of labeling for accurately attaching a plurality of labels to volume data at once by using learning data with mixed inconsistent labeling. A neural network (14) receives an input of multi-slice images of learning data Di (i=1, 2, . . . n) of which a class to be labeled is n types, and creates a prediction mask of n anatomical structures i by a convolutional neural network (CNN) or the like (S1). A machine learning unit (13) calculates a prediction accuracy acc(i) of the class corresponding to the learning data Di for each learning data Di (S2). The machine learning unit (13) calculates a weighted average M of an error di between the prediction accuracy acc(i) and a ground truth mask Gi. The machine learning unit (13) calculates a learning loss by a loss function Loss (S4). The machine learning unit (13) changes each coupling load of the neural network (14) from an output layer side to an input layer side according to a value of the learning loss calculated by the loss function Loss (S5).

Patent Agency Ranking