Patent ID: 11954881
Assignee: APPLE INC.
Field: Computer technology (Electrical engineering)
Classification: CPC G | IPC G

Claim 16:
17. A system comprising:
a non-transitory computer-readable storage medium; and
one or more processors coupled to the non-transitory computer-readable storage medium, wherein the non-transitory computer-readable storage medium comprises program instructions that, when executed on the one or more processors, cause the system to perform operations comprising:
obtaining a training set of training inputs and corresponding training labels, the training labels identifying known locations of parts of objects in the training inputs;
inputting the training inputs into a main task neural network to produce output labels predicting locations of the parts of the objects in the training inputs;
inputting data from pooled layers of the main task neural network into a clustering neural network to identify clusters comprising a first group of features in a first subset of the pooled layers that correspond with a first part of at least one object of the objects and a second group of features in a second subset of the pooled layers that correspond with a second part of the at least one object of the objects, wherein the first subset is different than the second subset; and
training the main task neural network and the clustering neural network based on a main task loss from the main task neural network and a clustering loss from the clustering neural network.