Learned image compression and decompression using long and short attention module

Abstract

A method for feature extraction using a neural network includes: extracting a first set of features from an input set of features by processing said input set of features through at least two residual blocks of the neural network connected successively to each other; extracting a second set of features from the input set of features, said extracting the second set of features comprising the steps of: extracting a third set of features and a fourth set of features from the input set of features by a non-local attention processing, implementing a group convolution by a multi-head attention mechanism to the input set of features, the third set of features and the fourth set of features to obtain the second set of features.

Dependencies are inferred from the claim text.