Neural network training technique
Apparatuses, systems, and techniques to train neural networks to perform image processing tasks. In at least one embodiment, one or more second neural networks are used to train one or more first neural networks based, at least in part, on a first object type in one or more images and a second object type in the one or more images, in parallel.
1 . One or more processors, comprising:
circuitry to train one or more first neural networks based, at least in part, on a comparison between one or more features identified, in one or more portions of synthetic data, by the one or more first neural networks and one or more features identified, in the one or more portions of synthetic data, by one or more pre-trained second neural networks, wherein the one or more features identified by the one or more first neural networks include a first object type in one or more images, the one or more features identified by the one or more pre-trained second neural networks include a second object type in the one or more images, and wherein the second object type is selected to be a negative sample of the first object type to reduce overfitting during training of the one or more first neural networks on positive samples of the first object type.
2 . The one or more processors of claim 1 , wherein the synthetic data includes one or more synthetic images.
3 . The one or more processors of claim 1 , wherein the one or more pre-trained second neural networks are pre-trained based on non-synthetic data.
4 . The one or more processors of claim 1 , wherein the one or more pre-trained second neural networks are frozen during training of the first one or more neural networks.
5 . The one or more processors of claim 1 , the circuitry to at least:
use the one or more first neural networks to generate a first embedding of an image;
use the one or more pre-trained second neural networks to generate a second embedding of the image; and
generate a loss value component indicative of a distance between the first and second embeddings.
6 . The one or more processors of claim 1 , the circuitry to at least:
use the one or more first neural networks to generate a first embedding of a first image;
use the one or more pre-trained second neural networks to generate a second
embedding of a second image; and
generate a loss value component based, at least in part, on a distance between the first and second embeddings, wherein the loss value component is reduced based on the distance.
7 . The one or more processors of claim 1 , wherein the one or more features identified by the one or more first neural networks include a first object type in one or more images, the one or more features identified by the one or more pre-trained second neural networks include a second object type in the one or more images, and wherein the second object type is selected to be different than the first object type.
8 . The one or more processors of claim 1 , wherein the one or more first neural networks and the one or more pre-trained second neural networks generate output based, at least in part, on an attention matrix.
9 . A system, comprising:
one or more processors to train one or more first neural networks based, at least in part, on a comparison between one or more features identified, in one or more portions of synthetic data, by the one or more first neural networks and one or more features identified, in the one or more portions of synthetic data, by one or more pre-trained second neural networks, wherein the one or more features identified by the one or more first neural networks include a first object type in one or more images, the one or more features identified by the one or more pre-trained second neural networks include a second object type in the one or more images, and wherein the second object type is selected to be a negative sample of the first object type to reduce overfitting during training of the one or more first neural networks on positive samples of the first object type.
10 . The system of claim 9 , wherein the one or more processors are to select the second object type based, at least in part, on one or more differences between the second object type and the first object type.
11 . The system of claim 9 , wherein the one or more pre-trained second neural networks are pre-trained based on images of real objects.
12 . The system of claim 9 , wherein the one or more pre-trained second neural networks are frozen during a training of the first one or more neural networks.
13 . The system of claim 9 , wherein the one or more processors to cause the system to at least generate a loss value component based on a distance between an embedding of an object of a first type by the one or more first neural networks and an embedding of the object of the first type by the one or more pre-trained second neural networks.
14 . The system of claim 9 , wherein the one or more processors to cause the system to at least generate a loss value component reduced based on distance between a first embedding and a second embedding, wherein the one or more features identified by the one or more first neural networks include a first object type in one or more images, wherein the one or more features identified by the one or more pre-trained second neural networks include a second object type in the one or more images, wherein the first embedding is generated by the one or more first neural networks based on a first image comprising a depiction of an object of the first object type, and wherein the second embedding is generated by the one or more pre-trained second neural networks based on a second image comprising a depiction of an object of the second object type.
15 . The system of claim 9 , wherein the one or more first neural networks and the one or more pre-trained second neural networks generate output based, at least in part, on an attention matrix.
16 . A non-transitory machine-readable medium having stored thereon a set of instructions, which cause one or more processors to at least:
train one or more first neural networks based, at least in part, on a comparison between one or more features identified, in one or more portions of synthetic data, by the one or more first neural networks and one or more features identified, in the one or more portions of synthetic data, by one or more pre-trained second neural networks, wherein the one or more features identified by the one or more first neural networks include a first object type in one or more images, the one or more features identified by the one or more pre-trained second neural networks include a second object type in the one or more images, and wherein the second object type is selected to be a negative sample of the first object type to reduce overfitting during training of the one or more first neural networks on positive samples of the first object type.
17 . The non-transitory machine-readable medium of claim 16 , having stored thereon a further set of instructions, which cause one or more processors to at least:
use the one or more pre-trained second neural networks to train the one or more first neural networks based, at least in part, on a first object type in one or more images and a second object type in the one or more images; and
select the second object type based, at least in part, on one or more differences between the second object type and the first object type.
18 . The non-transitory machine-readable medium of claim 16 , wherein the one or more pre-trained second neural networks are pre-trained based on images of real objects and frozen during training of the one or more first neural networks.
19 . The non-transitory machine-readable medium of claim 16 , having stored thereon a further set of instructions, which cause one or more processors to at least:
generate, by the one or more first neural networks, a first embedding of an image generate, by the one or more pre-trained second neural networks, a second embedding of the image; and
generate a loss value based, at least in part, on a distance between the first and second embeddings.
20 . The non-transitory machine-readable medium of claim 16 , having stored thereon a further set of instructions, which cause one or more processors to at least:
use the one or more pre-trained second neural networks to train the one or more first neural networks based, at least in part, on a first object type in one or more images and a second object type in the one or more images;
generate, by the one or more first neural networks, a first embedding of a first image of the one or more images, the first image comprising the first object type;
generate, by the one or more pre-trained second neural networks, a second embedding of a second image of the one or more images, the second image comprising the second object type;
and generate a loss value based, at least in part, on a distance between the first and second embeddings.
21 . The non-transitory machine-readable medium of claim 16 , wherein the one or more first neural networks are trained based, at least in part, on a contrastive loss signal.
22 . The non-transitory machine-readable medium of claim 16 , wherein the one or more first neural networks and the one or more pre-trained second neural networks generate output based, at least in part, on an attention matrix.
23 . A computing device, comprising:
one or more processors to perform an image processing task based, at least in part, on one or more first neural networks trained using one or more pre-trained second neural networks, based, at least in part on a comparison between one or more features identified, in one or more portions of synthetic data, by the one or more first neural networks and one or more features identified, in the one or more portions of synthetic data, by one or more pre-trained second neural networks, wherein the one or more features identified by the one or more first neural networks include a first object type in one or more images, the one or more features identified by the one or more pre-trained second neural networks include a second object type in the one or more images, and wherein the second object type is selected to be a negative sample of the first object type to reduce overfitting during training of the one or more first neural networks on positive samples of the first object type.
24 . The computing device of claim 23 , wherein the image processing task comprises recognition of an object.
25 . The computing device of claim 23 , wherein the one or more processors to use the one or more images in parallel to generate a loss signal for training the one or more first neural networks.
26 . The computing device of claim 23 , wherein the one or more pre-trained second neural networks are pre-trained based on images of real objects and frozen during a training of the one or more first neural networks.
27 . The computing device of claim 23 , wherein the one or more processors to cause the computing device to at least generate a loss value component increased in proportion to a distance between an embedding of an object of a first type by the one or more first neural networks and an embedding of the object of the first type by the one or more pre-trained second neural networks.
28 . The computing device of claim 23 , wherein the one or more processors to cause the computing device to at least generate a loss value component reduced in proportion to a distance between a first embedding and a second embedding.
29 . The computing device of claim 23 , wherein the one or more first neural networks and the one or more pre-trained second neural networks generate output based, at least in part, on an attention matrix.