| CPC H04N 19/126 (2014.11) [G06N 3/045 (2023.01); G06N 3/084 (2013.01); G06T 3/4046 (2013.01); G06T 9/002 (2013.01); G06V 10/774 (2022.01); H04N 19/13 (2014.11)] | 9 Claims |

|
1. A computer implemented method of training a first neural network and a second neural network based on training images in which each respective training image includes human scored data relating to a perceived level of distortion in the respective training image as evaluated by a group of humans, the first and second neural networks being for use in lossy image or video compression, transmission and decoding, the method including the steps of:
(i) generating a set of training images by forward passing a set of images through a trained AI-based compression pipeline, each training image having one or more artefacts representative of artefacts introduced by the trained AI-based compression pipeline, and by human scoring each generated training image based on a perceived level of distortion;
(ii) receiving an input training image from the generated set of training images;
(iii) encoding the input training image using the first neural network, to produce a latent representation;
(iv) quantizing the latent representation to produce a quantized latent;
(v) using the second neural network to produce an output image from the quantized latent, wherein the output image is an approximation of the input image;
(vi) evaluating a loss function based on differences between the output image and the input training image;
(vii) evaluating a gradient of the loss function;
(viii) back-propagating the gradient of the loss function through the second neural network and through the first neural network, to update weights of the second neural network and of the first neural network;
(ix) repeating steps (i) to (viii) using a set of training images, to produce a trained first neural network and a trained second neural network; and
(x) storing the weights of the trained first neural network and of the trained second neural network;
wherein the loss function includes a weighted sum of a rate term and a distortion term, and wherein the distortion term is a function based on human scored data of the respective training image from the generated set of training images.
|