WO2022232841A1 - Block-wise content-adaptive online training in neural image compression - Google Patents
Block-wise content-adaptive online training in neural image compression Download PDFInfo
- Publication number
- WO2022232841A1 WO2022232841A1 PCT/US2022/072022 US2022072022W WO2022232841A1 WO 2022232841 A1 WO2022232841 A1 WO 2022232841A1 US 2022072022 W US2022072022 W US 2022072022W WO 2022232841 A1 WO2022232841 A1 WO 2022232841A1
- Authority
- WO
- WIPO (PCT)
- Prior art keywords
- pretrained
- neural network
- parameters
- block
- video
- Prior art date
Links
- 238000012549 training Methods 0.000 title description 65
- 230000006835 compression Effects 0.000 title description 63
- 238000007906 compression Methods 0.000 title description 63
- 230000001537 neural effect Effects 0.000 title description 10
- 238000013528 artificial neural network Methods 0.000 claims abstract description 325
- 238000000034 method Methods 0.000 claims abstract description 160
- 238000012545 processing Methods 0.000 claims abstract description 56
- 238000004422 calculation algorithm Methods 0.000 claims description 13
- 230000008569 process Effects 0.000 description 87
- 239000000523 sample Substances 0.000 description 48
- 230000033001 locomotion Effects 0.000 description 33
- 238000005516 engineering process Methods 0.000 description 31
- 239000013598 vector Substances 0.000 description 22
- 238000011176 pooling Methods 0.000 description 20
- 230000006870 function Effects 0.000 description 16
- 239000000872 buffer Substances 0.000 description 15
- 230000003044 adaptive effect Effects 0.000 description 14
- 230000005540 biological transmission Effects 0.000 description 12
- 238000013527 convolutional neural network Methods 0.000 description 12
- 238000013139 quantization Methods 0.000 description 12
- 238000004891 communication Methods 0.000 description 11
- 238000010586 diagram Methods 0.000 description 11
- 239000011159 matrix material Substances 0.000 description 9
- 230000002123 temporal effect Effects 0.000 description 8
- 230000008901 benefit Effects 0.000 description 6
- 230000007246 mechanism Effects 0.000 description 6
- 230000002093 peripheral effect Effects 0.000 description 5
- 239000013074 reference sample Substances 0.000 description 5
- 230000000007 visual effect Effects 0.000 description 5
- 241000023320 Luma <angiosperm> Species 0.000 description 4
- 238000002347 injection Methods 0.000 description 4
- 239000007924 injection Substances 0.000 description 4
- OSWPMRLSEDHDFF-UHFFFAOYSA-N methyl salicylate Chemical compound COC(=O)C1=CC=CC=C1O OSWPMRLSEDHDFF-UHFFFAOYSA-N 0.000 description 4
- 238000010606 normalization Methods 0.000 description 4
- 238000005457 optimization Methods 0.000 description 4
- 230000011664 signaling Effects 0.000 description 4
- 230000004913 activation Effects 0.000 description 3
- 230000001143 conditioned effect Effects 0.000 description 3
- 230000006837 decompression Effects 0.000 description 3
- 230000001419 dependent effect Effects 0.000 description 3
- 239000000203 mixture Substances 0.000 description 3
- 238000009877 rendering Methods 0.000 description 3
- 238000005070 sampling Methods 0.000 description 3
- 230000002457 bidirectional effect Effects 0.000 description 2
- 230000001413 cellular effect Effects 0.000 description 2
- 239000012634 fragment Substances 0.000 description 2
- 238000013507 mapping Methods 0.000 description 2
- 230000003287 optical effect Effects 0.000 description 2
- 238000007781 pre-processing Methods 0.000 description 2
- 230000000306 recurrent effect Effects 0.000 description 2
- 230000009467 reduction Effects 0.000 description 2
- 238000013403 standard screening design Methods 0.000 description 2
- 230000004075 alteration Effects 0.000 description 1
- 238000013473 artificial intelligence Methods 0.000 description 1
- 238000004364 calculation method Methods 0.000 description 1
- 230000003750 conditioning effect Effects 0.000 description 1
- 238000012937 correction Methods 0.000 description 1
- 230000008878 coupling Effects 0.000 description 1
- 238000010168 coupling process Methods 0.000 description 1
- 238000005859 coupling reaction Methods 0.000 description 1
- 238000009795 derivation Methods 0.000 description 1
- 238000013461 design Methods 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 238000002474 experimental method Methods 0.000 description 1
- 238000001914 filtration Methods 0.000 description 1
- 239000011521 glass Substances 0.000 description 1
- 230000006872 improvement Effects 0.000 description 1
- 238000010348 incorporation Methods 0.000 description 1
- 238000012886 linear function Methods 0.000 description 1
- 239000004973 liquid crystal related substance Substances 0.000 description 1
- 230000007774 longterm Effects 0.000 description 1
- 238000010801 machine learning Methods 0.000 description 1
- 238000010295 mobile communication Methods 0.000 description 1
- 238000003062 neural network model Methods 0.000 description 1
- 210000002569 neuron Anatomy 0.000 description 1
- 238000013442 quality metrics Methods 0.000 description 1
- 238000009738 saturating Methods 0.000 description 1
- 230000035945 sensitivity Effects 0.000 description 1
- 239000000779 smoke Substances 0.000 description 1
- 239000007787 solid Substances 0.000 description 1
- 230000004936 stimulating effect Effects 0.000 description 1
- 230000000153 supplemental effect Effects 0.000 description 1
- 230000009466 transformation Effects 0.000 description 1
- 238000000844 transformation Methods 0.000 description 1
Classifications
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04N—PICTORIAL COMMUNICATION, e.g. TELEVISION
- H04N19/00—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals
- H04N19/50—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using predictive coding
- H04N19/593—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using predictive coding involving spatial prediction techniques
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04N—PICTORIAL COMMUNICATION, e.g. TELEVISION
- H04N19/00—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals
- H04N19/10—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding
- H04N19/169—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding characterised by the coding unit, i.e. the structural portion or semantic portion of the video signal being the object or the subject of the adaptive coding
- H04N19/17—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding characterised by the coding unit, i.e. the structural portion or semantic portion of the video signal being the object or the subject of the adaptive coding the unit being an image region, e.g. an object
- H04N19/176—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding characterised by the coding unit, i.e. the structural portion or semantic portion of the video signal being the object or the subject of the adaptive coding the unit being an image region, e.g. an object the region being a block, e.g. a macroblock
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
- G06N3/0455—Auto-encoder networks; Encoder-decoder networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/0464—Convolutional networks [CNN, ConvNet]
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
- G06N3/084—Backpropagation, e.g. using gradient descent
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T9/00—Image coding
- G06T9/002—Image coding using neural networks
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04N—PICTORIAL COMMUNICATION, e.g. TELEVISION
- H04N19/00—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals
- H04N19/10—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding
- H04N19/102—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding characterised by the element, parameter or selection affected or controlled by the adaptive coding
- H04N19/103—Selection of coding mode or of prediction mode
- H04N19/105—Selection of the reference unit for prediction within a chosen coding or prediction mode, e.g. adaptive choice of position and number of pixels used for prediction
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04N—PICTORIAL COMMUNICATION, e.g. TELEVISION
- H04N19/00—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals
- H04N19/10—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding
- H04N19/102—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding characterised by the element, parameter or selection affected or controlled by the adaptive coding
- H04N19/13—Adaptive entropy coding, e.g. adaptive variable length coding [AVLC] or context adaptive binary arithmetic coding [CABAC]
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04N—PICTORIAL COMMUNICATION, e.g. TELEVISION
- H04N19/00—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals
- H04N19/10—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding
- H04N19/102—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding characterised by the element, parameter or selection affected or controlled by the adaptive coding
- H04N19/132—Sampling, masking or truncation of coding units, e.g. adaptive resampling, frame skipping, frame interpolation or high-frequency transform coefficient masking
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04N—PICTORIAL COMMUNICATION, e.g. TELEVISION
- H04N19/00—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals
- H04N19/10—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding
- H04N19/169—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding characterised by the coding unit, i.e. the structural portion or semantic portion of the video signal being the object or the subject of the adaptive coding
- H04N19/184—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding characterised by the coding unit, i.e. the structural portion or semantic portion of the video signal being the object or the subject of the adaptive coding the unit being bits, e.g. of the compressed video stream
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04N—PICTORIAL COMMUNICATION, e.g. TELEVISION
- H04N19/00—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals
- H04N19/50—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using predictive coding
- H04N19/503—Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using predictive coding involving temporal prediction
- H04N19/51—Motion estimation or motion compensation
- H04N19/513—Processing of motion vectors
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/044—Recurrent networks, e.g. Hopfield networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/047—Probabilistic or stochastic networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/048—Activation functions
Definitions
- the present disclosure describes embodiments generally related to video coding.
- Uncompressed digital image and/or video can include a series of pictures, each picture having a spatial dimension of, for example, 1920 x 1080 luminance samples and associated chrominance samples.
- the series of pictures can have a fixed or variable picture rate (informally also known as frame rate), of, for example 60 pictures per second or 60 Hz.
- Uncompressed image and/or video has specific bitrate requirements. For example, 1080p604:2:0 video at 8 bit per sample (1920x1080 luminance sample resolution at 60 Hz frame rate) requires close to 1.5 Gbit/s bandwidth. An hour of such video requires more than 600 GBytes of storage space.
- Video coding and decoding can be the reduction of redundancy in the input image and/or video signal, through compression. Compression can help reduce the aforementioned bandwidth and/or storage space requirements, in some cases by two orders of magnitude or more.
- Video encoding/decoding as illustrative examples, the same techniques can be applied to image encoding/decoding in similar fashion without departing from the spirit of the present disclosure. Both lossless compression and lossy compression, as well as a combination thereof can be employed. Lossless compression refers to techniques where an exact copy of the original signal can be reconstructed from the compressed original signal.
- the reconstructed signal may not be identical to the original signal, but the distortion between original and reconstructed signals is small enough to make the reconstructed signal useful for the intended application.
- lossy compression is widely employed.
- the amount of distortion tolerated depends on the application; for example, users of certain consumer streaming applications may tolerate higher distortion than users of television distribution applications.
- the compression ratio achievable can reflect that: higher allowable/tolerable distortion can yield higher compression ratios.
- a video encoder and decoder can utilize techniques from several broad categories, including, for example, motion compensation, transform, quantization, and entropy coding.
- Video codec technologies can include techniques known as intra coding.
- intra coding sample values are represented without reference to samples or other data from previously reconstructed reference pictures.
- the picture is spatially subdivided into blocks of samples.
- Intra pictures and their derivations such as independent decoder refresh pictures, can be used to reset the decoder state and can, therefore, be used as the first picture in a coded video bitstream and a video session, or as a still image.
- the samples of an intra block can be exposed to a transform, and the transform coefficients can be quantized before entropy coding.
- Intra prediction can be a technique that minimizes sample values in the pre-transform domain.
- intra prediction is using reference data only from the current picture under reconstruction and not from reference pictures.
- intra prediction There can be many different forms of intra prediction.
- the technique in use can be coded in an intra prediction mode.
- modes can have submodes and/or parameters, and those can be coded individually or included in the mode codeword.
- Which codeword to use for a given mode, submode, and/or parameter combination can have an impact in the coding efficiency gain through intra prediction, and so can the entropy coding technology used to translate the codewords into a bitstream.
- a certain mode of intra prediction was introduced with H.264, refined in H.265, and further refined in newer coding technologies such as joint exploration model (JEM), versatile video coding (VVC), and benchmark set (BMS).
- JEM joint exploration model
- VVC versatile video coding
- BMS benchmark set
- a predictor block can be formed using neighboring sample values belonging to already available samples. Sample values of neighboring samples are copied into the predictor block according to a direction. A reference to the direction in use can be coded in the bitstream or may itself be predicted.
- FIG. 1A depicted in the lower right is a subset of nine predictor directions known from H.265 ’s 33 possible predictor directions (corresponding to the 33 angular modes of the 35 intra modes).
- the point where the arrows converge (101) represents the sample being predicted.
- the arrows represent the direction from which the sample is being predicted.
- arrow (102) indicates that sample (101) is predicted from a sample or samples to the upper right, at a 45 degree angle from the horizontal.
- arrow (103) indicates that sample (101) is predicted from a sample or samples to the lower left of sample (101), in a 22.5 degree angle from the horizontal.
- a square block (104) of 4 x 4 samples (indicated by a dashed, boldface line).
- the square block (104) includes 16 samples, each labelled with an “S”, its position in the Y dimension (e.g., row index) and its position in the X dimension (e.g., column index).
- sample S21 is the second sample in the Y dimension (from the top) and the first (from the left) sample in the X dimension.
- sample S44 is the fourth sample in block (104) in both the Y and X dimensions.
- S44 is at the bottom right. Further shown are reference samples that follow a similar numbering scheme.
- a reference sample is labelled with an R, its Y position (e.g., row index) and X position (column index) relative to block (104).
- R its Y position (e.g., row index) and X position (column index) relative to block (104).
- Y position e.g., row index
- X position column index
- Intra picture prediction can work by copying reference sample values from the neighboring samples as appropriated by the signaled prediction direction.
- the coded video bitstream includes signaling that, for this block, indicates a prediction direction consistent with arrow (102) — that is, samples are predicted from a prediction sample or samples to the upper right, at a 45 degree angle from the horizontal.
- samples S41, S32, S23, and S14 are predicted from the same reference sample R05.
- Sample S44 is then predicted from reference sample R08.
- the values of multiple reference samples may be combined, for example through interpolation, in order to calculate a reference sample; especially when the directions are not evenly divisible by 45 degrees.
- FIG. IB shows a schematic (110) that depicts 65 intra prediction directions according to JEM to illustrate the increasing number of prediction directions over time.
- the mapping of intra prediction directions bits in the coded video bitstream that represent the direction can be different from video coding technology to video coding technology; and can range, for example, from simple direct mappings of prediction direction to intra prediction mode, to codewords, to complex adaptive schemes involving most probable modes, and similar techniques. In all cases, however, there can be certain directions that are statistically less likely to occur in video content than certain other directions. As the goal of video compression is the reduction of redundancy, those less likely directions will, in a well working video coding technology, be represented by a larger number of bits than more likely directions.
- Motion compensation can be a lossy compression technique and can relate to techniques where a block of sample data from a previously reconstructed picture or part thereof (reference picture), after being spatially shifted in a direction indicated by a motion vector (MV henceforth), is used for the prediction of a newly reconstructed picture or picture part.
- the reference picture can be the same as the picture currently under reconstruction.
- MVs can have two dimensions X and Y, or three dimensions, the third being an indication of the reference picture in use (the latter, indirectly, can be a time dimension).
- an MV applicable to a certain area of sample data can be predicted from other MVs, for example from those related to another area of sample data spatially adjacent to the area under reconstruction, and preceding that MV in decoding order. Doing so can substantially reduce the amount of data required for coding the MV, thereby removing redundancy and increasing compression.
- MV prediction can work effectively, for example, because when coding an input video signal derived from a camera (known as natural video) there is a statistical likelihood that areas larger than the area to which a single MV is applicable move in a similar direction and, therefore, can in some cases be predicted using a similar motion vector derived from MVs of neighboring area.
- MV prediction can be an example of lossless compression of a signal (namely: the MVs) derived from the original signal (namely: the sample stream).
- MV prediction itself can be lossy, for example because of rounding errors when calculating a predictor from several surrounding MVs.
- a current block (201) comprises samples that have been found by the encoder during the motion search process to be predictable from a previous block of the same size that has been spatially shifted.
- the MV can be derived from metadata associated with one or more reference pictures, for example from the most recent (in decoding order) reference picture, using the MV associated with either one of five surrounding samples, denoted A0, Al, and B0, Bl, B2 (202 through 206, respectively).
- the MV prediction can use predictors from the same reference picture that the neighboring block is using.
- an apparatus for video decoding includes processing circuitry.
- the processing circuitry is configured to decode first neural network update information in a coded bitstream for a first neural network in a video decoder.
- the first neural network is configured with first pretrained parameters.
- the first neural network update information corresponds to a first block in an image to be reconstructed and indicates a first replacement parameter corresponding to a first pretrained parameter in the first pretrained parameters.
- the processing circuitry can update the first neural network in the video decoder based on the first replacement parameter and decode the first block based on the updated first neural network for the first block.
- the first neural network update information further indicates one or more replacement parameters for one or more remaining neural networks in the video decoder.
- the processing circuitry can update the one or more remaining neural networks based on the one or more replacement parameters.
- the processing circuitry decodes second neural network update information in the coded bitstream for a second neural network in the video decoder.
- the second neural network is configured with second pretrained parameters.
- the second neural network update information corresponds to a second block in the image to be reconstructed and indicates a second replacement parameter corresponding to a second pretrained parameter in the second pretrained parameters.
- the second neural network is different from the first neural network.
- the processing circuitry can update the second neural network in the video decoder based on the second replacement parameter and decode the second block based on the updated second neural network for the second block.
- the first pretrained parameter is a one of a pretrained weight coefficient and a pretrained bias term.
- the second pretrained parameter is another one of the pretrained weight coefficient and the pretrained bias term.
- the processing circuitry decodes a second block in the coded bitstream based on the updated first neural network for the first block.
- the first neural network update information indicates a difference between the first replacement parameter and the first pretrained parameter.
- the processing circuitry determines the first replacement parameter according to a sum of the difference and the first pretrained parameter.
- the processing circuitry decodes the first neural network update information based on one of a variation of a Lempel-Ziv-Markov chain algorithm (LZMA2) and a bzip2 algorithm.
- LZMA2 Lempel-Ziv-Markov chain algorithm
- the processing circuitry decodes the second neural network update information based on another one of the LZMA2 and the bzip2 algorithm.
- aspects of the disclosure also provide a non-transitory computer-readable storage medium storing a program executable by at least one processor to perform the methods for video decoding.
- FIG. 1A is a schematic illustration of an exemplary subset of intra prediction modes.
- FIG. IB is an illustration of exemplary intra prediction directions.
- FIG. 2 shows a current block (201) and surrounding samples in accordance with an embodiment.
- FIG. 3 is a schematic illustration of a simplified block diagram of a communication system (300) in accordance with an embodiment.
- FIG. 4 is a schematic illustration of a simplified block diagram of a communication system (400) in accordance with an embodiment.
- FIG. 5 is a schematic illustration of a simplified block diagram of a decoder in accordance with an embodiment.
- FIG. 6 is a schematic illustration of a simplified block diagram of an encoder in accordance with an embodiment.
- FIG. 7 shows a block diagram of an encoder in accordance with another embodiment.
- FIG. 8 shows a block diagram of a decoder in accordance with another embodiment.
- FIG. 9A shows an example of a block-wise image coding according to an embodiment of the disclosure.
- FIG. 9B shows an exemplary NIC framework according to an embodiment of the disclosure.
- FIG. 10 shows an exemplary convolution neural network (CNN) of a main encoder network according to an embodiment of the disclosure.
- CNN convolution neural network
- FIG. 11 shows an exemplary CNN of a main decoder network according to an embodiment of the disclosure.
- FIG. 12 shows an exemplary CNN of a hyper encoder according to an embodiment of the disclosure.
- FIG. 13 shows an exemplary CNN of a hyper decoder according to an embodiment of the disclosure.
- FIG. 14 shows an exemplary CNN of a context model network according to an embodiment of the disclosure.
- FIG. 15 shows an exemplary CNN of an entropy parameter network according to an embodiment of the disclosure.
- FIG. 16A shows an exemplary video encoder according to an embodiment of the disclosure.
- FIG. 16B shows an exemplary video decoder according to an embodiment of the disclosure.
- FIG. 17 shows an exemplary video encoder according to an embodiment of the disclosure.
- FIG. 18 shows an exemplary video decoder according to an embodiment of the disclosure.
- FIG. 19 shows a flow chart outlining a process according to an embodiment of the disclosure.
- FIG. 20 shows a flow chart outlining a process according to an embodiment of the disclosure.
- FIG. 21 is a schematic illustration of a computer system in accordance with an embodiment.
- FIG. 3 illustrates a simplified block diagram of a communication system (300) according to an embodiment of the present disclosure.
- the communication system (300) includes a plurality of terminal devices that can communicate with each other, via, for example, a network (350).
- the communication system (300) includes a first pair of terminal devices (310) and (320) interconnected via the network (350).
- the first pair of terminal devices (310) and (320) performs unidirectional transmission of data.
- the terminal device (310) may code video data (e.g., a stream of video pictures that are captured by the terminal device (310)) for transmission to the other terminal device (320) via the network (350).
- the encoded video data can be transmitted in the form of one or more coded video bitstreams.
- the terminal device (320) may receive the coded video data from the network (350), decode the coded video data to recover the video pictures and display video pictures according to the recovered video data.
- Unidirectional data transmission may be common in media serving applications and the like.
- the communication system (300) includes a second pair of terminal devices (330) and (340) that performs bidirectional transmission of coded video data that may occur, for example, during videoconferencing.
- each terminal device of the terminal devices (330) and (340) may code video data (e.g., a stream of video pictures that are captured by the terminal device) for transmission to the other terminal device of the terminal devices (330) and (340) via the network (350).
- Each terminal device of the terminal devices (330) and (340) also may receive the coded video data transmitted by the other terminal device of the terminal devices (330) and (340), and may decode the coded video data to recover the video pictures and may display video pictures at an accessible display device according to the recovered video data.
- the terminal devices (310), (320), (330) and (340) may be illustrated as servers, personal computers and smart phones but the principles of the present disclosure may be not so limited. Embodiments of the present disclosure find application with laptop computers, tablet computers, media players and/or dedicated video conferencing equipment.
- the network (350) represents any number of networks that convey coded video data among the terminal devices (310), (320), (330) and (340), including for example wireline (wired) and/or wireless communication networks.
- the communication network (350) may exchange data in circuit-switched and/or packet-switched channels.
- Representative networks include telecommunications networks, local area networks, wide area networks and/or the Internet. For the purposes of the present discussion, the architecture and topology of the network (350) may be immaterial to the operation of the present disclosure unless explained herein below.
- FIG. 4 illustrates, as an example for an application for the disclosed subject matter, the placement of a video encoder and a video decoder in a streaming environment.
- the disclosed subject matter can be equally applicable to other video enabled applications, including, for example, video conferencing, digital TV, storing of compressed video on digital media including CD, DVD, memory stick and the like, and so on.
- a streaming system may include a capture subsystem (413), that can include a video source (401), for example a digital camera, creating for example a stream of video pictures (402) that are uncompressed.
- the stream of video pictures (402) includes samples that are taken by the digital camera.
- the stream of video pictures (402), depicted as a bold line to emphasize a high data volume when compared to encoded video data (404) (or coded video bitstreams), can be processed by an electronic device (420) that includes a video encoder (403) coupled to the video source (401).
- the video encoder (403) can include hardware, software, or a combination thereof to enable or implement aspects of the disclosed subject matter as described in more detail below.
- the encoded video data (404) (or encoded video bitstream (404)), depicted as a thin line to emphasize the lower data volume when compared to the stream of video pictures (402), can be stored on a streaming server (405) for future use.
- One or more streaming client subsystems such as client subsystems (406) and (408) in FIG. 4 can access the streaming server (405) to retrieve copies (407) and (409) of the encoded video data (404).
- a client subsystem (406) can include a video decoder (410), for example, in an electronic device (430).
- the video decoder (410) decodes the incoming copy (407) of the encoded video data and creates an outgoing stream of video pictures (411) that can be rendered on a display (412) (e.g., display screen) or other rendering device (not depicted).
- the encoded video data (404), (407), and (409) e.g., video bitstreams
- video coding/compression standards examples include ITU-T Recommendation H.265.
- a video coding standard under development is informally known as Versatile Video Coding (VVC). The disclosed subject matter may be used in the context of VVC.
- the electronic devices (420) and (430) can include other components (not shown).
- the electronic device (420) can include a video decoder (not shown) and the electronic device (430) can include a video encoder (not shown) as well.
- FIG. 5 shows a block diagram of a video decoder (510) according to an embodiment of the present disclosure.
- the video decoder (510) can be included in an electronic device (530).
- the electronic device (530) can include a receiver (531) (e.g., receiving circuitry).
- the video decoder (510) can be used in the place of the video decoder (410) in the FIG. 4 example.
- the receiver (531) may receive one or more coded video sequences to be decoded by the video decoder (510); in the same or another embodiment, one coded video sequence at a time, where the decoding of each coded video sequence is independent from other coded video sequences.
- the coded video sequence may be received from a channel (501), which may be a hardware/software link to a storage device which stores the encoded video data.
- the receiver (531) may receive the encoded video data with other data, for example, coded audio data and/or ancillary data streams, that may be forwarded to their respective using entities (not depicted).
- the receiver (531) may separate the coded video sequence from the other data.
- a buffer memory (515) may be coupled in between the receiver (531) and an entropy decoder / parser (520) ("parser (520)" henceforth).
- the buffer memory (515) is part of the video decoder (510). In others, it can be outside of the video decoder (510) (not depicted). In still others, there can be a buffer memory (not depicted) outside of the video decoder (510), for example to combat network jitter, and in addition another buffer memory (515) inside the video decoder (510), for example to handle playout timing.
- the buffer memory (515) may not be needed, or can be small.
- the buffer memory (515) may be required, can be comparatively large and can be advantageously of adaptive size, and may at least partially be implemented in an operating system or similar elements (not depicted) outside of the video decoder (510).
- the video decoder (510) may include the parser (520) to reconstruct symbols (521) from the coded video sequence. Categories of those symbols include information used to manage operation of the video decoder (510), and potentially information to control a rendering device such as a render device (512) (e.g., a display screen) that is not an integral part of the electronic device (530) but can be coupled to the electronic device (530), as was shown in FIG. 5.
- the control information for the rendering device(s) may be in the form of Supplemental Enhancement Information (SEI messages) or Video Usability Information (VUI) parameter set fragments (not depicted).
- SEI messages Supplemental Enhancement Information
- VUI Video Usability Information
- the parser (520) may parse / entropy-decode the coded video sequence that is received.
- the coding of the coded video sequence can be in accordance with a video coding technology or standard, and can follow various principles, including variable length coding, Huffman coding, arithmetic coding with or without context sensitivity, and so forth.
- the parser (520) may extract from the coded video sequence, a set of subgroup parameters for at least one of the subgroups of pixels in the video decoder, based upon at least one parameter corresponding to the group. Subgroups can include Groups of Pictures (GOPs), pictures, tiles, slices, macroblocks, Coding Units (CUs), blocks, Transform Units (TUs), Prediction Units (PUs) and so forth.
- the parser (520) may also extract from the coded video sequence information such as transform coefficients, quantizer parameter values, motion vectors, and so forth.
- the parser (520) may perform an entropy decoding / parsing operation on the video sequence received from the buffer memory (515), so as to create symbols (521).
- Reconstruction of the symbols (521) can involve multiple different units depending on the type of the coded video picture or parts thereof (such as: inter and intra picture, inter and intra block), and other factors. Which units are involved, and how, can be controlled by the subgroup control information that was parsed from the coded video sequence by the parser (520). The flow of such subgroup control information between the parser (520) and the multiple units below is not depicted for clarity.
- the video decoder (510) can be conceptually subdivided into a number of functional units as described below. In a practical implementation operating under commercial constraints, many of these units interact closely with each other and can, at least partly, be integrated into each other. However, for the purpose of describing the disclosed subject matter, the conceptual subdivision into the functional units below is appropriate.
- a first unit is the scaler / inverse transform unit (551).
- the scaler / inverse transform unit (551) receives a quantized transform coefficient as well as control information, including which transform to use, block size, quantization factor, quantization scaling matrices, etc. as symbol(s) (521) from the parser (520).
- the scaler / inverse transform unit (551) can output blocks comprising sample values, that can be input into aggregator (555).
- the output samples of the scaler / inverse transform (551) can pertain to an intra coded block; that is: a block that is not using predictive information from previously reconstructed pictures, but can use predictive information from previously reconstructed parts of the current picture.
- Such predictive information can be provided by an intra picture prediction unit (552).
- the intra picture prediction unit (552) generates a block of the same size and shape of the block under reconstruction, using surrounding already reconstructed information fetched from the current picture buffer (558).
- the current picture buffer (558) buffers, for example, partly reconstructed current picture and/or fully reconstructed current picture.
- the aggregator (555) adds, on a per sample basis, the prediction information the intra prediction unit (552) has generated to the output sample information as provided by the scaler / inverse transform unit (551).
- the output samples of the scaler / inverse transform unit (551) can pertain to an inter coded, and potentially motion compensated block.
- a motion compensation prediction unit (553) can access reference picture memory (557) to fetch samples used for prediction. After motion compensating the fetched samples in accordance with the symbols (521) pertaining to the block, these samples can be added by the aggregator (555) to the output of the scaler / inverse transform unit (551) (in this case called the residual samples or residual signal) so as to generate output sample information.
- the addresses within the reference picture memory (557) from where the motion compensation prediction unit (553) fetches prediction samples can be controlled by motion vectors, available to the motion compensation prediction unit (553) in the form of symbols (521) that can have, for example X, Y, and reference picture components.
- Motion compensation also can include interpolation of sample values as fetched from the reference picture memory (557) when sub-sample exact motion vectors are in use, motion vector prediction mechanisms, and so forth.
- the output samples of the aggregator (555) can be subject to various loop filtering techniques in the loop filter unit (556).
- Video compression technologies can include in-loop filter technologies that are controlled by parameters included in the coded video sequence (also referred to as coded video bitstream) and made available to the loop filter unit (556) as symbols (521) from the parser (520), but can also be responsive to meta-information obtained during the decoding of previous (in decoding order) parts of the coded picture or coded video sequence, as well as responsive to previously reconstructed and loop-filtered sample values.
- the output of the loop filter unit (556) can be a sample stream that can be output to the render device (512) as well as stored in the reference picture memory (557) for use in future inter-picture prediction.
- Certain coded pictures once fully reconstructed, can be used as reference pictures for future prediction. For example, once a coded picture corresponding to a current picture is fully reconstructed and the coded picture has been identified as a reference picture (by, for example, the parser (520)), the current picture buffer (558) can become a part of the reference picture memory (557), and a fresh current picture buffer can be reallocated before commencing the reconstruction of the following coded picture.
- the video decoder (510) may perform decoding operations according to a predetermined video compression technology in a standard, such as ITU-T Rec. H.265.
- the coded video sequence may conform to a syntax specified by the video compression technology or standard being used, in the sense that the coded video sequence adheres to both the syntax of the video compression technology or standard and the profiles as documented in the video compression technology or standard.
- a profile can select certain tools as the only tools available for use under that profile from all the tools available in the video compression technology or standard.
- Also necessary for compliance can be that the complexity of the coded video sequence is within bounds as defined by the level of the video compression technology or standard.
- levels restrict the maximum picture size, maximum frame rate, maximum reconstruction sample rate (measured in, for example megasamples per second), maximum reference picture size, and so on. Limits set by levels can, in some cases, be further restricted through Hypothetical Reference Decoder (HRD) specifications and metadata for HRD buffer management signaled in the coded video sequence.
- HRD Hypothetical Reference Decoder
- the receiver (531) may receive additional (redundant) data with the encoded video.
- the additional data may be included as part of the coded video sequence(s).
- the additional data may be used by the video decoder (510) to properly decode the data and/or to more accurately reconstruct the original video data.
- Additional data can be in the form of, for example, temporal, spatial, or signal noise ratio (SNR) enhancement layers, redundant slices, redundant pictures, forward error correction codes, and so on.
- SNR signal noise ratio
- FIG. 6 shows a block diagram of a video encoder (603) according to an embodiment of the present disclosure.
- the video encoder (603) is included in an electronic device (620).
- the electronic device (620) includes a transmitter (640) (e.g., transmitting circuitry).
- the video encoder (603) can be used in the place of the video encoder (403) in the FIG. 4 example.
- the video encoder (603) may receive video samples from a video source (601) (that is not part of the electronic device (620) in the FIG. 6 example) that may capture video image(s) to be coded by the video encoder (603).
- a video source (601) that is not part of the electronic device (620) in the FIG. 6 example
- the video source (601) is a part of the electronic device (620).
- the video source (601) may provide the source video sequence to be coded by the video encoder (603) in the form of a digital video sample stream that can be of any suitable bit depth (for example: 8 bit, 10 bit, 12 bit, ...), any colorspace (for example, BT.601 Y CrCB, RGB, ...), and any suitable sampling structure (for example Y CrCb 4:2:0, Y CrCb 4:4:4).
- the video source (601) may be a storage device storing previously prepared video.
- the video source (601) may be a camera that captures local image information as a video sequence.
- Video data may be provided as a plurality of individual pictures that impart motion when viewed in sequence. The pictures themselves may be organized as a spatial array of pixels, wherein each pixel can comprise one or more samples depending on the sampling structure, color space, etc. in use.
- the video encoder (603) may code and compress the pictures of the source video sequence into a coded video sequence (643) in real time or under any other time constraints as required by the application. Enforcing appropriate coding speed is one function of a controller (650).
- the controller (650) controls other functional units as described below and is functionally coupled to the other functional units. The coupling is not depicted for clarity.
- Parameters set by the controller (650) can include rate control related parameters (picture skip, quantizer, lambda value of rate-distortion optimization techniques, ...), picture size, group of pictures (GOP) layout, maximum motion vector search range, and so forth.
- the controller (650) can be configured to have other suitable functions that pertain to the video encoder (603) optimized for a certain system design.
- the video encoder (603) is configured to operate in a coding loop.
- the coding loop can include a source coder (630) (e.g., responsible for creating symbols, such as a symbol stream, based on an input picture to be coded, and a reference picture(s)), and a (local) decoder (633) embedded in the video encoder (603).
- the decoder (633) reconstructs the symbols to create the sample data in a similar manner as a (remote) decoder also would create (as any compression between symbols and coded video bitstream is lossless in the video compression technologies considered in the disclosed subject matter).
- the reconstructed sample stream (sample data) is input to the reference picture memory (634).
- the content in the reference picture memory (634) is also bit exact between the local encoder and remote encoder.
- the prediction part of an encoder "sees” as reference picture samples exactly the same sample values as a decoder would "see” when using prediction during decoding.
- This fundamental principle of reference picture synchronicity (and resulting drift, if synchronicity cannot be maintained, for example because of channel errors) is used in some related arts as well.
- the operation of the "local" decoder (633) can be the same as of a "remote” decoder, such as the video decoder (510), which has already been described in detail above in conjunction with FIG. 5.
- a "remote” decoder such as the video decoder (510)
- the entropy decoding parts of the video decoder (510) including the buffer memory (515), and parser (520) may not be fully implemented in the local decoder (633).
- a decoder technology except the parsing/entropy decoding that is present in a decoder is present, in an identical or a substantially identical functional form, in a corresponding encoder. Accordingly, the disclosed subject matter focuses on decoder operation.
- encoder technologies can be abbreviated as they are the inverse of the comprehensively described decoder technologies. In certain areas a more detail description is provided below.
- the source coder (630) may perform motion compensated predictive coding, which codes an input picture predictively with reference to one or more previously coded picture from the video sequence that were designated as "reference pictures.”
- the coding engine (632) codes differences between pixel blocks of an input picture and pixel blocks of reference picture(s) that may be selected as prediction reference(s) to the input picture.
- the local video decoder (633) may decode coded video data of pictures that may be designated as reference pictures, based on symbols created by the source coder (630). Operations of the coding engine (632) may advantageously be lossy processes.
- the coded video data may be decoded at a video decoder (not shown in FIG. 6)
- the reconstructed video sequence typically may be a replica of the source video sequence with some errors.
- the local video decoder (633) replicates decoding processes that may be performed by the video decoder on reference pictures and may cause reconstructed reference pictures to be stored in the reference picture cache (634). In this manner, the video encoder (603) may store copies of reconstructed reference pictures locally that have common content as the reconstructed reference pictures that will be obtained by a far-end video decoder (absent transmission errors).
- the predictor (635) may perform prediction searches for the coding engine (632). That is, for a new picture to be coded, the predictor (635) may search the reference picture memory (634) for sample data (as candidate reference pixel blocks) or certain metadata such as reference picture motion vectors, block shapes, and so on, that may serve as an appropriate prediction reference for the new pictures.
- the predictor (635) may operate on a sample block-by-pixel block basis to find appropriate prediction references. In some cases, as determined by search results obtained by the predictor (635), an input picture may have prediction references drawn from multiple reference pictures stored in the reference picture memory (634).
- the controller (650) may manage coding operations of the source coder (630), including, for example, setting of parameters and subgroup parameters used for encoding the video data.
- Output of all aforementioned functional units may be subjected to entropy coding in the entropy coder (645).
- the entropy coder (645) translates the symbols as generated by the various functional units into a coded video sequence, by lossless compressing the symbols according to technologies such as Huffman coding, variable length coding, arithmetic coding, and so forth.
- the transmitter (640) may buffer the coded video sequence(s) as created by the entropy coder (645) to prepare for transmission via a communication channel (660), which may be a hardware/software link to a storage device which would store the encoded video data.
- the transmitter (640) may merge coded video data from the video coder (603) with other data to be transmitted, for example, coded audio data and/or ancillary data streams (sources not shown).
- the controller (650) may manage operation of the video encoder (603).
- the controller (650) may assign to each coded picture a certain coded picture type, which may affect the coding techniques that may be applied to the respective picture. For example, pictures often may be assigned as one of the following picture types:
- An Intra Picture may be one that may be coded and decoded without using any other picture in the sequence as a source of prediction.
- Some video codecs allow for different types of intra pictures, including, for example Independent Decoder Refresh (“IDR”) Pictures.
- IDR Independent Decoder Refresh
- a predictive picture may be one that may be coded and decoded using intra prediction or inter prediction using at most one motion vector and reference index to predict the sample values of each block.
- a bi-directionally predictive picture may be one that may be coded and decoded using intra prediction or inter prediction using at most two motion vectors and reference indices to predict the sample values of each block.
- multiple-predictive pictures can use more than two reference pictures and associated metadata for the reconstruction of a single block.
- Source pictures commonly may be subdivided spatially into a plurality of sample blocks (for example, blocks of 4x4, 8x8, 4x8, or 16x16 samples each) and coded on a block-by-block basis.
- Blocks may be coded predictively with reference to other (already coded) blocks as determined by the coding assignment applied to the blocks' respective pictures.
- blocks of I pictures may be coded non-predictively or they may be coded predictively with reference to already coded blocks of the same picture (spatial prediction or intra prediction).
- Pixel blocks of P pictures may be coded predictively, via spatial prediction or via temporal prediction with reference to one previously coded reference picture.
- Blocks of B pictures may be coded predictively, via spatial prediction or via temporal prediction with reference to one or two previously coded reference pictures.
- the video encoder (603) may perform coding operations according to a predetermined video coding technology or standard, such as ITU-T Rec. H.265. In its operation, the video encoder (603) may perform various compression operations, including predictive coding operations that exploit temporal and spatial redundancies in the input video sequence.
- the coded video data therefore, may conform to a syntax specified by the video coding technology or standard being used.
- the transmitter (640) may transmit additional data with the encoded video.
- the source coder (630) may include such data as part of the coded video sequence. Additional data may comprise temporal/spatial/SNR enhancement layers, other forms of redundant data such as redundant pictures and slices, SEI messages, VUI parameter set fragments, and so on.
- a video may be captured as a plurality of source pictures (video pictures) in a temporal sequence.
- Intra-picture prediction (often abbreviated to intra prediction) makes use of spatial correlation in a given picture
- inter-picture prediction makes uses of the (temporal or other) correlation between the pictures.
- a specific picture under encoding/decoding which is referred to as a current picture
- the block in the current picture can be coded by a vector that is referred to as a motion vector.
- the motion vector points to the reference block in the reference picture, and can have a third dimension identifying the reference picture, in case multiple reference pictures are in use.
- a bi-prediction technique can be used in the inter picture prediction.
- two reference pictures such as a first reference picture and a second reference picture that are both prior in decoding order to the current picture in the video (but may be in the past and future, respectively, in display order) are used.
- a block in the current picture can be coded by a first motion vector that points to a first reference block in the first reference picture, and a second motion vector that points to a second reference block in the second reference picture.
- the block can be predicted by a combination of the first reference block and the second reference block.
- a merge mode technique can be used in the inter-picture prediction to improve coding efficiency.
- predictions such as inter picture predictions and intra-picture predictions are performed in the unit of blocks.
- CTU coding tree units
- a picture in a sequence of video pictures is partitioned into coding tree units (CTU) for compression, the CTUs in a picture have the same size, such as 64x64 pixels, 32x32 pixels, or 16x16 pixels.
- a CTU includes three coding tree blocks (CTBs), which are one luma CTB and two chroma CTBs.
- Each CTU can be recursively quadtree split into one or multiple coding units (CUs).
- a CTU of 64x64 pixels can be split into one CU of 64x64 pixels, or 4 CUs of 32x32 pixels, or 16 CUs of 16x16 pixels.
- each CU is analyzed to determine a prediction type for the CU, such as an inter prediction type or an intra prediction type.
- the CU is split into one or more prediction units (PUs) depending on the temporal and/or spatial predictability.
- each PU includes a luma prediction block (PB), and two chroma PBs.
- PB luma prediction block
- a prediction operation in coding is performed in the unit of a prediction block.
- the prediction block includes a matrix of values (e.g., luma values) for pixels, such as 8x8 pixels, 16x16 pixels, 8x16 pixels, 16x8 pixels, and the like.
- FIG. 7 shows a diagram of a video encoder (703) according to another embodiment of the disclosure.
- the video encoder (703) is configured to receive a processing block (e.g., a prediction block) of sample values within a current video picture in a sequence of video pictures, and encode the processing block into a coded picture that is part of a coded video sequence.
- a processing block e.g., a prediction block
- the video encoder (703) is used in the place of the video encoder (403) in the FIG. 4 example.
- the video encoder (703) receives a matrix of sample values for a processing block, such as a prediction block of 8x8 samples, and the like.
- the video encoder (703) determines whether the processing block is best coded using intra mode, inter mode, or bi-prediction mode using, for example, rate-distortion optimization.
- the video encoder (703) may use an intra prediction technique to encode the processing block into the coded picture; and when the processing block is to be coded in inter mode or bi-prediction mode, the video encoder (703) may use an inter prediction or bi-prediction technique, respectively, to encode the processing block into the coded picture.
- merge mode can be an inter picture prediction submode where the motion vector is derived from one or more motion vector predictors without the benefit of a coded motion vector component outside the predictors.
- a motion vector component applicable to the subject block may be present.
- the video encoder (703) includes other components, such as a mode decision module (not shown) to determine the mode of the processing blocks.
- the video encoder (703) includes the inter encoder (730), an intra encoder (722), a residue calculator (723), a switch (726), a residue encoder (724), a general controller (721), and an entropy encoder (725) coupled together as shown in FIG. 7.
- the inter encoder (730) is configured to receive the samples of the current block (e.g., a processing block), compare the block to one or more reference blocks in reference pictures (e.g., blocks in previous pictures and later pictures), generate inter prediction information (e.g., description of redundant information according to inter encoding technique, motion vectors, merge mode information), and calculate inter prediction results (e.g., predicted block) based on the inter prediction information using any suitable technique.
- the reference pictures are decoded reference pictures that are decoded based on the encoded video information.
- the intra encoder (722) is configured to receive the samples of the current block (e.g., a processing block), in some cases compare the block to blocks already coded in the same picture, generate quantized coefficients after transform, and in some cases also intra prediction information (e.g., an intra prediction direction information according to one or more intra encoding techniques). In an example, the intra encoder (722) also calculates intra prediction results (e.g., predicted block) based on the intra prediction information and reference blocks in the same picture.
- intra prediction information e.g., an intra prediction direction information according to one or more intra encoding techniques
- the general controller (721) is configured to determine general control data and control other components of the video encoder (703) based on the general control data.
- the general controller (721) determines the mode of the block, and provides a control signal to the switch (726) based on the mode. For example, when the mode is the intra mode, the general controller (721) controls the switch (726) to select the intra mode result for use by the residue calculator (723), and controls the entropy encoder (725) to select the intra prediction information and include the intra prediction information in the bitstream; and when the mode is the inter mode, the general controller (721) controls the switch (726) to select the inter prediction result for use by the residue calculator (723), and controls the entropy encoder (725) to select the inter prediction information and include the inter prediction information in the bitstream.
- the residue calculator (723) is configured to calculate a difference (residue data) between the received block and prediction results selected from the intra encoder (722) or the inter encoder (730).
- the residue encoder (724) is configured to operate based on the residue data to encode the residue data to generate the transform coefficients.
- the residue encoder (724) is configured to convert the residue data from a spatial domain to a frequency domain, and generate the transform coefficients. The transform coefficients are then subject to quantization processing to obtain quantized transform coefficients.
- the video encoder (703) also includes a residue decoder (728).
- the residue decoder (728) is configured to perform inverse-transform, and generate the decoded residue data.
- the decoded residue data can be suitably used by the intra encoder (722) and the inter encoder (730).
- the inter encoder (730) can generate decoded blocks based on the decoded residue data and inter prediction information
- the intra encoder (722) can generate decoded blocks based on the decoded residue data and the intra prediction information.
- the decoded blocks are suitably processed to generate decoded pictures and the decoded pictures can be buffered in a memory circuit (not shown) and used as reference pictures in some examples.
- the entropy encoder (725) is configured to format the bitstream to include the encoded block.
- the entropy encoder (725) is configured to include various information according to a suitable standard, such as the HEVC standard.
- the entropy encoder (725) is configured to include the general control data, the selected prediction information (e.g., intra prediction information or inter prediction information), the residue information, and other suitable information in the bitstream. Note that, according to the disclosed subject matter, when coding a block in the merge submode of either inter mode or bi-prediction mode, there is no residue information.
- FIG. 8 shows a diagram of a video decoder (810) according to another embodiment of the disclosure.
- the video decoder (810) is configured to receive coded pictures that are part of a coded video sequence, and decode the coded pictures to generate reconstructed pictures.
- the video decoder (810) is used in the place of the video decoder (410) in the FIG. 4 example.
- the video decoder (810) includes an entropy decoder (871), an inter decoder (880), a residue decoder (873), a reconstruction module (874), and an intra decoder (872) coupled together as shown in FIG. 8.
- the entropy decoder (871) can be configured to reconstruct, from the coded picture, certain symbols that represent the syntax elements of which the coded picture is made up. Such symbols can include, for example, the mode in which a block is coded (such as, for example, intra mode, inter mode, bi-predicted mode, the latter two in merge submode or another submode), prediction information (such as, for example, intra prediction information or inter prediction information) that can identify certain sample or metadata that is used for prediction by the intra decoder (872) or the inter decoder (880), respectively, residual information in the form of, for example, quantized transform coefficients, and the like.
- the mode in which a block is coded such as, for example, intra mode, inter mode, bi-predicted mode, the latter two in merge submode or another submode
- prediction information such as, for example, intra prediction information or inter prediction information
- residual information in the form of, for example, quantized transform coefficients, and the like.
- the inter prediction information is provided to the inter decoder (880); and when the prediction type is the intra prediction type, the intra prediction information is provided to the intra decoder (872).
- the residual information can be subject to inverse quantization and is provided to the residue decoder (873).
- the inter decoder (880) is configured to receive the inter prediction information, and generate inter prediction results based on the inter prediction information.
- the intra decoder (872) is configured to receive the intra prediction information, and generate prediction results based on the intra prediction information.
- the residue decoder (873) is configured to perform inverse quantization to extract de-quantized transform coefficients, and process the de-quantized transform coefficients to convert the residual from the frequency domain to the spatial domain.
- the residue decoder (873) may also require certain control information (to include the Quantizer Parameter (QP)), and that information may be provided by the entropy decoder (871) (data path not depicted as this may be low volume control information only).
- QP Quantizer Parameter
- the reconstruction module (874) is configured to combine, in the spatial domain, the residual as output by the residue decoder (873) and the prediction results (as output by the inter or intra prediction modules as the case may be) to form a reconstructed block, that may be part of the reconstructed picture, which in turn may be part of the reconstructed video. It is noted that other suitable operations, such as a deblocking operation and the like, can be performed to improve the visual quality.
- the video encoders (403), (603), and (703), and the video decoders (410), (510), and (810) can be implemented using any suitable technique.
- the video encoders (403), (603), and (703), and the video decoders (410), (510), and (810) can be implemented using one or more integrated circuits.
- the video encoders (403), (603), and (603), and the video decoders (410), (510), and (810) can be implemented using one or more processors that execute software instructions.
- This disclosure describes video coding technologies related to neural image compression technologies and/or neural video compression technologies, such as artificial intelligence (AI) based neural image compression (NIC).
- AI artificial intelligence
- NIC neural image compression
- a neural network can include an artificial neural network (ANN), such as a deep neural network (DNN), a convolution neural network (CNN), or the like.
- ANN artificial neural network
- DNN deep neural network
- CNN convolution neural network
- a related hybrid video codec is difficult to be optimized as a whole.
- an improvement of a single module (e.g., an encoder) in the hybrid video codec may not result in a coding gain in the overall performance.
- different modules can be jointly optimized from an input to an output to improve a final objective (e.g., rate-distortion performance, such as a rate-distortion loss L described in the disclosure) by performing a learning process or a training process (e.g., a machine learning process), and thus resulting in an E2E optimized NIC.
- the NIC framework can use an input block x as an input to a neural network encoder (e.g., an encoder based on neural networks such as DNNs) to compute a compressed representation (e.g., a compact representation) x that can be compact, for example, for storage and transmission purposes.
- a neural network decoder e.g., a decoder based on neural networks such as DNNs
- the input block x and reconstructed block x are in a spatial domain and the compressed representation x is in a domain different from the spatial domain.
- the compressed representation x is quantized and entropy coded.
- a NIC framework can use a variational autoencoder (VAE) structure.
- VAE variational autoencoder
- the neural network encoder can directly use the entire input block x as the input to the neural network encoder.
- the entire input block x can pass through a set of neural network layers that work as a black box to compute the compressed representation x.
- the compressed representation x is an output of the neural network encoder.
- the neural network decoder can take the entire compressed representation x as an input.
- the compressed representation x can pass through another set of neural network layers that work as another black box to compute the reconstructed block x.
- a rate-distortion (R-D) loss L (x, x, x) can be optimized to achieve a trade-off between a distortion loss D(x, x) of the reconstructed block x and bit consumption R of the compact representation x with a trade-off hyperparameter l.
- a neural network e.g., an ANN
- An ANN can leam to perform tasks from examples, without task-specific programming.
- An ANN can be configured with connected nodes or artificial neurons.
- a connection between nodes can transmit a signal from a first node to a second node (e.g., a receiving node), and the signal can be modified by a weight which can be indicated by a weight coefficient for the connection.
- the receiving node can process signal(s) (i.e., input signal(s) for the receiving node) from node(s) that transmit the signal(s) to the receiving node and then generate an output signal by applying a function to the input signals.
- the function can be a linear function.
- the output signal is a weighted summation of the input signal(s).
- the output signal is further modified by a bias which can be indicated by a bias term, and thus the output signal is a sum of the bias and the weighted summation of the input signal(s).
- the function can include a nonlinear operation, for example, on the weighted sum or the sum of the bias and the weighted summation of the input signal(s).
- the output signal can be sent to node(s) (downstream node(s)) connected to the receiving node).
- the ANN can be represented or configured by parameters (e.g., weights of the connections and/or biases).
- the weights and/or the biases can be obtained by training the ANN with examples where the weights and/or the biases can be iteratively adjusted.
- the trained ANN configured with the determined weights and/or the determined biases can be used to perform tasks.
- Nodes in an ANN can be organized in any suitable architecture.
- nodes in an ANN are organized in layers including an input layer that receives input signal(s) to the ANN and an output layer that outputs output signal(s) from the ANN.
- the ANN further includes layer(s) such as hidden layer(s) between the input layer and the output layer. Different layers may perform different kinds of transformations on respective inputs of the different layers. Signals can travel from the input layer to the output layer.
- a DNN is a feedforward network where data flows from the input layer to the output layer without looping back.
- a DNN is a fully connected network where each node in one layer is connected to all nodes in the next layer.
- a DNN is a recurrent neural network (RNN) where data can flow in any direction.
- RNN recurrent neural network
- a DNN is a CNN.
- a CNN can include an input layer, an output layer, and hidden layer(s) between the input layer and the output layer.
- the hidden layer(s) can include convolutional layer(s) (e.g., used in an encoder) that perform convolutions, such as a two-dimensional (2D) convolution.
- a 2D convolution performed in a convolution layer is between a convolution kernel (also referred to as a filter or a channel, such as a 5 x 5 matrix) and an input signal (e.g., a 2D matrix such as a 2D block, a 256 x 256 matrix) to the convolution layer.
- a dimension of the convolution kernel e.g., 5 x 5
- a dimension of the input signal e.g., 256 x 256).
- a portion (e.g., a 5 x 5 area) in the input signal (e.g., a 256 x 256 matrix) that is covered by the convolution kernel is smaller than an area (e.g., a 256 x 256 area) of the input signal, and thus can be referred to as a receptive field in the respective node in the next layer.
- each element of the convolution kernel is a weight that is applied to a corresponding sample in the receptive field, and thus the convolution kernel includes weights.
- a convolution kernel represented by a 5 x 5 matrix has 25 weights.
- a bias is applied to the output signal of the convolution layer, and the output signal is based on a sum of the dot product and the bias.
- the convolution kernel can shift along the input signal (e.g., a 2D matrix) by a size referred to as a stride, and thus the convolution operation generates a feature map or an activation map (e.g., another 2D matrix), which in turn contributes to an input of the next layer in the CNN.
- the input signal is a 2D block having 256 x 256 samples
- a stride is 2 samples (e.g., a stride of 2).
- the convolution kernel shifts along an X direction (e.g., a horizontal direction) and/or a Y direction (e.g., a vertical direction) by 2 samples.
- convolution kernels can be applied in the same convolution layer to the input signal to generate multiple feature maps, respectively, where each feature map can represent a specific feature of the input signal.
- a convolution layer with N channels i.e., N convolution kernels
- each convolution kernel having M x M samples, and a stride S can be specified as Conv: MxM cN sS.
- the hidden layer(s) can include deconvolutional layer(s) (e.g., used in a decoder) that perform deconvolutions, such as a 2D deconvolution.
- a deconvolution is an inverse of a convolution.
- a deconvolution layer with 192 channels, each deconvolution kernel having 5 x 5 samples, and a stride of 2 is specified as DeConv: 5x5 cl92 s2.
- a CNN has the following benefits.
- a number of learnable parameters (i.e., parameters to be trained) in a CNN can be significantly smaller than a number of learnable parameters in a DNN, such as a feedforward DNN.
- a relatively large number of nodes can share a same filter (e.g., same weights) and a same bias (if the bias is used), and thus the memory footprint can be reduced because a single bias and a single vector of weights can be used across all receptive fields that share the same filter.
- a convolution layer with a convolution kernel having 5 x 5 samples has 25 learnable parameters (e.g., weights).
- one channel uses 26 learnable parameters (e.g., 25 weights and one bias). If the convolution layer has N channels, the total learnable parameters is 26 x N. On the other hand, for a fully connected a layer in a DNN, 100 x 100 (i.e., 10000) weights are used for each node in the next layer. If the next layer has L nodes, then the total learnable parameters is 10000 x L.
- a CNN can further include one or more other layer(s), such as pooling layer(s), fully connected layer(s) that can connect every node in one layer to every node in another layer, normalization layer(s), and/or the like.
- Layers in a CNN can be arranged in any suitable order and in any suitable architecture (e.g., a feed-forward architecture, a recurrent architecture).
- a convolutional layer is followed by other layer(s), such as pooling layer(s), fully connected layer(s), normalization layer(s), and/or the like.
- a pooling layer can be used to reduce dimensions of data by combining outputs from a plurality of nodes at one layer into a single node in the next layer.
- a pooling operation for a pooling layer having a feature map as an input is described below. The description can be suitably adapted to other input signals.
- the feature map can be divided into sub-regions (e.g., rectangular sub-regions), and features in the respective sub-regions can be independently down-sampled (or pooled) to a single value, for example, by taking an average value in an average pooling or a maximum value in a max pooling.
- the pooling layer can perform a pooling, such as a local pooling, a global pooling, a max pooling, an average pooling, and/or the like.
- a pooling is a form of nonlinear down-sampling.
- a local pooling combines a small number of nodes (e.g., a local cluster of nodes, such as 2 x 2 nodes) in the feature map.
- a global pooling can combine all nodes, for example, of the feature map.
- the pooling layer can reduce a size of the representation, and thus reduce a number of parameters, a memory footprint, and an amount of computation in a CNN.
- a pooling layer is inserted between successive convolutional layers in a CNN.
- a pooling layer is followed by an activation function, such as a rectified linear unit (ReLU) layer.
- a pooling layer is omitted between successive convolutional layers in a CNN.
- a normalization layer can be an ReLU, a leaky ReLU, a generalized divisive normalization (GDN), an inverse GDN (IGDN), or the like.
- An ReLU can apply a non saturating activation function to remove negative values from an input signal, such as a feature map, by setting the negative values to zero.
- a leaky ReLU can have a small slope (e.g., 0.01) for negative values instead of a flat slope (e.g., 0). Accordingly, if a value x is larger than 0, then an output from the leaky ReLU is x. Otherwise, the output from the leaky ReLU is the value x multiplied by the small slope (e.g., 0.01). In an example, the slope is determined before training, and thus is not learnt during training.
- FIG. 9A shows an example of a block-wise image coding according to an embodiment of the disclosure.
- An image (980) can be partitioned into blocks, e.g., blocks (981)-(996).
- the blocks (981)-(996) can be compressed, for example, according to a scanning order. IN an example shown in FIG. 9A, the blocks (981)-(989) are already compressed, and the blocks (990)-(996) are to be compressed.
- An image can be treated as a block.
- the image is compressed without being split into blocks.
- the entire image can be the input of an E2E NIC framework.
- FIG. 9B shows an exemplary NIC framework (900) (e.g., a NIC system) according to an embodiment of the disclosure.
- the NIC framework (900) can be based on neural networks, such as DNNs and/or CNNs.
- the NIC framework (900) can be used to compress (e.g., encode) blocks and decompress (e.g., decode or reconstruct) compressed blocks (e.g., encoded blocks).
- the NIC framework (900) can include two sub-neural networks, a first sub-NN (951) and a second sub-NN (952) that are implemented using neural networks.
- the first sub-NN (951) can resemble an autoencoder and can be trained to generate a compressed block x of an input block x and decompress the compressed block x to obtain a reconstructed block x.
- the first sub-NN (951) can include a plurality of components (or modules), such as a main encoder neural network (or a main encoder network) (911), a quantizer (912), an entropy encoder (913), an entropy decoder (914), and a main decoder neural network (or a main encoder network) (915). Referring to FIG.
- the main encoder network (911) can generate a latent or a latent representation y from the input block x (e.g., an block to be compressed or encoded).
- the main encoder network (911) is implemented using a CNN.
- a relationship between the latent representation y and the input block x can be described using Eq. 2.
- y fi(x ⁇ 3 ⁇ 4)
- Eq. 2 where a parameter q ⁇ represents parameters, such as weights used in convolution kernels in the main encoder network (911) and biases (if biases are used in the main encoder network
- the latent representation y can be quantized using the quantizer (912) to generate a quantized latent y.
- the quantized latent y can be compressed, for example, using lossless compression by the entropy encoder (913) to generate the compressed block (e.g., an encoded block) x (931) that is a compressed representation x of the input block x.
- the entropy encoder (913) can use entropy coding techniques such as Huffman coding, arithmetic coding, or the like.
- the entropy encoder (913) uses arithmetic encoding and is an arithmetic encoder.
- the encoded block (931) is transmitted in a coded bitstream.
- the encoded block (931) can be decompressed (e.g., entropy decoded) by the entropy decoder (914) to generate an output.
- the entropy decoder (914) can use entropy coding techniques such as Huffman coding, arithmetic coding, or the like that correspond to the entropy encoding techniques used in the entropy encoder (913).
- the entropy decoder (914) uses arithmetic decoding and is an arithmetic decoder.
- lossless compression is used in the entropy encoder (913)
- lossless decompression is used in the entropy decoder (914)
- noises such as due to the transmission of the encoded block (931) are omissible
- the output from the entropy decoder (914) is the quantized latent y.
- the main decoder network (915) can decode the quantized latent y to generate the reconstructed block x.
- the main decoder network (915) is implemented using a CNN.
- a relationship between the reconstructed block x (i.e., the output of the main decoder network (915)) and the quantized latent y (i.e., the input of the main decoder network (915)) can be described using Eq. 3.
- x f (y; 3 ⁇ 4)
- a parameter q 2 represents parameters, such as weights used in convolution kernels in the main decoder network (915) and biases (if biases are used in the main decoder network (915)).
- the first sub-NN (951) can compress (e.g., encode) the input block x to obtain the encoded block (931) and decompress (e.g., decode) the encoded block (931) to obtain the reconstructed block x.
- the reconstructed block x can be different from the input block x due to quantization loss introduced by the quantizer (912).
- the second sub-NN (952) can learn the entropy model (e.g., a prior probabilistic model) over the quantized latent y used for entropy coding.
- the entropy model can be a conditioned entropy model, e.g., a Gaussian mixture model (GMM), a Gaussian scale model (GSM) that is dependent on the input block x.
- the second sub-NN (952) can include a context model NN (916), an entropy parameter NN (917), a hyper encoder (921), a quantizer (922), an entropy encoder (923), an entropy decoder (924), and a hyper decoder (925).
- the entropy model used in the context model NN (916) can be an autoregressive model over latent (e.g., the quantized latent y).
- the hyper encoder (921), the quantizer (922), the entropy encoder (923), the entropy decoder (924), and the hyper decoder (925) form a hyper neural network (e.g., a hyperprior NN).
- the hyper neural network can represent information useful for correcting context-based predictions.
- Data from the context model NN (916) and the hyper neural network can be combined by the entropy parameter NN (917).
- the entropy parameter NN (917) can generate parameters, such as mean and scale parameters for the entropy model such as a conditional Gaussian entropy model (e.g., the GMM).
- the quantized latent y from the quantizer (912) is fed into the context model NN (916).
- the quantized latent y from the entropy decoder (914) is fed into the context model NN (916).
- the context model NN (916) can be implemented using a neural network, such as a CNN.
- the context model NN (916) can generate an output o cm i based on a context y ⁇ L that is the quantized latent y available to the context model NN (916).
- the context y ⁇ L can include previously quantized latent at the encoder side or previously entropy decoded quantized latent at the decoder side.
- a relationship between the output o cm i and the input (e.g., y ⁇ i ) of the context model NN (916) can be described using Eq. 4.
- o C m,i h y ⁇ i , ⁇ ⁇ 3 ⁇ 4)
- a parameter q 3 represents parameters, such as weights used in convolution kernels in the context model NN (916) and biases (if biases are used in the context model NN (916)).
- the output o cm i from the context model NN (916) and an output o hc from the hyper decoder (925) are fed into the entropy parameter NN (917) to generate an output o ep .
- the entropy parameter NN (917) can be implemented using a neural network, such as a CNN.
- a relationship between the output o ep and the inputs (e.g., o cm i and o hc) of the entropy parameter NN (917) can be described using Eq. 5.
- a parameter q 4 represents parameters, such as weights used in convolution kernels in the entropy parameter NN (917) and biases (if biases are used in the entropy parameter NN (917)).
- the output o ep of the entropy parameter NN (917) can be used in determining (e.g., conditioning) the entropy model, and thus the conditioned entropy model can be dependent on the input block x, for example, via the output o hc from the hyper decoder (925).
- the output o ep includes parameters, such as the mean and scale parameters, used to condition the entropy model (e.g., GMM).
- the entropy model (e.g., the conditioned entropy model) can be employed by the entropy encoder (913) and the entropy decoder (914) in entropy coding and entropy decoding, respectively.
- the second sub-NN (952) can be described below.
- the latent y can be fed into the hyper encoder (921) to generate a hyper latent z.
- the hyper encoder (921) is implemented using a neural network, such as a CNN.
- a relationship between the hyper latent z and the latent y can be described using Eq. 6.
- z / 5 (y; ⁇ 3 ⁇ 4)
- Eq. 6 where a parameter q 5 represents parameters, such as weights used in convolution kernels in the hyper encoder (921) and biases (if biases are used in the hyper encoder (921)).
- the hyper latent z is quantized by the quantizer (922) to generate a quantized latent z.
- the quantized latent z can be compressed, for example, using lossless compression by the entropy encoder (923) to generate side information, such as encoded bits (932) from the hyper neural network.
- the entropy encoder (923) can use entropy coding techniques such as Huffman coding, arithmetic coding, or the like.
- the entropy encoder (923) uses arithmetic encoding and is an arithmetic encoder.
- the side information such as the encoded bits (932), can be transmitted in the coded bitstream, for example, together with the encoded block (931).
- the side information such as the encoded bits (932) can be decompressed (e.g., entropy decoded) by the entropy decoder (924) to generate an output.
- the entropy decoder (924) can use entropy coding techniques such as Huffman coding, arithmetic coding, or the like.
- the entropy decoder (924) uses arithmetic decoding and is an arithmetic decoder.
- the output from the entropy decoder (924) can be the quantized latent z.
- the hyper decoder (925) can decode the quantized latent z to generate the output o hc .
- a relationship between the output o hc and the quantized latent z can be described using Eq. 7.
- a parameter q 6 represents parameters, such as weights used in convolution kernels in the hyper decoder (925) and biases (if biases are used in the hyper decoder (925)).
- the compressed or encoded bits (932) can be added to the coded bitstream as the side information, which enables the entropy decoder (914) to use the conditional entropy model.
- the entropy model can be block-dependent and spatially adaptive, and thus can be more accurate than a fixed entropy model.
- the NIC framework (900) can be suitably adapted, for example, to omit one or more components shown in FIG. 9, to modify one or more components shown in FIG. 9, and/or to include one or more components not shown in FIG. 9.
- a NIC framework using a fixed entropy model includes the first sub-NN (951), and does not include the second sub-NN (952).
- a NIC framework includes the components in the NIC framework (900) except the entropy encoder (923) and the entropy decoder (924).
- one or more components in the NIC framework (900) shown in FIG. 9 are implemented using neural network(s), such as CNN(s).
- Each NN-based component e.g., the main encoder network (911), the main decoder network (915), the context model NN (916), the entropy parameter NN (917), the hyper encoder (921), or the hyper decoder (925)
- a NIC framework e.g., the NIC framework (900)
- a suitable architecture e.g., have any suitable combinations of layers
- include any suitable types of parameters e.g., weights, biases, a combination of weights and biases, and/or the like
- parameters e.g., weights, biases, a combination of weights and biases, and/or the like
- the main encoder network (911), the main decoder network (915), the context model NN (916), the entropy parameter NN (917), the hyper encoder (921), and the hyper decoder (925) are implemented using respective CNNs.
- FIG. 10 shows an exemplary CNN of the main encoder network (911) according to an embodiment of the disclosure.
- the main encoder network (911) includes four sets of layers where each set of layers includes a convolution layer 5x5 cl92 s2 followed by a GDN layer.
- Each set of layers includes a convolution layer 5x5 cl92 s2 followed by a GDN layer.
- One or more layers shown in FIG. 10 can be modified and /or omitted. Additional layer(s) can be added to the main encoder network (911).
- FIG. 11 shows an exemplary CNN of the main decoder network (915) according to an embodiment of the disclosure.
- the main decoder network (915) includes three sets of layers where each set of layers includes a deconvolution layer 5x5 cl92 s2 followed by an IGDN layer.
- the three sets of layers are followed by a deconvolution layer 5x5 c3 s2 followed by an IGDN layer.
- One or more layers shown in FIG. 11 can be modified and /or omitted. Additional layer(s) can be added to the main decoder network (915).
- FIG. 12 shows an exemplary CNN of the hyper encoder (921) according to an embodiment of the disclosure.
- the hyper encoder (921) includes a convolution layer 3x3 cl92 si followed by a leaky ReLU, a convolution layer 5x5 cl92 s2 followed by a leaky ReLU, and a convolution layer 5x5 cl92 s2.
- One or more layers shown in FIG. 12 can be modified and /or omitted. Additional layer(s) can be added to the hyper encoder (921).
- FIG. 13 shows an exemplary CNN of the hyper decoder (925) according to an embodiment of the disclosure.
- the hyper decoder (925) includes a deconvolution layer 5x5 cl92 s2 followed by a leaky ReLU, a deconvolution layer 5x5 c288 s2 followed by a leaky ReLU, and a deconvolution layer 3x3 c384 si.
- One or more layers shown in FIG. 13 can be modified and /or omitted. Additional layer(s) can be added to the hyper encoder (925).
- FIG. 14 shows an exemplary CNN of the context model NN (916) according to an embodiment of the disclosure.
- the context model NN (916) includes a masked convolution 5x5 c384 si for context prediction, and thus the context y ⁇ L in Eq. 4 includes a limited context (e.g., a 5x5 convolution kernel).
- the convolution layer in FIG. 14 can be modified. Additional layer(s) can be added to the context model NN (916).
- FIG. 15 shows an exemplary CNN of the entropy parameter NN (917) according to an embodiment of the disclosure.
- the entropy parameter NN (917) includes a convolution layer lxl c640 si followed by a leaky ReLU, a convolution layer lxl c512 si followed by leaky ReLU, and a convolution layer lxl c384 si.
- One or more layers shown in FIG. 15 can be modified and /or omitted. Additional layer(s) can be added to the entropy parameter NN (917).
- the NIC framework (900) can be implemented using CNNs, as described with reference to FIGs. 10-15.
- the NIC framework (900) can be suitably adapted such that one or more components (e.g., (911), (915), (916), (917), (921), and/or (925)) in the NIC framework (900) are implemented using any suitable types of neural networks (e.g., CNNs or non-CNN based neural networks).
- One or more other components the NIC framework (900) can be implemented using neural network(s).
- the NIC framework (900) that includes neural networks can be trained to leam the parameters used in the neural networks.
- the parameters represented by q 1 -q 6, such as the weights used in the convolution kernels in the main encoder network (911) and biases (if biases are used in the main encoder network (911)), the weights used in the convolution kernels in the main decoder network (915) and biases (if biases are used in the main decoder network (915)), the weights used in the convolution kernels in the hyper encoder (921) and biases (if biases are used in the hyper encoder (921)), the weights used in the convolution kernels in the hyper decoder (925) and biases (if biases are used in the hyper decoder (925)), the weights used in the convolution kernel(s) in the context model NN (916) and biases (if biases are used in the context
- the main encoder network (911) includes four convolution layers where each convolution layer has a convolution kernel of 5x5 and 192 channels.
- a number of the weights used in the convolution kernels in the main encoder network (911) is 19200 (i.e., 4x5x5x192).
- the parameters used in the main encoder network (911) include the 19200 weights and optional biases. Additional parameter(s) can be included when biases and/or additional NN(s) are used in the main encoder network (911).
- the NIC framework (900) includes at least one component or module built on neural network(s).
- the at least one component can include one or more of the main encoder network (911), the main decoder network (915), the hyper encoder (921), the hyper decoder (925), the context model NN (916), and the entropy parameter NN (917).
- the at least one component can be trained individually.
- the training process is used to learn the parameters for each component separately.
- the at least one component can be trained jointly as a group.
- the training process is used to learn the parameters for a subset of the at least one component jointly.
- the training process is used to learn the parameters for all of the at least one component, and thus is referred to as an E2E optimization.
- the weights (or the weight coefficients) of the one or more components can be initialized.
- the weights are initialized based on pre-trained corresponding neural network model(s) (e.g., DNN models, CNN models).
- the weights are initialized by setting the weights to random numbers.
- a set of training blocks can be employed to train the one or more components, for example, after the weights are initialized.
- the set of training blocks can include any suitable blocks having any suitable size(s).
- the set of training blocks includes blocks from raw images, natural images, computer-generated images, and/or the like that are in the spatial domain.
- the set of training blocks includes blocks from residue blocks or residue images having residue data in the spatial domain.
- the residue data can be calculated by a residue calculator (e.g., the residue calculator (723)).
- raw images and/or residue images including residue data can be used directly to train neural networks in a NIC framework.
- raw images, residue images, blocks from raw images, and/or blocks from residue images can be used to train neural networks in a NIC framework.
- a training block t of the set of training blocks can be passed through the encoding process in FIG. 9B to generate a compressed representation (e.g., encoded information, for example, to a bitstream).
- the encoded information can be passed through the decoding process described in FIG. 9B to compute and reconstruct a reconstructed block t.
- a quality loss function e.g., a distortion or distortion loss
- D(t, t) can be used to indicate the reconstruction quality, such as a difference between the reconstruction (e.g., the reconstructed block t) and an original block (e.g., the training block t).
- a rate (or a rate loss) R can be used to indicate the bit consumption of the compressed representation.
- the rate loss R further includes the side information, for example, used in determining a context model.
- differentiable approximations of quantization can be used in E2E optimization.
- noise injection is used to simulate quantization, and thus quantization is simulated by the noise injection instead of being performed by a quantizer (e.g., the quantizer (912)).
- a quantizer e.g., the quantizer (912)
- a bits per pixel (BPP) estimator can be used to simulate an entropy coder, and thus entropy coding is simulated by the BPP estimator instead of being performed by an entropy encoder (e.g., (913)) and an entropy decoder (e.g., (914)). Therefore, the rate loss R in the loss function L shown in Eq.
- a trade-off hyperparameter l in Eq. 1 can be used to optimize a joint R-D loss L where L as a summation of /.D and R can be optimized.
- the training process can be used to adjust the parameters of the one or more components (e.g., (911) (915)) in the NIC framework (900) such that the joint R-D loss L is minimized or optimized.
- a trade-off hyperparameter l can be used to optimize the joint Rate-Distortion (R-D) loss as:
- L(x, x, r x , ... , r N , y) (x, x) + R( ⁇ ? s t , ⁇ ? iq) + bE Eq. 8
- E measures the distortion of the decoded block residuals compared with the original block residuals before encoding, which acts as regularization loss for the residual encoding/decoding DNNs and the encoding/decoding DNNs.
- b is a hyperparameter used to balance the importance of the regularization loss.
- the distortion loss D ( t , t) is expressed as a peak signal-to-noise ratio (PSNR) that is a metric based on mean squared error, a multiscale structural similarity (MS-SSIM) quality index, a weighted combination of the PSNR and MS-SSIM, or the like.
- PSNR peak signal-to-noise ratio
- MS-SSIM multiscale structural similarity
- the target of the training process is to train the encoding neural network (e.g., the encoding DNN), such as a video encoder to be used on an encoder side and the decoding neural network (e.g., the decoding DNN), such as a video decoder to be used on a decoder side.
- the encoding neural network can include the main encoder network (911), the hyper encoder (921), the hyper decoder (925), the context model NN (916), and the entropy parameter NN (917).
- the decoding neural network can include the main decoder network (915), the hyper decoder (925), the context model NN (916), and the entropy parameter NN (917).
- the video encoder and/or the video decoder can include other component(s) that are based on NN(s) and/or not based on NN(s).
- the NIC framework (e.g., the NIC framework (900)) can be trained in an E2E fashion.
- the encoding neural network and the decoding neural network are updated jointly in the training process based on backpropagated gradients in an E2E fashion.
- one or more components in the NIC framework (900) can be used to encode and/or decode blocks.
- the video encoder is configured to encode the input block x into the encoded block (931) to be transmitted in the bitstream.
- the video encoder can include multiple components in the NIC framework (900).
- the corresponding video decoder is configured to decode the encoded block (931) in the bitstream into the reconstructed block x.
- the video decoder can include multiple components in the NIC framework (900).
- the video encoder includes all the components in the NIC framework (900), for example, when content-adaptive online training is employed.
- FIG. 16A shows an exemplary video encoder (1600 A) according to an embodiment of the disclosure.
- the video encoder (1600 A) includes the main encoder network (911), the quantizer (912), the entropy encoder (913), and the second sub-NN (952) that are described with reference to FIG. 9B and detailed descriptions are omitted for purposes of brevity.
- FIG. 16B shows an exemplary video decoder (1600B) according to an embodiment of the disclosure.
- the video decoder (1600B) can correspond to the video encoder (1600A).
- the video decoder (1600B) can include the main decoder network (915), the entropy decoder (914), the context model NN (916), the entropy parameter NN (917), the entropy decoder (924), and the hyper decoder (925).
- the video encoder (1600A) can generate the encoded block (931) and the encoded bits (932) to be transmitted in the bitstream.
- the video decoder (1600B) can receive and decode the encoded block (931) and the encoded bits (932).
- FIGs. 17-18 show an exemplary video encoder (1700) and a corresponding video decoder (1800), respectively, according to embodiments of the disclosure.
- the encoder (1700) includes the main encoder network (911), the quantizer (912), and the entropy encoder (913). Examples of the main encoder network (911), the quantizer (912), and the entropy encoder (913) are described with reference to FIG. 9B.
- the video decoder (1800) includes the main decoder network (915) and the entropy decoder (914). Examples of the main decoder network (915) and the entropy decoder (914) are described with reference to FIG. 9B.
- the video encoder (1700) can generate the encoded block (931) to be transmitted in the bitstream.
- the video decoder (1800) can receive and decode the encoded block (931).
- the NIC framework (900) including the video encoder and the video decoder can be trained based on images and/or blocks in the set of training images.
- one or more blocks to be compressed (e.g., encoded) and/or transmitted have properties that are significantly different from the set of training blocks.
- encoding and decoding the one or more blocks using the video encoder and the video decoder trained based on the set of training blocks, respectively can lead to a relatively poor R-D loss F (e.g., a relatively large distortion and/or a relatively large bit rate). Therefore, aspects of the disclosure describe a content-adaptive online training method for NIC, such as a block- wise content-adaptive online training method for NIC.
- an input image can be split into blocks and one or more of the blocks can be used to update one or more parameters in a pretrained NIC framework to be one or more replacement parameters by optimizing rate-distortion performance.
- Neural network update information indicating the one or more replacement parameters or a subset of the one or more replacement parameters can be encoded into a bitstream along with the encoded one or more of the blocks.
- a video decoder can decode the encoded one or more of the blocks and can achieve better compression performance by using the one or more replacement parameters or the subset of the one or more replacement parameters.
- the block-wise content-adaptive online training method can be used as a preprocessing step (e.g., a pre-encoding step) for boosting the compression performance of a pretrained E2E NIC compression method.
- the NIC framework (900), the video encoder, and the video decoder that are trained by the set of training blocks are referred to as the pretrained NIC framework (900), the pretrained video encoder, and the pretrained video decoder, respectively.
- Parameters in the pretrained NIC framework (900), the pretrained video encoder, or the pretrained video decoder are referred to as NIC pretrained parameters, encoder pretrained parameters, and decoder pretrained parameters, respectively.
- the NIC pretrained parameters includes the encoder pretrained parameters and the decoder pretrained parameters.
- the encoder pretrained parameters and the decoder pretrained parameters do not overlap where none of the encoder pretrained parameters is included in the decoder pretrained parameters.
- the encoder pretrained parameters e.g., pretrained parameters in the main encoder network (911)) in (1700) and the decoder pretrained parameters (e.g., pretrained parameters in the main decoder network (915)) in (1800) do not overlap.
- the encoder pretrained parameters and the decoder pretrained parameters overlap where at least one of the encoder pretrained parameters is included in the decoder pretrained parameters.
- the encoder pretrained parameters e.g., pretrained parameters in the context model NN (916)
- the decoder pretrained parameters e.g., the pretrained parameters in the context model NN (916)
- the NIC pretrained parameters can be obtained based on blocks and/or images in the set of training blocks.
- the content- adaptive online training process can be referred to as a finetuning process and is described below.
- One or more pretrained parameters in the NIC pretrained parameters in the pretrained NIC framework (900) can further be trained (e.g., finetuned) based on the one or more blocks to be encoded and/or transmitted where the one or more blocks can be different from the set of training blocks.
- the one or more pretrained parameters used in the NIC pretrained parameters can be finetuned by optimizing the joint R-D loss L based on the one or more blocks.
- the one or more pretrained parameters that have been finetuned by the one or more blocks are referred to as the one or more replacement parameters or the one or more finetuned parameters.
- neural network update information is encoded into a bitstream to indicate the one or more replacement parameters or a subset of the one or more replacement parameters.
- the NIC framework (900) is updated (or finetuned) where the one or more pretrained parameters are replaced by the one or more replacement parameters, respectively.
- the one or more pretrained parameters includes a first subset of the one or more pretrained parameters and a second subset of the one or more pretrained parameters.
- the one or more replacement parameters includes a first subset of the one or more replacement parameters and a second subset of the one or more replacement parameters.
- the first subset of the one or more pretrained parameters is used in the pretrained video encoder and is replaced by the first subset of the one or more replacement parameters, for example, in the training process.
- the pretrained video encoder is updated to the updated video encoder by the training process.
- the neural network update information can indicate the second subset of the one or more replacement parameters that is to replace the second subset of the one or more replacement parameters.
- the one or more blocks can be encoded using the updated video encoder and transmitted in the bitstream with the neural network update information.
- the second subset of the one or more pretrained parameters is used in the pretrained video decoder.
- the pretrained video decoder receives and decodes the neural network update information to determine the second subset of the one or more replacement parameters.
- the pretrained video decoder is updated to the updated video decoder when the second subset of the one or more pretrained parameters in the pretrained video decoder is replaced by the second subset of the one or more replacement parameters.
- the one or more encoded blocks can be decoded using the updated video decoder.
- FIGs. 16A-16B show an example of the first scenario.
- the one or more pretrained parameters include N 1 pretrained parameters in the pretrained context model NN (916) and N2 pretrained parameters in the pretrained main decoder network (915).
- the first subset of the one or more pretrained parameters include the N 1 pretrained parameters
- the second subset of the one or more pretrained parameters are identical to the one or more pretrained parameters.
- the N1 pretrained parameters in the pretrained context model NN (916) can be replaced by N1 corresponding replacement parameters such that the pretrained video encoder (1600A) can be updated to the updated video encoder (1600A).
- the pretrained context model NN (916) is also updated to be the updated context model NN (916).
- the N1 pretrained parameters can be replaced by the N 1 corresponding replacement parameters and the N2 pretrained parameters can be replaced by N2 corresponding replacement parameters, updating the pretrained context model NN (916) to be the updated context model NN (916) and updating the pretrained main decoder network (915) to be the updated main decoder network (915).
- the pretrained video decoder (1600B) can be updated to the updated video decoder (1600B).
- the pretrained video encoder is not updated and continues to be the pretrained video encoder after the training process.
- the neural network update information indicates the one or more replacement parameters.
- the one or more blocks can be encoded using the pretrained video encoder and transmitted in the bitstream with the neural network update information.
- the pretrained video decoder can receive and decode the neural network update information to determine the one or more replacement parameters.
- the pretrained video decoder is updated to the updated video decoder when the one or more pretrained parameters in the pretrained video decoder is replaced by the one or more replacement parameters.
- the one or more encoded blocks can be decoded using the updated video decoder.
- FIGs. 16A-16B show an example of the second scenario.
- the one or more pretrained parameters include N2 pretrained parameters in the pretrained main decoder network (915).
- the pretrained video encoder e.g., the pretrained video encoder (1600A)
- the pretrained video encoder (1600A) continues to be the pretrained video encoder after the training process.
- the N2 pretrained parameters can be replaced by N2 corresponding replacement parameters, which updates the pretrained main decoder network (915) to the updated main decoder network (915).
- the pretrained video decoder (1600B) can be updated to the updated video decoder (1600B).
- the one or more pretrained parameters are used in the pretrained video encoder and are replaced by the one or more replacement parameters, for example, in the training process.
- the pretrained video encoder is updated to the updated video encoder by the training process.
- the one or more blocks can be encoded using the updated video encoder and transmitted in the bitstream. No neural network update information is encoded in the bitstream.
- the pretrained video decoder is not updated and remains to be the pretrained video decoder.
- the one or more encoded blocks can be decoded using the pretrained video decoder.
- FIGs. 16A-16B show an example of the third scenario.
- the one or more pretrained parameters are in the pretrained main encoder network (911).
- the one or more pretrained parameters in the pretrained main encoder network (911) can be replaced by the one or more replacement parameters such that the pretrained video encoder (1600 A) can be updated to be the updated video encoder (1600A).
- the pretrained main encoder network (911) is also updated to be the updated main encoder network (911).
- the pre trained video decoder (1600B) is not updated.
- video decoding may be performed by pretrained decoders having different capabilities, including decoders with and without capabilities to update the pretrained parameters.
- compression performance can be increased by coding the one or more blocks with the updated video encoder and/or the updated video decoder as compared to coding the one or more blocks with the pretrained video encoder and the pretrained video decoder. Therefore, the content-adaptive online training method can be used to adapt a pretrained NIC framework (e.g., the pretrained NIC framework (900)) to target block content (e.g., the one or more blocks to be transmitted), and thus finetuning the pretrained NIC framework. Accordingly, the video encoder on the encoder side and/or the video decoder on the decoder side can be updated.
- a pretrained NIC framework e.g., the pretrained NIC framework (900)
- target block content e.g., the one or more blocks to be transmitted
- the content- adaptive online training method can be used as a preprocessing step (e.g., pre-encoding step) for boosting the compression performance of a pretrained E2E NIC compression method.
- the one or more blocks include a single input block, and the finetuning process is performed with the single input block.
- the NIC framework (900) is trained and updated (e.g., finetuned) based on the single input block.
- the updated video encoder on the encoder side and/or the updated video decoder on the decoder side can be used to code the single input block and optionally other input blocks.
- the neural network update information can be encoded into the bitstream together with the encoded single input block.
- the one or more blocks include multiple input blocks, and the finetuning process is performed with the multiple input blocks.
- the NIC framework (900) is trained and updated (e.g., finetuned) based on the multiple input blocks.
- the updated video encoder on the encoder side and/or the updated decoder on the decoder side can be used to code the multiple input blocks and optionally other input blocks.
- the neural network update information can be encoded into the bitstream together with the encoded multiple input blocks.
- the rate loss R can increase with the signaling of the neural network update information in the bitstream.
- the neural network update information is signaled for each encoded block, and a first increase to the rate loss R is used to indicate the increase to the rate loss R due to the signaling of the neural network update information per block.
- the neural network update information is signaled for and shared by the multiple input blocks, and a second increase to the rate loss R is used to indicate the increase to the rate loss R due to the signaling of the neural network update information per block. Because the neural network update information is shared by the multiple input blocks, the second increase to the rate loss R can be less than the first increase to the rate loss R.
- it can be advantageous to finetune the NIC framework using the multiple input blocks.
- the one or more pretrained parameters to be updated are in one component of the pretrained NIC framework (900).
- the one component of the pretrained NIC framework (900) is updated based on the one or more replacement parameters, and other components of the pretrained NIC framework (900) are not updated.
- the one component can be the pretrained context model NN (916), the pretrained entropy parameter NN (917), the pretrained main encoder network (911), the pretrained main decoder network (915), the pretrained hyper encoder (921), or the pretrained hyper decoder (925).
- the pretrained video encoder and/or the pretrained video decoder can be updated depending on which of the components in the pretrained NIC framework (900) is updated.
- the one or more pretrained parameters to be updated are in the pretrained context model NN (916), and thus the pretrained context model NN (916) is updated and the remaining components (911), (915), (921), (917), and (925) are not updated.
- the pretrained video encoder on the encoder side and the pretrained video decoder on the decoder side include the pretrained context model NN (916), and thus both the pretrained video encoder and the pretrained video decoder are updated.
- the one or more pretrained parameters to be updated are in the pretrained hyper decoder (925), and thus the pretrained hyper decoder (925) is updated and the remaining components (911), (915), (916), (917), and (921) are not updated.
- the pretrained video encoder is not updated and the pretrained video decoder is updated.
- the one or more pretrained parameters to be updated are in multiple components of the pretrained NIC framework (900).
- the multiple components of the pretrained NIC framework (900) are updated based on the one or more replacement parameters.
- the multiple components of the pretrained NIC framework (900) include all the components configured with neural networks (e.g., DNNs, CNNs).
- the multiple components of the pretrained NIC framework (900) include the CNN- based components: the pretrained main encoder network (911), the pretrained main decoder network (915), the pretrained context model NN (916), the pretrained entropy parameter NN (917), the pretrained hyper encoder (921), and the pretrained hyper decoder (925).
- the one or more pretrained parameters to be updated are in the pre trained video encoder of the pretrained NIC framework (900). In an example, the one or more pretrained parameters to be updated are in the pretrained video decoder of the NIC framework (900). In an example, the one or more pretrained parameters to be updated are in the pretrained video encoder and the pretrained video decoder of the pretrained NIC framework (900).
- the NIC framework (900) can be based on neural networks, for example, one or more components in the NIC framework (900) can include neural networks, such as CNNs, DNNs, and/or the like. As described above, the neural networks can be specified by different types of parameters, such as weights, biases, and the like.
- Each neural network- based component e.g., the context model NN (916), the entropy parameter NN (917), the main encoder network (911), the main decoder network (915), the hyper encoder (921), or the hyper decoder (925)
- suitable parameters such as respective weights, biases, or a combination of weights and biases.
- the weights can include elements in convolution kernels.
- One or more types of parameters can be used to specify the neural networks.
- the one or more pretrained parameters to be updated are bias term(s), and only the bias term(s) are replaced by the one or more replacement parameters.
- the one or more pretrained parameters to be updated are weights, and only the weights are replaced by the one or more replacement parameters.
- the one or more pretrained parameters to be updated include the weights and bias term(s), and all the pretrained parameters including the weights and bias term(s) are replaced by the one or more replacement parameters.
- other parameters can be used to specify the neural networks, and the other parameters can be finetuned.
- the finetuning process can include multiple epochs (e.g., iterations) where the one or more pretrained parameters are updated in an iterative finetuning process.
- the finetuning process can stop when a training loss has flattened or is about to flatten.
- the finetuning process stops when the training loss (e.g., a R-D loss L) is below a first threshold.
- the finetuning process stops when a difference between two successive training losses is below a second threshold.
- Two hyperparameters e.g., a step size and a maximum number of steps
- a loss function e.g., an R-D loss L
- the maximum number of iterations can be used as a threshold of a maximum number of iterations to terminate the finetuning process.
- the finetuning process stops when a number of iterations reaches the maximum number of iterations.
- the step size can indicate a learning rate of the online training process (e.g., the online finetuning process).
- the step size can be used in a gradient descent algorithm or a backpropagation calculation performed in the finetuning process.
- a step size can be determined using any suitable method.
- the step size for each block in an image can be different.
- different step sizes can be assigned for an image in order to achieve a better compression result (e.g., a better R-D loss L).
- a video encoder and a video decoder based on a NIC framework can encode and decode an image directly.
- the block-wise content-adaptive online training method can be adapted to update certain parameters in the NIC framework and thus the video encoder and/or the video decoder by using one or more images directly. Different images can have different step sizes to achieve optimized compression result.
- step sizes are used for blocks with different types of contents to achieve optimal results. Different types can refer to different variances.
- the step size is determined based on a variance of a block used to update a NIC framework. For example, a step size of a block having a high variance is larger than a step size of a block having a low variance where the high variance is larger than the low variance.
- a step size is chosen based on characteristics of a block or an image, such as RGB variance of the block.
- a step size is chosen based on RD performance (e.g., a R-D loss L) of the block.
- RD performance e.g., a R-D loss L
- Multiple sets of replacement parameter(s) can be generated based on different step sizes, and the set with the better compression performance (e.g., a smaller R-D loss) can be chosen.
- a first step size can be used to ran a certain number (e.g., 100) iterations. Then, a second step size (e.g., the first step size plus or minus a size increment) can be used to ran the certain number of iterations. Results from the first step size and the second step size can be compared to determine a step size to be used. More than two step sizes may be tested to determine an optimal step size.
- a certain number e.g. 100
- a second step size e.g., the first step size plus or minus a size increment
- a step size can vary during the finetuning process.
- the step size can have an initial value at an onset of the finetuning process, and the initial value can be reduced (e.g., halved) at a later stage of the finetuning process, for example, after a certain number of iterations to achieve a finer tuning.
- the step size or the learning rate can be varied by a scheduler during the iterative online training.
- the scheduler can include a parameter adjustment method used to adjust the step size.
- the scheduler can determine a value for the step size such that the step size can increase, decrease, or remain constant in a number of intervals. In an example, the learning rate is altered in each step by the scheduler.
- a single scheduler or multiple different schedulers can be used for different blocks.
- multiple sets of replacement parameter(s) can be generated based on the multiple schedulers, and one of the multiple sets of replacement parameter(s) with the better compression performance (e.g., a smaller R-D loss) can be chosen.
- multiple learning rate schedules are assigned for different blocks in order to achieve better compression result.
- all blocks in an image share a same learning rate schedule.
- selection of learning rate schedules is based on characteristics of a block, such as a RGB variance of the block.
- selection of learning rate schedules is based on the RD performance of the block.
- different blocks can be used to update different parameters in different components (e.g., the context model NN (916) or the hyper decoder (925)) in the NIC framework.
- a first block is used to update parameters in the context model NN (916)
- a second block is used to update parameters in the hyper decoder (925).
- different blocks can be used to update different types of parameters (e.g., biases or weights) in the NIC framework.
- a first block is used to update at least one bias in one or more neural networks in the NIC framework
- a second block is used to update at least one weight in one or more neural networks in the NIC framework.
- multiple blocks e.g., all blocks
- the one or more parameters to be updated are chosen based on a characteristic of a block, such as a RGB variance of the block. In an embodiment, the one or more parameters to be updated are chosen based on a RD performance of the block.
- one or more updated parameters can be computed for the respective one or more replacement parameters.
- the one or more updated parameters are calculated as differences between the one or more replacement parameters and the corresponding one or more pretrained parameters.
- the one or more updated parameters are the one or more replacement parameters, respectively.
- the one or more updated parameters can be generated from the one or more replacement parameters, for example, using a certain linear or nonlinear transform, and the one or more updated parameters are representative parameter(s) generated based on the one or more replacement parameters.
- the one or more replacement parameters are transformed into the one or more updated parameters for better compression.
- a first subset of the one or more updated parameters corresponds to the first subset of the one or more replacement parameters, and a second subset of the one or more updated parameters corresponds to the second subset of the one or more replacement parameters.
- different blocks have different relationships between the one or more updated parameters and the one or more replacement parameters.
- the one or more updated parameters are calculated as differences between the one or more replacement parameters and the corresponding one or more pretrained parameters.
- the one or more updated parameters are the one or more replacement parameters, respectively.
- multiple blocks e.g., all blocks in an image have a same relationship between the one or more updated parameters and the one or more replacement parameters.
- the relationship between the one or more updated parameters and the one or more replacement parameters is chosen based on characteristics of a block, such as a RGB variance of the block. In an embodiment, the relationship between the one or more updated parameters and the one or more replacement parameters is chosen based on a RD performance of the block.
- the one or more updated parameters can be compressed, for example, using LZMA2 that is a variation of a Lempel-Ziv-Markov chain algorithm (LZMA), a bzip2 algorithm, or the like.
- LZMA2 Lempel-Ziv-Markov chain algorithm
- bzip2 bzip2 algorithm
- compression is omitted for the one or more updated parameters.
- the one or more updated parameters or the second subset of the one or more updated parameters can be encoded into the bitstream as the neural network update information where the neural network update information indicates the one or more replacement parameters or the second subset of the one or more replacement parameters.
- compression methods for the one or more updated parameters are different for different blocks.
- the LZMA2 is used to compress the one or more updated parameters
- the bzip2 is used to compress the one or more updated parameters.
- a same compression method is used to compress the one or more updated parameters for multiple blocks (e.g., all blocks) in an image.
- a compression method is chosen based on characteristics of a block, such as a RGB variance of the block.
- a compression method is chosen based on RD performance of the block.
- the pretrained video encoder on the encoder side can be updated or finetuned based on (i) the first subset of the one or more replacement parameters or (ii) the one or more replacement parameters.
- An input block e.g., one of the one or more blocks used to in the finetuning process
- the bitstream includes both the encoded block and the neural network update information.
- the neural network update information is decoded (e.g., decompressed) by the pretrained video decoder to obtain the one or more updated parameters or the second subset of the one or more updated parameters.
- the one or more replacement parameters or the second subset of the one or more replacement parameters can be obtained based on the relationship between the one or more updated parameters and the one or more replacement parameters described above.
- the pretrained video decoder can be finetuned and the updated video decoded can be used to decode the encoded block, as described above.
- the NIC framework can include any type of neural networks and use any neural network-based image compression methods, such as a context-hyperprior encoder- decoder framework (e.g., the NIC framework shown in FIG. 9B), a scale-hyperprior encoder- decoder framework, a Gaussian Mixture Likelihoods framework and variants of the Gaussian Mixture Likelihoods framework, an RNN-based recursive compression method and variants of the RNN-based recursive compression method, and the like.
- a context-hyperprior encoder- decoder framework e.g., the NIC framework shown in FIG. 9B
- a scale-hyperprior encoder- decoder framework e.g., the NIC framework shown in FIG. 9B
- a Gaussian Mixture Likelihoods framework and variants of the Gaussian Mixture Likelihoods framework e.g., the NIC framework shown in FIG. 9B
- the content-adaptive online training methods and apparatus in the disclosure can have the following benefits. Adaptive online training mechanisms are exploited to improve the NIC coding efficiency.
- pretrained parameters in the various types of pretrained frameworks can be replaced by using online training with blocks to be encoded and transmitted ⁇
- FIG. 19 shows a flow chart outlining a process (1900) according to an embodiment of the disclosure.
- the process (1900) can be used to encode a block, such as a block in a raw image or a block in a residue image.
- the process (1900) is executed by processing circuitry, such as the processing circuitry in the terminal devices (310), (320), (330) and (340), the processing circuitry that performs functions of the video encoder (1600A), the processing circuitry that performs functions of the video encoder (1700).
- the processing circuitry performs a combination of functions of (i) one of the video encoder (403), (603), and (703) and (ii) one of the video encoder (1600A) and the video encoder (1700).
- the process (1900) is implemented in software instructions, thus when the processing circuitry executes the software instructions, the processing circuitry performs the process (1900).
- the process starts at (S1901).
- an NIC framework is based on neural networks.
- the NIC framework is the NIC framework (900) described with reference to FIG. 9B.
- the NIC framework can be based on CNNs, such as described with reference to FIGs. 10-15.
- a video encoder e.g., (1600A) or (1700)
- a corresponding video decoder e.g., (1600B) or (1800)
- the NIC framework based on neural networks are pretrained, and thus the video encoder and the video decoder are pretrained.
- the process (1900) proceeds to (S1910).
- a finetuning process is performed on the NIC framework based on one or more blocks (or input block(s)).
- the input block(s) can be any suitable block(s) having any suitable size(s).
- the input block(s) include block(s) in a raw image, a natural image, a computer-generated image, and/or the like that are in the spatial domain.
- the input block(s) include residue data in the spatial domain, for example, calculated by a residue calculator (e.g., the residue calculator (723)).
- a residue calculator e.g., the residue calculator (723)
- Components in various apparatuses can be suitably combined to achieve (S1910), for example, referring to FIGs. 7 and 9, the residue block from the residue calculator can be fed into the main encoder network (911) in the NIC framework.
- One of more parameters (e.g., one or more pretrained parameters) in one or more neural networks (e.g., one or more pretrained neural networks) in the NIC framework (e.g., the pretrained NIC framework) can be updated to be one or more replacement parameters, respectively, as described above.
- the one of more parameters in the one or more neural networks are being updated during the training process described in (S1910), for example, in each step.
- At least one neural network in the video encoder (e.g., the pretrained video encoder) is configured with a first subset of the one or more pretrained parameters, and thus the at least one neural network in the video encoder can be updated based on a corresponding first subset of the one or more replacement parameters.
- the first subset of the one or more replacement parameters includes all of the one or more replacement parameters.
- the at least one neural network in the video encoder is updated when the first subset of the one or more pretrained parameters is replaced with the first subset of the one or more replacement parameters, respectively.
- the at least one neural network in the video encoder is updated iteratively in the finetuning process.
- none of the one or more pretrained parameters are included in the video encoder, and thus the video encoder is not updated and remains the pretrained video encoder.
- one of the one or more blocks can be encoded using the video encoder having the at least one updated neural network where the video encoder is configured with the first subset of the one or more replacement parameters.
- the one of the one or more blocks is encoded after the at least one neural network in the video encoder is updated.
- the step (SI 920) can be suitably adapted.
- the video encoder is not updated when none of the one or more replacement parameters are included in the at least one neural network in the video encoder, and thus the one of the one or more blocks can be encoded using the pretrained video encoder (e.g., the video encoder including the at least one pretrained neural network).
- neural network update information indicating a second subset of the one or more replacement parameters can be encoded into the bitstream.
- the second subset of the one or more replacement parameters is to be used to update at least one neural network in the video decoder on the decoder side.
- the step (S1930) can be omitted, and none of the neural networks in the video decoder is updated, for example, if the second subset of the one or more replacement parameters includes no parameters and no neural network update information is signaled in the bitstream.
- the bitstream including the encoded one of the one or more blocks and the neural network update information can be transmitted.
- the step (S1940) can be suitably adapted. For example, if the step (S1930) is omitted, the bitstream does not include the neural network update information.
- the process (1900) proceeds to (S1999), and terminates.
- the process (1900) can be suitably adapted to various scenarios and steps in the process (1900) can be adjusted accordingly.
- One or more of the steps in the process (1900) can be adapted, omitted, repeated, and/or combined. Any suitable order can be used to implement the process (1900).
- Additional step(s) can be added.
- the one or more blocks are encoded in (S1920), and are transmitted in (S1940).
- the one of the one or more blocks is encoded by the updated video encoder and transmitted in the bitstream.
- the finetuning process is based on the one or more blocks, the finetuning process is based on the context to be encoded, and thus is context-based.
- the neural network update information further indicates what parameter(s) the second subset of the one or more pretrained parameters (or the corresponding second subset of the one or more replacement parameters) are so that corresponding pretrained parameter(s) in the video decoder can be updated.
- the neural network update information can indicate component information (e.g., (915)), layer information (e.g., the fourth layer DeConv: 5x5 c3 s2), channel information (e.g., the second channel), and/or the like of the second subset of the one or more pretrained parameters. Therefore, referring to FIG.
- the second subset of the one or more replacement parameters includes the convolution kernel of the second channel of DeConv: 5x5 c3 s2 in the main decoder network (915).
- the convolution kernel of the second channel of DeConv: 5x5 c3 s2 in the pretrained main decoder network (915) is updated.
- the component information e.g., (915)
- the layer information e.g., the fourth layer DeConv:
- the channel information e.g., the second channel
- the like of the second subset of the one or more pretrained parameters are pre-determined and stored in the pretrained video decoder, and thus are not signaled.
- FIG. 20 shows a flow chart outlining a process (2000) according to an embodiment of the disclosure.
- the process (2000) can be used in the reconstruction of an encoded block.
- the process (2000) is executed by processing circuitry, such as the processing circuitry in the terminal devices (310), (320), (330) and (340), the processing circuitry that performs functions of the video decoder (1600B), the processing circuitry that performs functions of the video decoder (1800).
- the processing circuitry performs a combination of functions of (i) one of the video decoder (410), the video decoder (510), and the video decoder (810) and (ii) one of the video decoder (1600B) or the video decoder (1800).
- the process (2000) is implemented in software instructions, thus when the processing circuitry executes the software instructions, the processing circuitry performs the process (2000).
- the process starts at (S2001).
- an NIC framework is based on neural networks.
- the NIC framework is the NIC framework (900) described with reference to FIG. 9B.
- the NIC framework can be based on CNNs, such as described with reference to FIGs. 10-15.
- a video decoder e.g., (1600B) or (1800)
- the NIC framework based on neural networks can be pretrained.
- the video decoder can be pretrained with pretrained parameters.
- the process (2000) proceeds to (S2010).
- first neural network update information in a coded bitstream can be decoded.
- the first neural network update information can be for a first neural network in the video decoder.
- the first neural network can be configured with first pretrained parameters.
- the first neural network update information can correspond to a first block in an image to be reconstructed and indicate a first replacement parameter corresponding to a first pretrained parameter in the first pretrained parameters.
- the first pretrained parameter is a pretrained bias term.
- the first pretrained parameter is a pretrained weight coefficient.
- the video decoder includes multiple neural networks.
- the first neural network update information can indicate update information for one or more remaining neural networks of the multiple neural networks.
- the first neural network update information further indicates one or more replacement parameters for the one or more remaining neural networks in the multiple neural networks.
- the one or more replacement parameters correspond to one or more respective pretrained parameters for the one or more remaining neural networks.
- each of the first pre trained parameter and the one or more pretrained parameters is a respective pretrained bias term.
- each of the first pretrained parameter and the one or more pretrained parameters is a respective pretrained weight coefficient.
- the first pretrained parameter and the one or more pretrained parameters include one or more pretrained bias terms and one or more pretrained weight coefficients in the multiple neural networks.
- the first neural network update information indicates update information for a subset of the multiple neural networks, and a remaining subset of the multiple neural networks is not updated.
- the video decoder is the video decoder (1800) shown in FIG. 18.
- the first neural network is the main decoder network (915).
- the video decoder is the video decoder (1600B) shown in FIG. 16B.
- the multiple neural networks in the video decoder include the main decoder network (915), the context model NN (916), the entropy parameter NN (917), and the hyper decoder (925).
- the first neural network is one of the main decoder network (915), the context model NN (916), the entropy parameter NN (917), and the hyper decoder (925), such as the context model NN (916).
- the first neural network update information further includes the one or more replacement parameters for the one or more remaining neural networks (e.g., the main decoder network (915), the entropy parameter NN (917), and/or the hyper decoder (925)) in the video decoder.
- the main decoder network (915) the entropy parameter NN (917), and/or the hyper decoder (925)
- the first replacement parameter can be determined based on the first neural network update information.
- an updated parameter is obtained from the first neural network update information.
- the updated parameter can be obtained from the first neural network update information by decompression (e.g., the LZMA2 or the bzip2 algorithm).
- the first neural network update information indicates the updated parameter being a difference between the first replacement parameter and the first pretrained parameter.
- the first replacement parameter can be calculated according to a sum of the updated parameter and the first pretrained parameter.
- the first replacement parameter is determined to be the updated parameter.
- the updated parameter is a representative parameter generated (e.g., using a linear or a nonlinear transform) based on the first replacement parameter on an encoder side, and the first replacement parameter is obtained based on the representative parameter.
- the first neural network in the video decoder can be updated (or finetuned) based on the first replacement parameter, for example, by replacing the first pretrained parameter with the first replacement parameter in the first neural network.
- the video decoder includes the multiple neural networks
- the first neural network update information indicates the update information (e.g., additional replacement parameter(s)) for the multiple neural networks
- the multiple neural networks can be updated.
- the first neural network update information further includes the one or more replacement parameters for the one or more remaining neural networks in the video decoder, and the one or more remaining neural networks can be updated based on the one or more replacement parameters.
- the encoded first block in the bitstream can be decoded by the video decoder that is updated, for example, based on the updated first neural network.
- An output block generated at (S2040) can be any suitable block having any suitable size.
- the output block is a reconstructed block in a reconstructed image in the spatial domain.
- the output block of the video decoder includes residue data in the spatial domain, and thus further processing can be used to generate a reconstructed block based on the output block.
- the reconstruction module (874) is configured to combine, in the spatial domain, the residue data and prediction results (as output by the inter or intra prediction modules) to form reconstructed blocks that may be part of a reconstructed image. Additional suitable operations, such as a deblocking operation and the like, can be performed to improve the visual quality. Components in various apparatuses can be suitably combined to achieve (S2040), for example, referring to FIGs. 8 and 9, the residue data from the main decoder network (915) in the video decoder and the corresponding prediction results are fed into the reconstruction module (874) to generate the reconstructed image.
- the bitstream further includes one or more encoded bits used to determine a context model for decoding the encoded block.
- the video decoder can include a main decoder network (e.g., (911)), a context model network (e.g., (916)), an entropy parameter network (e.g., (917)), and a hyper decoder network (e.g., (925)).
- the neural network is one of the main decoder network, the context model network, the entropy parameter NN, and the hyper decoder network.
- the one or more encoded bits can be decoded using the hyper decoder network.
- An entropy model (e.g., a context model) can be determined using the context model network and the entropy parameter network based on the decoded bits and quantized latent of the encoded block that is available to the context model network.
- the encoded block can be decoded using the main decoder network and the entropy model.
- the process (2000) can be suitably adapted to various scenarios and steps in the process (2000) can be adjusted accordingly. One or more of the steps in the process (2000) can be adapted, omitted, repeated, and/or combined. Any suitable order can be used to implement the process (2000). Additional step(s) can be added.
- another block in the coded bitstream is decoded based on the updated first neural network for the first block.
- second neural network update information in the coded bitstream for a second neural network in the video decoder is decoded.
- the second neural network is configured with second pretrained parameters.
- the second neural network update information corresponds to a second block in the image to be reconstructed and indicates a second replacement parameter corresponding to a second pretrained parameter in the second pretrained parameters.
- the second neural network e.g., the context model NN (916)
- the second neural network in the video decoder can be updated based on the second replacement parameter.
- the second block can be decoded based on the updated second neural network for the second block.
- the first pretrained parameter is a one of a pretrained weight coefficient and a pretrained bias term.
- the second pretrained parameter is another one of the pretrained weight coefficient and the pretrained bias term.
- Embodiments in the disclosure may be used separately or combined in any order. Further, each of the methods (or embodiments), an encoder, and a decoder may be implemented by processing circuitry (e.g., one or more processors or one or more integrated circuits). In one example, the one or more processors execute a program that is stored in a non-transitory computer-readable medium.
- processing circuitry e.g., one or more processors or one or more integrated circuits.
- the one or more processors execute a program that is stored in a non-transitory computer-readable medium.
- Neural network(s) used in an encoder, a decoder, and/or the like can be any suitable types of neural network(s), such as a DNN, a CNN, and the like.
- the content- adaptive online training methods of this disclosure can accommodate different types of NIC frameworks, e.g., different types of encoding DNNs, decoding DNNs, encoding CNNs, decoding CNNs, and/or the like.
- FIG. 21 shows a computer system (2100) suitable for implementing certain embodiments of the disclosed subject matter.
- the computer software can be coded using any suitable machine code or computer language, that may be subject to assembly, compilation, linking, or like mechanisms to create code comprising instructions that can be executed directly, or through interpretation, micro-code execution, and the like, by one or more computer central processing units (CPUs), Graphics Processing Units (GPUs), and the like.
- CPUs computer central processing units
- GPUs Graphics Processing Units
- the instructions can be executed on various types of computers or components thereof, including, for example, personal computers, tablet computers, servers, smartphones, gaming devices, internet of things devices, and the like.
- FIG. 21 for computer system (2100) are exemplary in nature and are not intended to suggest any limitation as to the scope of use or functionality of the computer software implementing embodiments of the present disclosure. Neither should the configuration of components be interpreted as having any dependency or requirement relating to any one or combination of components illustrated in the exemplary embodiment of a computer system (2100).
- Computer system (2100) may include certain human interface input devices.
- a human interface input device may be responsive to input by one or more human users through, for example, tactile input (such as: keystrokes, swipes, data glove movements), audio input (such as: voice, clapping), visual input (such as: gestures), olfactory input (not depicted).
- the human interface devices can also be used to capture certain media not necessarily directly related to conscious input by a human, such as audio (such as: speech, music, ambient sound), images (such as: scanned images, photographic images obtain from a still image camera), video (such as two-dimensional video, three-dimensional video including stereoscopic video).
- Input human interface devices may include one or more of (only one of each depicted): keyboard (2101), mouse (2102), trackpad (2103), touch screen (2110), data-glove (not shown), joystick (2105), microphone (2106), scanner (2107), camera (2108).
- Computer system (2100) may also include certain human interface output devices.
- Such human interface output devices may be stimulating the senses of one or more human users through, for example, tactile output, sound, light, and smell/taste.
- Such human interface output devices may include tactile output devices (for example tactile feedback by the touch-screen (2110), data-glove (not shown), or joystick (2105), but there can also be tactile feedback devices that do not serve as input devices), audio output devices (such as: speakers (2109), headphones (not depicted)), visual output devices (such as screens (2110) to include CRT screens, LCD screens, plasma screens, OLED screens, each with or without touch-screen input capability, each with or without tactile feedback capability — some of which may be capable to output two dimensional visual output or more than three dimensional output through means such as stereographic output; virtual-reality glasses (not depicted), holographic displays and smoke tanks (not depicted)), and printers (not depicted).
- Computer system (2100) can also include human accessible storage devices and their associated media such as optical media including CD/DVD ROM/RW (2120) with CD/DVD or the like media (2121), thumb-drive (2122), removable hard drive or solid state drive (2123), legacy magnetic media such as tape and floppy disc (not depicted), specialized ROM/ASIC/PLD based devices such as security dongles (not depicted), and the like.
- optical media including CD/DVD ROM/RW (2120) with CD/DVD or the like media (2121), thumb-drive (2122), removable hard drive or solid state drive (2123), legacy magnetic media such as tape and floppy disc (not depicted), specialized ROM/ASIC/PLD based devices such as security dongles (not depicted), and the like.
- Computer system (2100) can also include an interface (2154) to one or more communication networks (2155).
- Networks can for example be wireless, wireline, optical.
- Networks can further be local, wide-area, metropolitan, vehicular and industrial, real-time, delay-tolerant, and so on.
- Examples of networks include local area networks such as Ethernet, wireless LANs, cellular networks to include GSM, 3G, 4G, 5G, LTE and the like, TV wireline or wireless wide area digital networks to include cable TV, satellite TV, and terrestrial broadcast TV, vehicular and industrial to include CANBus, and so forth.
- Certain networks commonly require external network interface adapters that attached to certain general purpose data ports or peripheral buses (2149) (such as, for example USB ports of the computer system (2100)); others are commonly integrated into the core of the computer system (2100) by attachment to a system bus as described below (for example Ethernet interface into a PC computer system or cellular network interface into a smartphone computer system).
- computer system (2100) can communicate with other entities.
- Such communication can be uni-directional, receive only (for example, broadcast TV), uni-directional send-only (for example CANbus to certain CANbus devices), or bi directional, for example to other computer systems using local or wide area digital networks.
- Certain protocols and protocol stacks can be used on each of those networks and network interfaces as described above.
- Aforementioned human interface devices, human-accessible storage devices, and network interfaces can be attached to a core (2140) of the computer system (2100).
- the core (2140) can include one or more Central Processing Units (CPU) (2141), Graphics Processing Units (GPU) (2142), specialized programmable processing units in the form of Field Programmable Gate Areas (FPGA) (2143), hardware accelerators for certain tasks (2144), graphics adapters (2150), and so forth.
- CPU Central Processing Unit
- GPU Graphics Processing Unit
- FPGA Field Programmable Gate Areas
- These devices, along with Read- only memory (ROM) (2145), Random-access memory (2146), internal mass storage such as internal non-user accessible hard drives, SSDs, and the like (2147), may be connected through a system bus (2148).
- the system bus (2148) can be accessible in the form of one or more physical plugs to enable extensions by additional CPUs, GPU, and the like.
- peripheral devices can be attached either directly to the core’s system bus (2148), or through a peripheral bus (2149).
- the screen (2110) can be connected to the graphics adapter (2150).
- Architectures for a peripheral bus include PCI, USB, and the like.
- CPUs (2141), GPUs (2142), FPGAs (2143), and accelerators (2144) can execute certain instructions that, in combination, can make up the aforementioned computer code. That computer code can be stored in ROM (2145) or RAM (2146). Transitional data can be also be stored in RAM (2146), whereas permanent data can be stored for example, in the internal mass storage (2147). Fast storage and retrieve to any of the memory devices can be enabled through the use of cache memory, that can be closely associated with one or more CPU (2141), GPU (2142), mass storage (2147), ROM (2145), RAM (2146), and the like.
- the computer readable media can have computer code thereon for performing various computer-implemented operations.
- the media and computer code can be those specially designed and constructed for the purposes of the present disclosure, or they can be of the kind well known and available to those having skill in the computer software arts.
- the computer system having architecture (2100), and specifically the core (2140) can provide functionality as a result of processor(s) (including CPUs, GPUs, FPGA, accelerators, and the like) executing software embodied in one or more tangible, computer-readable media.
- processor(s) including CPUs, GPUs, FPGA, accelerators, and the like
- Such computer-readable media can be media associated with user-accessible mass storage as introduced above, as well as certain storage of the core (2140) that are of non-transitory nature, such as core-internal mass storage (2147) or ROM (2145).
- the software implementing various embodiments of the present disclosure can be stored in such devices and executed by core (2140).
- a computer- readable medium can include one or more memory devices or chips, according to particular needs.
- the software can cause the core (2140) and specifically the processors therein (including CPU, GPU, FPGA, and the like) to execute particular processes or particular parts of particular processes described herein, including defining data structures stored in RAM (2146) and modifying such data structures according to the processes defined by the software.
- the computer system can provide functionality as a result of logic hardwired or otherwise embodied in a circuit (for example: accelerator (2144)), which can operate in place of or together with software to execute particular processes or particular parts of particular processes described herein.
- Reference to software can encompass logic, and vice versa, where appropriate.
- Reference to a computer-readable media can encompass a circuit (such as an integrated circuit (IC)) storing software for execution, a circuit embodying logic for execution, or both, where appropriate.
- the present disclosure encompasses any suitable combination of hardware and software.
- Appendix A Acronyms JEM: joint exploration model VVC: versatile video coding BMS: benchmark set MV: Motion Vector HEVC: High Efficiency Video Coding SEI: Supplementary Enhancement Information VUI: Video Usability Information GOPs: Groups of Pictures TUs: Transform Units,
- CPUs Central Processing Units
- GPUs Graphics Processing Units
- OLED Organic Light-Emitting Diode
- CD Compact Disc
- RAM Random Access Memory
- ASIC Application-Specific Integrated Circuit
- PLD Programmable Logic Device
- LAN Local Area Network
- GSM Global System for Mobile communications
- CANBus Controller Area Network Bus
- USB Universal Serial Bus
- PCI Peripheral Component Interconnect
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- Multimedia (AREA)
- Signal Processing (AREA)
- General Physics & Mathematics (AREA)
- Artificial Intelligence (AREA)
- Evolutionary Computation (AREA)
- Data Mining & Analysis (AREA)
- Computational Linguistics (AREA)
- Biophysics (AREA)
- Biomedical Technology (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- General Engineering & Computer Science (AREA)
- Life Sciences & Earth Sciences (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Health & Medical Sciences (AREA)
- Compression Or Coding Systems Of Tv Signals (AREA)
Abstract
Aspects of the disclosure provide a method, an apparatus, and a non-transitory computer-readable storage medium for video decoding. The apparatus can include processing circuitry. The processing circuitry is configured to decode first neural network update information in a coded bitstream for a first neural network in the video decoder. The first neural network is configured with first pretrained parameters. The first neural network update information corresponds to a first block in an image to be reconstructed and indicates a first replacement parameter corresponding to a first pretrained parameter in the first pre trained parameters. The processing circuitry is configured to update the first neural network in the video decoder based on the first replacement parameter. The processing circuitry can decode the first block based on the updated first neural network for the first block.
Description
BLOCK- WISE CONTENT- ADAPTIVE ONLINE TRAINING IN NEURAL IMAGE
COMPRESSION
INCORPORATION BY REFERENCE
[0001] This present application claims the benefit of priority to U.S. Patent Application No. 17/729,978, “BLOCK- WISE CONTENT- ADAPTIVE ONLINE TRAINING IN NEURAL IMAGE” filed on April 26, 2022, which claims the benefit of priority to U.S. Provisional Application No. 63/182,366, "BLOCK- WISE CONTENT- ADAPTIVE ONLINE TRAINING IN NEURAL IMAGE COMPRESSION" filed on April 30, 2021. The disclosures of the prior applications are hereby incorporated by reference in their entirety.
TECHNICAL FIELD
[0002] The present disclosure describes embodiments generally related to video coding.
BACKGROUND
[0003] The background description provided herein is for the purpose of generally presenting the context of the disclosure. Work of the presently named inventors, to the extent the work is described in this background section, as well as aspects of the description that may not otherwise qualify as prior art at the time of filing, are neither expressly nor impliedly admitted as prior art against the present disclosure.
[0004] Video coding and decoding can be performed using inter-picture prediction with motion compensation. Uncompressed digital image and/or video can include a series of pictures, each picture having a spatial dimension of, for example, 1920 x 1080 luminance samples and associated chrominance samples. The series of pictures can have a fixed or variable picture rate (informally also known as frame rate), of, for example 60 pictures per second or 60 Hz. Uncompressed image and/or video has specific bitrate requirements. For example, 1080p604:2:0 video at 8 bit per sample (1920x1080 luminance sample resolution at 60 Hz frame rate) requires close to 1.5 Gbit/s bandwidth. An hour of such video requires more than 600 GBytes of storage space.
[0005] One purpose of video coding and decoding can be the reduction of redundancy in the input image and/or video signal, through compression. Compression can help reduce the aforementioned bandwidth and/or storage space requirements, in some cases by two orders of magnitude or more. Although the descriptions herein use video encoding/decoding as illustrative examples, the same techniques can be applied to image encoding/decoding in similar fashion without departing from the spirit of the present disclosure. Both lossless
compression and lossy compression, as well as a combination thereof can be employed. Lossless compression refers to techniques where an exact copy of the original signal can be reconstructed from the compressed original signal. When using lossy compression, the reconstructed signal may not be identical to the original signal, but the distortion between original and reconstructed signals is small enough to make the reconstructed signal useful for the intended application. In the case of video, lossy compression is widely employed. The amount of distortion tolerated depends on the application; for example, users of certain consumer streaming applications may tolerate higher distortion than users of television distribution applications. The compression ratio achievable can reflect that: higher allowable/tolerable distortion can yield higher compression ratios.
[0006] A video encoder and decoder can utilize techniques from several broad categories, including, for example, motion compensation, transform, quantization, and entropy coding.
[0007] Video codec technologies can include techniques known as intra coding. In intra coding, sample values are represented without reference to samples or other data from previously reconstructed reference pictures. In some video codecs, the picture is spatially subdivided into blocks of samples. When all blocks of samples are coded in intra mode, that picture can be an intra picture. Intra pictures and their derivations such as independent decoder refresh pictures, can be used to reset the decoder state and can, therefore, be used as the first picture in a coded video bitstream and a video session, or as a still image. The samples of an intra block can be exposed to a transform, and the transform coefficients can be quantized before entropy coding. Intra prediction can be a technique that minimizes sample values in the pre-transform domain. In some cases, the smaller the DC value after a transform is, and the smaller the AC coefficients are, the fewer the bits that are required at a given quantization step size to represent the block after entropy coding.
[0008] Traditional intra coding such as known from, for example MPEG-2 generation coding technologies, does not use intra prediction. However, some newer video compression technologies include techniques that attempt, from, for example, surrounding sample data and/or metadata obtained during the encoding and/or decoding of spatially neighboring, and preceding in decoding order, blocks of data. Such techniques are henceforth called “intra prediction” techniques. Note that in at least some cases, intra prediction is using reference data only from the current picture under reconstruction and not from reference pictures.
[0009] There can be many different forms of intra prediction. When more than one of such techniques can be used in a given video coding technology, the technique in use can be
coded in an intra prediction mode. In certain cases, modes can have submodes and/or parameters, and those can be coded individually or included in the mode codeword. Which codeword to use for a given mode, submode, and/or parameter combination can have an impact in the coding efficiency gain through intra prediction, and so can the entropy coding technology used to translate the codewords into a bitstream.
[0010] A certain mode of intra prediction was introduced with H.264, refined in H.265, and further refined in newer coding technologies such as joint exploration model (JEM), versatile video coding (VVC), and benchmark set (BMS). A predictor block can be formed using neighboring sample values belonging to already available samples. Sample values of neighboring samples are copied into the predictor block according to a direction. A reference to the direction in use can be coded in the bitstream or may itself be predicted.
[0011] Referring to FIG. 1A, depicted in the lower right is a subset of nine predictor directions known from H.265 ’s 33 possible predictor directions (corresponding to the 33 angular modes of the 35 intra modes). The point where the arrows converge (101) represents the sample being predicted. The arrows represent the direction from which the sample is being predicted. For example, arrow (102) indicates that sample (101) is predicted from a sample or samples to the upper right, at a 45 degree angle from the horizontal. Similarly, arrow (103) indicates that sample (101) is predicted from a sample or samples to the lower left of sample (101), in a 22.5 degree angle from the horizontal.
[0012] Still referring to FIG. 1A, on the top left there is depicted a square block (104) of 4 x 4 samples (indicated by a dashed, boldface line). The square block (104) includes 16 samples, each labelled with an “S”, its position in the Y dimension (e.g., row index) and its position in the X dimension (e.g., column index). For example, sample S21 is the second sample in the Y dimension (from the top) and the first (from the left) sample in the X dimension. Similarly, sample S44 is the fourth sample in block (104) in both the Y and X dimensions. As the block is 4 x 4 samples in size, S44 is at the bottom right. Further shown are reference samples that follow a similar numbering scheme. A reference sample is labelled with an R, its Y position (e.g., row index) and X position (column index) relative to block (104). In both H.264 and H.265, prediction samples neighbor the block under reconstruction; therefore no negative values need to be used.
[0013] Intra picture prediction can work by copying reference sample values from the neighboring samples as appropriated by the signaled prediction direction. For example, assume the coded video bitstream includes signaling that, for this block, indicates a prediction direction consistent with arrow (102) — that is, samples are predicted from a
prediction sample or samples to the upper right, at a 45 degree angle from the horizontal. In that case, samples S41, S32, S23, and S14 are predicted from the same reference sample R05. Sample S44 is then predicted from reference sample R08.
[0014] In certain cases, the values of multiple reference samples may be combined, for example through interpolation, in order to calculate a reference sample; especially when the directions are not evenly divisible by 45 degrees.
[0015] The number of possible directions has increased as video coding technology has developed. In H.264 (year 2003), nine different direction could be represented. That increased to 33 in H.265 (year 2013), and JEM/VVC/BMS, at the time of disclosure, can support up to 65 directions. Experiments have been conducted to identify the most likely directions, and certain techniques in the entropy coding are used to represent those likely directions in a small number of bits, accepting a certain penalty for less likely directions. Further, the directions themselves can sometimes be predicted from neighboring directions used in neighboring, already decoded, blocks.
[0016] FIG. IB shows a schematic (110) that depicts 65 intra prediction directions according to JEM to illustrate the increasing number of prediction directions over time.
[0017] The mapping of intra prediction directions bits in the coded video bitstream that represent the direction can be different from video coding technology to video coding technology; and can range, for example, from simple direct mappings of prediction direction to intra prediction mode, to codewords, to complex adaptive schemes involving most probable modes, and similar techniques. In all cases, however, there can be certain directions that are statistically less likely to occur in video content than certain other directions. As the goal of video compression is the reduction of redundancy, those less likely directions will, in a well working video coding technology, be represented by a larger number of bits than more likely directions.
[0018] Motion compensation can be a lossy compression technique and can relate to techniques where a block of sample data from a previously reconstructed picture or part thereof (reference picture), after being spatially shifted in a direction indicated by a motion vector (MV henceforth), is used for the prediction of a newly reconstructed picture or picture part. In some cases, the reference picture can be the same as the picture currently under reconstruction. MVs can have two dimensions X and Y, or three dimensions, the third being an indication of the reference picture in use (the latter, indirectly, can be a time dimension).
[0019] In some video compression techniques, an MV applicable to a certain area of sample data can be predicted from other MVs, for example from those related to another area
of sample data spatially adjacent to the area under reconstruction, and preceding that MV in decoding order. Doing so can substantially reduce the amount of data required for coding the MV, thereby removing redundancy and increasing compression. MV prediction can work effectively, for example, because when coding an input video signal derived from a camera (known as natural video) there is a statistical likelihood that areas larger than the area to which a single MV is applicable move in a similar direction and, therefore, can in some cases be predicted using a similar motion vector derived from MVs of neighboring area. That results in the MV found for a given area to be similar or the same as the MV predicted from the surrounding MVs, and that in turn can be represented, after entropy coding, in a smaller number of bits than what would be used if coding the MV directly. In some cases, MV prediction can be an example of lossless compression of a signal (namely: the MVs) derived from the original signal (namely: the sample stream). In other cases, MV prediction itself can be lossy, for example because of rounding errors when calculating a predictor from several surrounding MVs.
[0020] Various MV prediction mechanisms are described in H.265/HEVC (ITU-T Rec. H.265, "High Efficiency Video Coding", December 2016). Out of the many MV prediction mechanisms that H.265 offers, described here is a technique henceforth referred to as "spatial merge".
[0021] Referring to FIG. 2, a current block (201) comprises samples that have been found by the encoder during the motion search process to be predictable from a previous block of the same size that has been spatially shifted. Instead of coding that MV directly, the MV can be derived from metadata associated with one or more reference pictures, for example from the most recent (in decoding order) reference picture, using the MV associated with either one of five surrounding samples, denoted A0, Al, and B0, Bl, B2 (202 through 206, respectively). In H.265, the MV prediction can use predictors from the same reference picture that the neighboring block is using.
SUMMARY
[0022] Aspects of the disclosure provide methods and apparatuses for video encoding and decoding. In some examples, an apparatus for video decoding includes processing circuitry. The processing circuitry is configured to decode first neural network update information in a coded bitstream for a first neural network in a video decoder. The first neural network is configured with first pretrained parameters. The first neural network update information corresponds to a first block in an image to be reconstructed and indicates a first replacement parameter corresponding to a first pretrained parameter in the first pretrained
parameters. The processing circuitry can update the first neural network in the video decoder based on the first replacement parameter and decode the first block based on the updated first neural network for the first block.
[0023] In an embodiment, the first neural network update information further indicates one or more replacement parameters for one or more remaining neural networks in the video decoder. The processing circuitry can update the one or more remaining neural networks based on the one or more replacement parameters.
[0024] In an embodiment, the processing circuitry decodes second neural network update information in the coded bitstream for a second neural network in the video decoder. The second neural network is configured with second pretrained parameters. The second neural network update information corresponds to a second block in the image to be reconstructed and indicates a second replacement parameter corresponding to a second pretrained parameter in the second pretrained parameters. In an example, the second neural network is different from the first neural network. The processing circuitry can update the second neural network in the video decoder based on the second replacement parameter and decode the second block based on the updated second neural network for the second block.
[0025] In an embodiment, the first pretrained parameter is a one of a pretrained weight coefficient and a pretrained bias term.
[0026] In an embodiment, the second pretrained parameter is another one of the pretrained weight coefficient and the pretrained bias term.
[0027] In an embodiment, the processing circuitry decodes a second block in the coded bitstream based on the updated first neural network for the first block.
[0028] In an embodiment, the first neural network update information indicates a difference between the first replacement parameter and the first pretrained parameter. The processing circuitry determines the first replacement parameter according to a sum of the difference and the first pretrained parameter.
[0029] In an embodiment, the processing circuitry decodes the first neural network update information based on one of a variation of a Lempel-Ziv-Markov chain algorithm (LZMA2) and a bzip2 algorithm.
[0030] In an example, the processing circuitry decodes the second neural network update information based on another one of the LZMA2 and the bzip2 algorithm.
[0031] Aspects of the disclosure also provide a non-transitory computer-readable storage medium storing a program executable by at least one processor to perform the methods for video decoding.
BRIEF DESCRIPTION OF THE DRAWINGS
[0032] Further features, the nature, and various advantages of the disclosed subject matter will be more apparent from the following detailed description and the accompanying drawings in which:
[0033] FIG. 1A is a schematic illustration of an exemplary subset of intra prediction modes.
[0034] FIG. IB is an illustration of exemplary intra prediction directions.
[0035] FIG. 2 shows a current block (201) and surrounding samples in accordance with an embodiment.
[0036] FIG. 3 is a schematic illustration of a simplified block diagram of a communication system (300) in accordance with an embodiment.
[0037] FIG. 4 is a schematic illustration of a simplified block diagram of a communication system (400) in accordance with an embodiment.
[0038] FIG. 5 is a schematic illustration of a simplified block diagram of a decoder in accordance with an embodiment.
[0039] FIG. 6 is a schematic illustration of a simplified block diagram of an encoder in accordance with an embodiment.
[0040] FIG. 7 shows a block diagram of an encoder in accordance with another embodiment.
[0041] FIG. 8 shows a block diagram of a decoder in accordance with another embodiment.
[0042] FIG. 9A shows an example of a block-wise image coding according to an embodiment of the disclosure.
[0043] FIG. 9B shows an exemplary NIC framework according to an embodiment of the disclosure.
[0044] FIG. 10 shows an exemplary convolution neural network (CNN) of a main encoder network according to an embodiment of the disclosure.
[0045] FIG. 11 shows an exemplary CNN of a main decoder network according to an embodiment of the disclosure.
[0046] FIG. 12 shows an exemplary CNN of a hyper encoder according to an embodiment of the disclosure.
[0047] FIG. 13 shows an exemplary CNN of a hyper decoder according to an embodiment of the disclosure.
[0048] FIG. 14 shows an exemplary CNN of a context model network according to an embodiment of the disclosure.
[0049] FIG. 15 shows an exemplary CNN of an entropy parameter network according to an embodiment of the disclosure.
[0050] FIG. 16A shows an exemplary video encoder according to an embodiment of the disclosure.
[0051] FIG. 16B shows an exemplary video decoder according to an embodiment of the disclosure.
[0052] FIG. 17 shows an exemplary video encoder according to an embodiment of the disclosure.
[0053] FIG. 18 shows an exemplary video decoder according to an embodiment of the disclosure.
[0054] FIG. 19 shows a flow chart outlining a process according to an embodiment of the disclosure.
[0055] FIG. 20 shows a flow chart outlining a process according to an embodiment of the disclosure.
[0056] FIG. 21 is a schematic illustration of a computer system in accordance with an embodiment.
DETAILED DESCRIPTION OF EMBODIMENTS
[0057] FIG. 3 illustrates a simplified block diagram of a communication system (300) according to an embodiment of the present disclosure. The communication system (300) includes a plurality of terminal devices that can communicate with each other, via, for example, a network (350). For example, the communication system (300) includes a first pair of terminal devices (310) and (320) interconnected via the network (350). In the FIG. 3 example, the first pair of terminal devices (310) and (320) performs unidirectional transmission of data. For example, the terminal device (310) may code video data (e.g., a stream of video pictures that are captured by the terminal device (310)) for transmission to the other terminal device (320) via the network (350). The encoded video data can be transmitted in the form of one or more coded video bitstreams. The terminal device (320) may receive the coded video data from the network (350), decode the coded video data to recover the video pictures and display video pictures according to the recovered video data. Unidirectional data transmission may be common in media serving applications and the like.
[0058] In another example, the communication system (300) includes a second pair of terminal devices (330) and (340) that performs bidirectional transmission of coded video data
that may occur, for example, during videoconferencing. For bidirectional transmission of data, in an example, each terminal device of the terminal devices (330) and (340) may code video data (e.g., a stream of video pictures that are captured by the terminal device) for transmission to the other terminal device of the terminal devices (330) and (340) via the network (350). Each terminal device of the terminal devices (330) and (340) also may receive the coded video data transmitted by the other terminal device of the terminal devices (330) and (340), and may decode the coded video data to recover the video pictures and may display video pictures at an accessible display device according to the recovered video data.
[0059] In the FIG. 3 example, the terminal devices (310), (320), (330) and (340) may be illustrated as servers, personal computers and smart phones but the principles of the present disclosure may be not so limited. Embodiments of the present disclosure find application with laptop computers, tablet computers, media players and/or dedicated video conferencing equipment. The network (350) represents any number of networks that convey coded video data among the terminal devices (310), (320), (330) and (340), including for example wireline (wired) and/or wireless communication networks. The communication network (350) may exchange data in circuit-switched and/or packet-switched channels. Representative networks include telecommunications networks, local area networks, wide area networks and/or the Internet. For the purposes of the present discussion, the architecture and topology of the network (350) may be immaterial to the operation of the present disclosure unless explained herein below.
[0060] FIG. 4 illustrates, as an example for an application for the disclosed subject matter, the placement of a video encoder and a video decoder in a streaming environment. The disclosed subject matter can be equally applicable to other video enabled applications, including, for example, video conferencing, digital TV, storing of compressed video on digital media including CD, DVD, memory stick and the like, and so on.
[0061] A streaming system may include a capture subsystem (413), that can include a video source (401), for example a digital camera, creating for example a stream of video pictures (402) that are uncompressed. In an example, the stream of video pictures (402) includes samples that are taken by the digital camera. The stream of video pictures (402), depicted as a bold line to emphasize a high data volume when compared to encoded video data (404) (or coded video bitstreams), can be processed by an electronic device (420) that includes a video encoder (403) coupled to the video source (401). The video encoder (403) can include hardware, software, or a combination thereof to enable or implement aspects of the disclosed subject matter as described in more detail below. The encoded video data (404)
(or encoded video bitstream (404)), depicted as a thin line to emphasize the lower data volume when compared to the stream of video pictures (402), can be stored on a streaming server (405) for future use. One or more streaming client subsystems, such as client subsystems (406) and (408) in FIG. 4 can access the streaming server (405) to retrieve copies (407) and (409) of the encoded video data (404). A client subsystem (406) can include a video decoder (410), for example, in an electronic device (430). The video decoder (410) decodes the incoming copy (407) of the encoded video data and creates an outgoing stream of video pictures (411) that can be rendered on a display (412) (e.g., display screen) or other rendering device (not depicted). In some streaming systems, the encoded video data (404), (407), and (409) (e.g., video bitstreams) can be encoded according to certain video coding/compression standards. Examples of those standards include ITU-T Recommendation H.265. In an example, a video coding standard under development is informally known as Versatile Video Coding (VVC). The disclosed subject matter may be used in the context of VVC.
[0062] It is noted that the electronic devices (420) and (430) can include other components (not shown). For example, the electronic device (420) can include a video decoder (not shown) and the electronic device (430) can include a video encoder (not shown) as well.
[0063] FIG. 5 shows a block diagram of a video decoder (510) according to an embodiment of the present disclosure. The video decoder (510) can be included in an electronic device (530). The electronic device (530) can include a receiver (531) (e.g., receiving circuitry). The video decoder (510) can be used in the place of the video decoder (410) in the FIG. 4 example.
[0064] The receiver (531) may receive one or more coded video sequences to be decoded by the video decoder (510); in the same or another embodiment, one coded video sequence at a time, where the decoding of each coded video sequence is independent from other coded video sequences. The coded video sequence may be received from a channel (501), which may be a hardware/software link to a storage device which stores the encoded video data. The receiver (531) may receive the encoded video data with other data, for example, coded audio data and/or ancillary data streams, that may be forwarded to their respective using entities (not depicted). The receiver (531) may separate the coded video sequence from the other data. To combat network jitter, a buffer memory (515) may be coupled in between the receiver (531) and an entropy decoder / parser (520) ("parser (520)" henceforth). In certain applications, the buffer memory (515) is part of the video decoder
(510). In others, it can be outside of the video decoder (510) (not depicted). In still others, there can be a buffer memory (not depicted) outside of the video decoder (510), for example to combat network jitter, and in addition another buffer memory (515) inside the video decoder (510), for example to handle playout timing. When the receiver (531) is receiving data from a store/forward device of sufficient bandwidth and controllability, or from an Asynchronous network, the buffer memory (515) may not be needed, or can be small. For use on best effort packet networks such as the Internet, the buffer memory (515) may be required, can be comparatively large and can be advantageously of adaptive size, and may at least partially be implemented in an operating system or similar elements (not depicted) outside of the video decoder (510).
[0065] The video decoder (510) may include the parser (520) to reconstruct symbols (521) from the coded video sequence. Categories of those symbols include information used to manage operation of the video decoder (510), and potentially information to control a rendering device such as a render device (512) (e.g., a display screen) that is not an integral part of the electronic device (530) but can be coupled to the electronic device (530), as was shown in FIG. 5. The control information for the rendering device(s) may be in the form of Supplemental Enhancement Information (SEI messages) or Video Usability Information (VUI) parameter set fragments (not depicted). The parser (520) may parse / entropy-decode the coded video sequence that is received. The coding of the coded video sequence can be in accordance with a video coding technology or standard, and can follow various principles, including variable length coding, Huffman coding, arithmetic coding with or without context sensitivity, and so forth. The parser (520) may extract from the coded video sequence, a set of subgroup parameters for at least one of the subgroups of pixels in the video decoder, based upon at least one parameter corresponding to the group. Subgroups can include Groups of Pictures (GOPs), pictures, tiles, slices, macroblocks, Coding Units (CUs), blocks, Transform Units (TUs), Prediction Units (PUs) and so forth. The parser (520) may also extract from the coded video sequence information such as transform coefficients, quantizer parameter values, motion vectors, and so forth.
[0066] The parser (520) may perform an entropy decoding / parsing operation on the video sequence received from the buffer memory (515), so as to create symbols (521).
[0067] Reconstruction of the symbols (521) can involve multiple different units depending on the type of the coded video picture or parts thereof (such as: inter and intra picture, inter and intra block), and other factors. Which units are involved, and how, can be controlled by the subgroup control information that was parsed from the coded video
sequence by the parser (520). The flow of such subgroup control information between the parser (520) and the multiple units below is not depicted for clarity.
[0068] Beyond the functional blocks already mentioned, the video decoder (510) can be conceptually subdivided into a number of functional units as described below. In a practical implementation operating under commercial constraints, many of these units interact closely with each other and can, at least partly, be integrated into each other. However, for the purpose of describing the disclosed subject matter, the conceptual subdivision into the functional units below is appropriate.
[0069] A first unit is the scaler / inverse transform unit (551). The scaler / inverse transform unit (551) receives a quantized transform coefficient as well as control information, including which transform to use, block size, quantization factor, quantization scaling matrices, etc. as symbol(s) (521) from the parser (520). The scaler / inverse transform unit (551) can output blocks comprising sample values, that can be input into aggregator (555).
[0070] In some cases, the output samples of the scaler / inverse transform (551) can pertain to an intra coded block; that is: a block that is not using predictive information from previously reconstructed pictures, but can use predictive information from previously reconstructed parts of the current picture. Such predictive information can be provided by an intra picture prediction unit (552). In some cases, the intra picture prediction unit (552) generates a block of the same size and shape of the block under reconstruction, using surrounding already reconstructed information fetched from the current picture buffer (558). The current picture buffer (558) buffers, for example, partly reconstructed current picture and/or fully reconstructed current picture. The aggregator (555), in some cases, adds, on a per sample basis, the prediction information the intra prediction unit (552) has generated to the output sample information as provided by the scaler / inverse transform unit (551).
[0071] In other cases, the output samples of the scaler / inverse transform unit (551) can pertain to an inter coded, and potentially motion compensated block. In such a case, a motion compensation prediction unit (553) can access reference picture memory (557) to fetch samples used for prediction. After motion compensating the fetched samples in accordance with the symbols (521) pertaining to the block, these samples can be added by the aggregator (555) to the output of the scaler / inverse transform unit (551) (in this case called the residual samples or residual signal) so as to generate output sample information. The addresses within the reference picture memory (557) from where the motion compensation prediction unit (553) fetches prediction samples can be controlled by motion vectors, available to the motion compensation prediction unit (553) in the form of symbols (521) that
can have, for example X, Y, and reference picture components. Motion compensation also can include interpolation of sample values as fetched from the reference picture memory (557) when sub-sample exact motion vectors are in use, motion vector prediction mechanisms, and so forth.
[0072] The output samples of the aggregator (555) can be subject to various loop filtering techniques in the loop filter unit (556). Video compression technologies can include in-loop filter technologies that are controlled by parameters included in the coded video sequence (also referred to as coded video bitstream) and made available to the loop filter unit (556) as symbols (521) from the parser (520), but can also be responsive to meta-information obtained during the decoding of previous (in decoding order) parts of the coded picture or coded video sequence, as well as responsive to previously reconstructed and loop-filtered sample values.
[0073] The output of the loop filter unit (556) can be a sample stream that can be output to the render device (512) as well as stored in the reference picture memory (557) for use in future inter-picture prediction.
[0074] Certain coded pictures, once fully reconstructed, can be used as reference pictures for future prediction. For example, once a coded picture corresponding to a current picture is fully reconstructed and the coded picture has been identified as a reference picture (by, for example, the parser (520)), the current picture buffer (558) can become a part of the reference picture memory (557), and a fresh current picture buffer can be reallocated before commencing the reconstruction of the following coded picture.
[0075] The video decoder (510) may perform decoding operations according to a predetermined video compression technology in a standard, such as ITU-T Rec. H.265. The coded video sequence may conform to a syntax specified by the video compression technology or standard being used, in the sense that the coded video sequence adheres to both the syntax of the video compression technology or standard and the profiles as documented in the video compression technology or standard. Specifically, a profile can select certain tools as the only tools available for use under that profile from all the tools available in the video compression technology or standard. Also necessary for compliance can be that the complexity of the coded video sequence is within bounds as defined by the level of the video compression technology or standard. In some cases, levels restrict the maximum picture size, maximum frame rate, maximum reconstruction sample rate (measured in, for example megasamples per second), maximum reference picture size, and so on. Limits set by levels can, in some cases, be further restricted through Hypothetical Reference Decoder (HRD)
specifications and metadata for HRD buffer management signaled in the coded video sequence.
[0076] In an embodiment, the receiver (531) may receive additional (redundant) data with the encoded video. The additional data may be included as part of the coded video sequence(s). The additional data may be used by the video decoder (510) to properly decode the data and/or to more accurately reconstruct the original video data. Additional data can be in the form of, for example, temporal, spatial, or signal noise ratio (SNR) enhancement layers, redundant slices, redundant pictures, forward error correction codes, and so on.
[0077] FIG. 6 shows a block diagram of a video encoder (603) according to an embodiment of the present disclosure. The video encoder (603) is included in an electronic device (620). The electronic device (620) includes a transmitter (640) (e.g., transmitting circuitry). The video encoder (603) can be used in the place of the video encoder (403) in the FIG. 4 example.
[0078] The video encoder (603) may receive video samples from a video source (601) (that is not part of the electronic device (620) in the FIG. 6 example) that may capture video image(s) to be coded by the video encoder (603). In another example, the video source (601) is a part of the electronic device (620).
[0079] The video source (601) may provide the source video sequence to be coded by the video encoder (603) in the form of a digital video sample stream that can be of any suitable bit depth (for example: 8 bit, 10 bit, 12 bit, ...), any colorspace (for example, BT.601 Y CrCB, RGB, ...), and any suitable sampling structure (for example Y CrCb 4:2:0, Y CrCb 4:4:4). In a media serving system, the video source (601) may be a storage device storing previously prepared video. In a videoconferencing system, the video source (601) may be a camera that captures local image information as a video sequence. Video data may be provided as a plurality of individual pictures that impart motion when viewed in sequence. The pictures themselves may be organized as a spatial array of pixels, wherein each pixel can comprise one or more samples depending on the sampling structure, color space, etc. in use.
A person skilled in the art can readily understand the relationship between pixels and samples. The description below focuses on samples.
[0080] According to an embodiment, the video encoder (603) may code and compress the pictures of the source video sequence into a coded video sequence (643) in real time or under any other time constraints as required by the application. Enforcing appropriate coding speed is one function of a controller (650). In some embodiments, the controller (650) controls other functional units as described below and is functionally coupled to the other
functional units. The coupling is not depicted for clarity. Parameters set by the controller (650) can include rate control related parameters (picture skip, quantizer, lambda value of rate-distortion optimization techniques, ...), picture size, group of pictures (GOP) layout, maximum motion vector search range, and so forth. The controller (650) can be configured to have other suitable functions that pertain to the video encoder (603) optimized for a certain system design.
[0081] In some embodiments, the video encoder (603) is configured to operate in a coding loop. As an oversimplified description, in an example, the coding loop can include a source coder (630) (e.g., responsible for creating symbols, such as a symbol stream, based on an input picture to be coded, and a reference picture(s)), and a (local) decoder (633) embedded in the video encoder (603). The decoder (633) reconstructs the symbols to create the sample data in a similar manner as a (remote) decoder also would create (as any compression between symbols and coded video bitstream is lossless in the video compression technologies considered in the disclosed subject matter). The reconstructed sample stream (sample data) is input to the reference picture memory (634). As the decoding of a symbol stream leads to bit-exact results independent of decoder location (local or remote), the content in the reference picture memory (634) is also bit exact between the local encoder and remote encoder. In other words, the prediction part of an encoder "sees" as reference picture samples exactly the same sample values as a decoder would "see" when using prediction during decoding. This fundamental principle of reference picture synchronicity (and resulting drift, if synchronicity cannot be maintained, for example because of channel errors) is used in some related arts as well.
[0082] The operation of the "local" decoder (633) can be the same as of a "remote" decoder, such as the video decoder (510), which has already been described in detail above in conjunction with FIG. 5. Briefly referring also to FIG. 5, however, as symbols are available and encoding/decoding of symbols to a coded video sequence by an entropy coder (645) and the parser (520) can be lossless, the entropy decoding parts of the video decoder (510), including the buffer memory (515), and parser (520) may not be fully implemented in the local decoder (633).
[0083] In an embodiment, a decoder technology except the parsing/entropy decoding that is present in a decoder is present, in an identical or a substantially identical functional form, in a corresponding encoder. Accordingly, the disclosed subject matter focuses on decoder operation. The description of encoder technologies can be abbreviated as they are
the inverse of the comprehensively described decoder technologies. In certain areas a more detail description is provided below.
[0084] During operation, in some examples, the source coder (630) may perform motion compensated predictive coding, which codes an input picture predictively with reference to one or more previously coded picture from the video sequence that were designated as "reference pictures.” In this manner, the coding engine (632) codes differences between pixel blocks of an input picture and pixel blocks of reference picture(s) that may be selected as prediction reference(s) to the input picture.
[0085] The local video decoder (633) may decode coded video data of pictures that may be designated as reference pictures, based on symbols created by the source coder (630). Operations of the coding engine (632) may advantageously be lossy processes. When the coded video data may be decoded at a video decoder (not shown in FIG. 6), the reconstructed video sequence typically may be a replica of the source video sequence with some errors.
The local video decoder (633) replicates decoding processes that may be performed by the video decoder on reference pictures and may cause reconstructed reference pictures to be stored in the reference picture cache (634). In this manner, the video encoder (603) may store copies of reconstructed reference pictures locally that have common content as the reconstructed reference pictures that will be obtained by a far-end video decoder (absent transmission errors).
[0086] The predictor (635) may perform prediction searches for the coding engine (632). That is, for a new picture to be coded, the predictor (635) may search the reference picture memory (634) for sample data (as candidate reference pixel blocks) or certain metadata such as reference picture motion vectors, block shapes, and so on, that may serve as an appropriate prediction reference for the new pictures. The predictor (635) may operate on a sample block-by-pixel block basis to find appropriate prediction references. In some cases, as determined by search results obtained by the predictor (635), an input picture may have prediction references drawn from multiple reference pictures stored in the reference picture memory (634).
[0087] The controller (650) may manage coding operations of the source coder (630), including, for example, setting of parameters and subgroup parameters used for encoding the video data.
[0088] Output of all aforementioned functional units may be subjected to entropy coding in the entropy coder (645). The entropy coder (645) translates the symbols as generated by the various functional units into a coded video sequence, by lossless
compressing the symbols according to technologies such as Huffman coding, variable length coding, arithmetic coding, and so forth.
[0089] The transmitter (640) may buffer the coded video sequence(s) as created by the entropy coder (645) to prepare for transmission via a communication channel (660), which may be a hardware/software link to a storage device which would store the encoded video data. The transmitter (640) may merge coded video data from the video coder (603) with other data to be transmitted, for example, coded audio data and/or ancillary data streams (sources not shown).
[0090] The controller (650) may manage operation of the video encoder (603).
During coding, the controller (650) may assign to each coded picture a certain coded picture type, which may affect the coding techniques that may be applied to the respective picture. For example, pictures often may be assigned as one of the following picture types:
[0091] An Intra Picture (I picture) may be one that may be coded and decoded without using any other picture in the sequence as a source of prediction. Some video codecs allow for different types of intra pictures, including, for example Independent Decoder Refresh (“IDR”) Pictures. A person skilled in the art is aware of those variants of I pictures and their respective applications and features.
[0092] A predictive picture (P picture) may be one that may be coded and decoded using intra prediction or inter prediction using at most one motion vector and reference index to predict the sample values of each block.
[0093] A bi-directionally predictive picture (B Picture) may be one that may be coded and decoded using intra prediction or inter prediction using at most two motion vectors and reference indices to predict the sample values of each block. Similarly, multiple-predictive pictures can use more than two reference pictures and associated metadata for the reconstruction of a single block.
[0094] Source pictures commonly may be subdivided spatially into a plurality of sample blocks (for example, blocks of 4x4, 8x8, 4x8, or 16x16 samples each) and coded on a block-by-block basis. Blocks may be coded predictively with reference to other (already coded) blocks as determined by the coding assignment applied to the blocks' respective pictures. For example, blocks of I pictures may be coded non-predictively or they may be coded predictively with reference to already coded blocks of the same picture (spatial prediction or intra prediction). Pixel blocks of P pictures may be coded predictively, via spatial prediction or via temporal prediction with reference to one previously coded reference
picture. Blocks of B pictures may be coded predictively, via spatial prediction or via temporal prediction with reference to one or two previously coded reference pictures.
[0095] The video encoder (603) may perform coding operations according to a predetermined video coding technology or standard, such as ITU-T Rec. H.265. In its operation, the video encoder (603) may perform various compression operations, including predictive coding operations that exploit temporal and spatial redundancies in the input video sequence. The coded video data, therefore, may conform to a syntax specified by the video coding technology or standard being used.
[0096] In an embodiment, the transmitter (640) may transmit additional data with the encoded video. The source coder (630) may include such data as part of the coded video sequence. Additional data may comprise temporal/spatial/SNR enhancement layers, other forms of redundant data such as redundant pictures and slices, SEI messages, VUI parameter set fragments, and so on.
[0097] A video may be captured as a plurality of source pictures (video pictures) in a temporal sequence. Intra-picture prediction (often abbreviated to intra prediction) makes use of spatial correlation in a given picture, and inter-picture prediction makes uses of the (temporal or other) correlation between the pictures. In an example, a specific picture under encoding/decoding, which is referred to as a current picture, is partitioned into blocks. When a block in the current picture is similar to a reference block in a previously coded and still buffered reference picture in the video, the block in the current picture can be coded by a vector that is referred to as a motion vector. The motion vector points to the reference block in the reference picture, and can have a third dimension identifying the reference picture, in case multiple reference pictures are in use.
[0098] In some embodiments, a bi-prediction technique can be used in the inter picture prediction. According to the bi-prediction technique, two reference pictures, such as a first reference picture and a second reference picture that are both prior in decoding order to the current picture in the video (but may be in the past and future, respectively, in display order) are used. A block in the current picture can be coded by a first motion vector that points to a first reference block in the first reference picture, and a second motion vector that points to a second reference block in the second reference picture. The block can be predicted by a combination of the first reference block and the second reference block.
[0099] Further, a merge mode technique can be used in the inter-picture prediction to improve coding efficiency.
[0100] According to some embodiments of the disclosure, predictions, such as inter picture predictions and intra-picture predictions are performed in the unit of blocks. For example, according to the HEVC standard, a picture in a sequence of video pictures is partitioned into coding tree units (CTU) for compression, the CTUs in a picture have the same size, such as 64x64 pixels, 32x32 pixels, or 16x16 pixels. In general, a CTU includes three coding tree blocks (CTBs), which are one luma CTB and two chroma CTBs. Each CTU can be recursively quadtree split into one or multiple coding units (CUs). For example, a CTU of 64x64 pixels can be split into one CU of 64x64 pixels, or 4 CUs of 32x32 pixels, or 16 CUs of 16x16 pixels. In an example, each CU is analyzed to determine a prediction type for the CU, such as an inter prediction type or an intra prediction type. The CU is split into one or more prediction units (PUs) depending on the temporal and/or spatial predictability. Generally, each PU includes a luma prediction block (PB), and two chroma PBs. In an embodiment, a prediction operation in coding (encoding/decoding) is performed in the unit of a prediction block. Using a luma prediction block as an example of a prediction block, the prediction block includes a matrix of values (e.g., luma values) for pixels, such as 8x8 pixels, 16x16 pixels, 8x16 pixels, 16x8 pixels, and the like.
[0101] FIG. 7 shows a diagram of a video encoder (703) according to another embodiment of the disclosure. The video encoder (703) is configured to receive a processing block (e.g., a prediction block) of sample values within a current video picture in a sequence of video pictures, and encode the processing block into a coded picture that is part of a coded video sequence. In an example, the video encoder (703) is used in the place of the video encoder (403) in the FIG. 4 example.
[0102] In an HEVC example, the video encoder (703) receives a matrix of sample values for a processing block, such as a prediction block of 8x8 samples, and the like. The video encoder (703) determines whether the processing block is best coded using intra mode, inter mode, or bi-prediction mode using, for example, rate-distortion optimization. When the processing block is to be coded in intra mode, the video encoder (703) may use an intra prediction technique to encode the processing block into the coded picture; and when the processing block is to be coded in inter mode or bi-prediction mode, the video encoder (703) may use an inter prediction or bi-prediction technique, respectively, to encode the processing block into the coded picture. In certain video coding technologies, merge mode can be an inter picture prediction submode where the motion vector is derived from one or more motion vector predictors without the benefit of a coded motion vector component outside the predictors. In certain other video coding technologies, a motion vector component applicable
to the subject block may be present. In an example, the video encoder (703) includes other components, such as a mode decision module (not shown) to determine the mode of the processing blocks.
[0103] In the FIG. 7 example, the video encoder (703) includes the inter encoder (730), an intra encoder (722), a residue calculator (723), a switch (726), a residue encoder (724), a general controller (721), and an entropy encoder (725) coupled together as shown in FIG. 7.
[0104] The inter encoder (730) is configured to receive the samples of the current block (e.g., a processing block), compare the block to one or more reference blocks in reference pictures (e.g., blocks in previous pictures and later pictures), generate inter prediction information (e.g., description of redundant information according to inter encoding technique, motion vectors, merge mode information), and calculate inter prediction results (e.g., predicted block) based on the inter prediction information using any suitable technique. In some examples, the reference pictures are decoded reference pictures that are decoded based on the encoded video information.
[0105] The intra encoder (722) is configured to receive the samples of the current block (e.g., a processing block), in some cases compare the block to blocks already coded in the same picture, generate quantized coefficients after transform, and in some cases also intra prediction information (e.g., an intra prediction direction information according to one or more intra encoding techniques). In an example, the intra encoder (722) also calculates intra prediction results (e.g., predicted block) based on the intra prediction information and reference blocks in the same picture.
[0106] The general controller (721) is configured to determine general control data and control other components of the video encoder (703) based on the general control data.
In an example, the general controller (721) determines the mode of the block, and provides a control signal to the switch (726) based on the mode. For example, when the mode is the intra mode, the general controller (721) controls the switch (726) to select the intra mode result for use by the residue calculator (723), and controls the entropy encoder (725) to select the intra prediction information and include the intra prediction information in the bitstream; and when the mode is the inter mode, the general controller (721) controls the switch (726) to select the inter prediction result for use by the residue calculator (723), and controls the entropy encoder (725) to select the inter prediction information and include the inter prediction information in the bitstream.
[0107] The residue calculator (723) is configured to calculate a difference (residue data) between the received block and prediction results selected from the intra encoder (722) or the inter encoder (730). The residue encoder (724) is configured to operate based on the residue data to encode the residue data to generate the transform coefficients. In an example, the residue encoder (724) is configured to convert the residue data from a spatial domain to a frequency domain, and generate the transform coefficients. The transform coefficients are then subject to quantization processing to obtain quantized transform coefficients. In various embodiments, the video encoder (703) also includes a residue decoder (728). The residue decoder (728) is configured to perform inverse-transform, and generate the decoded residue data. The decoded residue data can be suitably used by the intra encoder (722) and the inter encoder (730). For example, the inter encoder (730) can generate decoded blocks based on the decoded residue data and inter prediction information, and the intra encoder (722) can generate decoded blocks based on the decoded residue data and the intra prediction information. The decoded blocks are suitably processed to generate decoded pictures and the decoded pictures can be buffered in a memory circuit (not shown) and used as reference pictures in some examples.
[0108] The entropy encoder (725) is configured to format the bitstream to include the encoded block. The entropy encoder (725) is configured to include various information according to a suitable standard, such as the HEVC standard. In an example, the entropy encoder (725) is configured to include the general control data, the selected prediction information (e.g., intra prediction information or inter prediction information), the residue information, and other suitable information in the bitstream. Note that, according to the disclosed subject matter, when coding a block in the merge submode of either inter mode or bi-prediction mode, there is no residue information.
[0109] FIG. 8 shows a diagram of a video decoder (810) according to another embodiment of the disclosure. The video decoder (810) is configured to receive coded pictures that are part of a coded video sequence, and decode the coded pictures to generate reconstructed pictures. In an example, the video decoder (810) is used in the place of the video decoder (410) in the FIG. 4 example.
[0110] In the FIG. 8 example, the video decoder (810) includes an entropy decoder (871), an inter decoder (880), a residue decoder (873), a reconstruction module (874), and an intra decoder (872) coupled together as shown in FIG. 8.
[0111] The entropy decoder (871) can be configured to reconstruct, from the coded picture, certain symbols that represent the syntax elements of which the coded picture is
made up. Such symbols can include, for example, the mode in which a block is coded (such as, for example, intra mode, inter mode, bi-predicted mode, the latter two in merge submode or another submode), prediction information (such as, for example, intra prediction information or inter prediction information) that can identify certain sample or metadata that is used for prediction by the intra decoder (872) or the inter decoder (880), respectively, residual information in the form of, for example, quantized transform coefficients, and the like. In an example, when the prediction mode is inter or bi-predicted mode, the inter prediction information is provided to the inter decoder (880); and when the prediction type is the intra prediction type, the intra prediction information is provided to the intra decoder (872). The residual information can be subject to inverse quantization and is provided to the residue decoder (873).
[0112] The inter decoder (880) is configured to receive the inter prediction information, and generate inter prediction results based on the inter prediction information.
[0113] The intra decoder (872) is configured to receive the intra prediction information, and generate prediction results based on the intra prediction information.
[0114] The residue decoder (873) is configured to perform inverse quantization to extract de-quantized transform coefficients, and process the de-quantized transform coefficients to convert the residual from the frequency domain to the spatial domain. The residue decoder (873) may also require certain control information (to include the Quantizer Parameter (QP)), and that information may be provided by the entropy decoder (871) (data path not depicted as this may be low volume control information only).
[0115] The reconstruction module (874) is configured to combine, in the spatial domain, the residual as output by the residue decoder (873) and the prediction results (as output by the inter or intra prediction modules as the case may be) to form a reconstructed block, that may be part of the reconstructed picture, which in turn may be part of the reconstructed video. It is noted that other suitable operations, such as a deblocking operation and the like, can be performed to improve the visual quality.
[0116] It is noted that the video encoders (403), (603), and (703), and the video decoders (410), (510), and (810) can be implemented using any suitable technique. In an embodiment, the video encoders (403), (603), and (703), and the video decoders (410), (510), and (810) can be implemented using one or more integrated circuits. In another embodiment, the video encoders (403), (603), and (603), and the video decoders (410), (510), and (810) can be implemented using one or more processors that execute software instructions.
[0117] This disclosure describes video coding technologies related to neural image compression technologies and/or neural video compression technologies, such as artificial intelligence (AI) based neural image compression (NIC). Aspects of the disclosure include content- adaptive online training in NIC, such as block-wise content- adaptive online training NIC methods for an end-to-end (E2E) optimized image coding framework based on neural networks. A neural network (NN) can include an artificial neural network (ANN), such as a deep neural network (DNN), a convolution neural network (CNN), or the like.
[0118] In an embodiment, a related hybrid video codec is difficult to be optimized as a whole. For example, an improvement of a single module (e.g., an encoder) in the hybrid video codec may not result in a coding gain in the overall performance. In a NN-based video coding framework, different modules can be jointly optimized from an input to an output to improve a final objective (e.g., rate-distortion performance, such as a rate-distortion loss L described in the disclosure) by performing a learning process or a training process (e.g., a machine learning process), and thus resulting in an E2E optimized NIC.
[0119] An exemplary NIC framework or system can be described as follows. The NIC framework can use an input block x as an input to a neural network encoder (e.g., an encoder based on neural networks such as DNNs) to compute a compressed representation (e.g., a compact representation) x that can be compact, for example, for storage and transmission purposes. A neural network decoder (e.g., a decoder based on neural networks such as DNNs) can use the compressed representation x as an input to reconstruct an output block (also referred to as a reconstructed block) x. In various embodiments, the input block x and reconstructed block x are in a spatial domain and the compressed representation x is in a domain different from the spatial domain. In some examples, the compressed representation x is quantized and entropy coded.
[0120] In some examples, a NIC framework can use a variational autoencoder (VAE) structure. In the VAE structure, the neural network encoder can directly use the entire input block x as the input to the neural network encoder. The entire input block x can pass through a set of neural network layers that work as a black box to compute the compressed representation x. The compressed representation x is an output of the neural network encoder. The neural network decoder can take the entire compressed representation x as an input. The compressed representation x can pass through another set of neural network layers that work as another black box to compute the reconstructed block x. A rate-distortion (R-D) loss L (x, x, x) can be optimized to achieve a trade-off between a distortion loss D(x, x) of the
reconstructed block x and bit consumption R of the compact representation x with a trade-off hyperparameter l.
L (x, x,x) = D ( x , x) + R (x) Eq. 1
[0121] A neural network (e.g., an ANN) can leam to perform tasks from examples, without task-specific programming. An ANN can be configured with connected nodes or artificial neurons. A connection between nodes can transmit a signal from a first node to a second node (e.g., a receiving node), and the signal can be modified by a weight which can be indicated by a weight coefficient for the connection. The receiving node can process signal(s) (i.e., input signal(s) for the receiving node) from node(s) that transmit the signal(s) to the receiving node and then generate an output signal by applying a function to the input signals. The function can be a linear function. In an example, the output signal is a weighted summation of the input signal(s). In an example, the output signal is further modified by a bias which can be indicated by a bias term, and thus the output signal is a sum of the bias and the weighted summation of the input signal(s). The function can include a nonlinear operation, for example, on the weighted sum or the sum of the bias and the weighted summation of the input signal(s). The output signal can be sent to node(s) (downstream node(s)) connected to the receiving node). The ANN can be represented or configured by parameters (e.g., weights of the connections and/or biases). The weights and/or the biases can be obtained by training the ANN with examples where the weights and/or the biases can be iteratively adjusted. The trained ANN configured with the determined weights and/or the determined biases can be used to perform tasks.
[0122] Nodes in an ANN can be organized in any suitable architecture. In various embodiments, nodes in an ANN are organized in layers including an input layer that receives input signal(s) to the ANN and an output layer that outputs output signal(s) from the ANN. In an embodiment, the ANN further includes layer(s) such as hidden layer(s) between the input layer and the output layer. Different layers may perform different kinds of transformations on respective inputs of the different layers. Signals can travel from the input layer to the output layer.
[0123] An ANN with multiple layers between an input layer and an output layer can be referred to as a DNN. In an embodiment, a DNN is a feedforward network where data flows from the input layer to the output layer without looping back. In an example, a DNN is a fully connected network where each node in one layer is connected to all nodes in the next layer. In an embodiment, a DNN is a recurrent neural network (RNN) where data can flow in any direction. In an embodiment, a DNN is a CNN.
[0124] A CNN can include an input layer, an output layer, and hidden layer(s) between the input layer and the output layer. The hidden layer(s) can include convolutional layer(s) (e.g., used in an encoder) that perform convolutions, such as a two-dimensional (2D) convolution. In an embodiment, a 2D convolution performed in a convolution layer is between a convolution kernel (also referred to as a filter or a channel, such as a 5 x 5 matrix) and an input signal (e.g., a 2D matrix such as a 2D block, a 256 x 256 matrix) to the convolution layer. In various examples, a dimension of the convolution kernel (e.g., 5 x 5) is smaller than a dimension of the input signal (e.g., 256 x 256). Thus, a portion (e.g., a 5 x 5 area) in the input signal (e.g., a 256 x 256 matrix) that is covered by the convolution kernel is smaller than an area (e.g., a 256 x 256 area) of the input signal, and thus can be referred to as a receptive field in the respective node in the next layer.
[0125] During the convolution, a dot product of the convolution kernel and the corresponding receptive field in the input signal is calculated. Thus, each element of the convolution kernel is a weight that is applied to a corresponding sample in the receptive field, and thus the convolution kernel includes weights. For example, a convolution kernel represented by a 5 x 5 matrix has 25 weights. In some examples, a bias is applied to the output signal of the convolution layer, and the output signal is based on a sum of the dot product and the bias.
[0126] The convolution kernel can shift along the input signal (e.g., a 2D matrix) by a size referred to as a stride, and thus the convolution operation generates a feature map or an activation map (e.g., another 2D matrix), which in turn contributes to an input of the next layer in the CNN. For example, the input signal is a 2D block having 256 x 256 samples, a stride is 2 samples (e.g., a stride of 2). For the stride of 2, the convolution kernel shifts along an X direction (e.g., a horizontal direction) and/or a Y direction (e.g., a vertical direction) by 2 samples.
[0127] Multiple convolution kernels can be applied in the same convolution layer to the input signal to generate multiple feature maps, respectively, where each feature map can represent a specific feature of the input signal. In general, a convolution layer with N channels (i.e., N convolution kernels), each convolution kernel having M x M samples, and a stride S can be specified as Conv: MxM cN sS. For example, a convolution layer with 192 channels, each convolution kernel having 5 x 5 samples, and a stride of 2 is specified as Conv: 5x5 cl92 s2. The hidden layer(s) can include deconvolutional layer(s) (e.g., used in a decoder) that perform deconvolutions, such as a 2D deconvolution. A deconvolution is an
inverse of a convolution. A deconvolution layer with 192 channels, each deconvolution kernel having 5 x 5 samples, and a stride of 2 is specified as DeConv: 5x5 cl92 s2.
[0128] In various embodiments, a CNN has the following benefits. A number of learnable parameters (i.e., parameters to be trained) in a CNN can be significantly smaller than a number of learnable parameters in a DNN, such as a feedforward DNN. In the CNN, a relatively large number of nodes can share a same filter (e.g., same weights) and a same bias (if the bias is used), and thus the memory footprint can be reduced because a single bias and a single vector of weights can be used across all receptive fields that share the same filter. For example, for an input signal having 100 x 100 samples, a convolution layer with a convolution kernel having 5 x 5 samples has 25 learnable parameters (e.g., weights). If a bias is used, then one channel uses 26 learnable parameters (e.g., 25 weights and one bias). If the convolution layer has N channels, the total learnable parameters is 26 x N. On the other hand, for a fully connected a layer in a DNN, 100 x 100 (i.e., 10000) weights are used for each node in the next layer. If the next layer has L nodes, then the total learnable parameters is 10000 x L.
[0129] A CNN can further include one or more other layer(s), such as pooling layer(s), fully connected layer(s) that can connect every node in one layer to every node in another layer, normalization layer(s), and/or the like. Layers in a CNN can be arranged in any suitable order and in any suitable architecture (e.g., a feed-forward architecture, a recurrent architecture). In an example, a convolutional layer is followed by other layer(s), such as pooling layer(s), fully connected layer(s), normalization layer(s), and/or the like.
[0130] A pooling layer can be used to reduce dimensions of data by combining outputs from a plurality of nodes at one layer into a single node in the next layer. A pooling operation for a pooling layer having a feature map as an input is described below. The description can be suitably adapted to other input signals. The feature map can be divided into sub-regions (e.g., rectangular sub-regions), and features in the respective sub-regions can be independently down-sampled (or pooled) to a single value, for example, by taking an average value in an average pooling or a maximum value in a max pooling.
[0131] The pooling layer can perform a pooling, such as a local pooling, a global pooling, a max pooling, an average pooling, and/or the like. A pooling is a form of nonlinear down-sampling. A local pooling combines a small number of nodes (e.g., a local cluster of nodes, such as 2 x 2 nodes) in the feature map. A global pooling can combine all nodes, for example, of the feature map.
[0132] The pooling layer can reduce a size of the representation, and thus reduce a number of parameters, a memory footprint, and an amount of computation in a CNN. In an example, a pooling layer is inserted between successive convolutional layers in a CNN. In an example, a pooling layer is followed by an activation function, such as a rectified linear unit (ReLU) layer. In an example, a pooling layer is omitted between successive convolutional layers in a CNN.
[0133] A normalization layer can be an ReLU, a leaky ReLU, a generalized divisive normalization (GDN), an inverse GDN (IGDN), or the like. An ReLU can apply a non saturating activation function to remove negative values from an input signal, such as a feature map, by setting the negative values to zero. A leaky ReLU can have a small slope (e.g., 0.01) for negative values instead of a flat slope (e.g., 0). Accordingly, if a value x is larger than 0, then an output from the leaky ReLU is x. Otherwise, the output from the leaky ReLU is the value x multiplied by the small slope (e.g., 0.01). In an example, the slope is determined before training, and thus is not learnt during training.
[0134] In NN-based image compression methods, such as DNN-based or CNN-based image compression methods, instead of directly encoding an entire image, the block-based or block-wise coding mechanism can be effective for compressing images in a DNN-based video coding standards such as FVC. An entire image can be partitioned into blocks of the same (or various) sizes, and the blocks can be compressed individually. In an embodiment, an image may be split into blocks with an equal size or non-equal sizes. The spilt blocks instead of the image can be compressed. FIG. 9A shows an example of a block-wise image coding according to an embodiment of the disclosure. An image (980) can be partitioned into blocks, e.g., blocks (981)-(996). The blocks (981)-(996) can be compressed, for example, according to a scanning order. IN an example shown in FIG. 9A, the blocks (981)-(989) are already compressed, and the blocks (990)-(996) are to be compressed.
[0135] An image can be treated as a block. In an embodiment, the image is compressed without being split into blocks. The entire image can be the input of an E2E NIC framework.
[0136] FIG. 9B shows an exemplary NIC framework (900) (e.g., a NIC system) according to an embodiment of the disclosure. The NIC framework (900) can be based on neural networks, such as DNNs and/or CNNs. The NIC framework (900) can be used to compress (e.g., encode) blocks and decompress (e.g., decode or reconstruct) compressed blocks (e.g., encoded blocks). The NIC framework (900) can include two sub-neural
networks, a first sub-NN (951) and a second sub-NN (952) that are implemented using neural networks.
[0137] The first sub-NN (951) can resemble an autoencoder and can be trained to generate a compressed block x of an input block x and decompress the compressed block x to obtain a reconstructed block x. The first sub-NN (951) can include a plurality of components (or modules), such as a main encoder neural network (or a main encoder network) (911), a quantizer (912), an entropy encoder (913), an entropy decoder (914), and a main decoder neural network (or a main encoder network) (915). Referring to FIG. 9, the main encoder network (911) can generate a latent or a latent representation y from the input block x (e.g., an block to be compressed or encoded). In an example, the main encoder network (911) is implemented using a CNN. A relationship between the latent representation y and the input block x can be described using Eq. 2. y = fi(x <¾) Eq. 2 where a parameter qί represents parameters, such as weights used in convolution kernels in the main encoder network (911) and biases (if biases are used in the main encoder network
(911))·
[0138] The latent representation y can be quantized using the quantizer (912) to generate a quantized latent y. The quantized latent y can be compressed, for example, using lossless compression by the entropy encoder (913) to generate the compressed block (e.g., an encoded block) x (931) that is a compressed representation x of the input block x. The entropy encoder (913) can use entropy coding techniques such as Huffman coding, arithmetic coding, or the like. In an example, the entropy encoder (913) uses arithmetic encoding and is an arithmetic encoder. In an example, the encoded block (931) is transmitted in a coded bitstream.
[0139] The encoded block (931) can be decompressed (e.g., entropy decoded) by the entropy decoder (914) to generate an output. The entropy decoder (914) can use entropy coding techniques such as Huffman coding, arithmetic coding, or the like that correspond to the entropy encoding techniques used in the entropy encoder (913). In an example, the entropy decoder (914) uses arithmetic decoding and is an arithmetic decoder. In an example, lossless compression is used in the entropy encoder (913), lossless decompression is used in the entropy decoder (914), and noises, such as due to the transmission of the encoded block (931) are omissible, the output from the entropy decoder (914) is the quantized latent y.
[0140] The main decoder network (915) can decode the quantized latent y to generate the reconstructed block x. In an example, the main decoder network (915) is implemented using a CNN. A relationship between the reconstructed block x (i.e., the output of the main decoder network (915)) and the quantized latent y (i.e., the input of the main decoder network (915)) can be described using Eq. 3. x = f (y; ¾) Eq. 3 where a parameter q2 represents parameters, such as weights used in convolution kernels in the main decoder network (915) and biases (if biases are used in the main decoder network (915)). Thus, the first sub-NN (951) can compress (e.g., encode) the input block x to obtain the encoded block (931) and decompress (e.g., decode) the encoded block (931) to obtain the reconstructed block x. The reconstructed block x can be different from the input block x due to quantization loss introduced by the quantizer (912).
[0141] The second sub-NN (952) can learn the entropy model (e.g., a prior probabilistic model) over the quantized latent y used for entropy coding. Thus, the entropy model can be a conditioned entropy model, e.g., a Gaussian mixture model (GMM), a Gaussian scale model (GSM) that is dependent on the input block x. The second sub-NN (952) can include a context model NN (916), an entropy parameter NN (917), a hyper encoder (921), a quantizer (922), an entropy encoder (923), an entropy decoder (924), and a hyper decoder (925). The entropy model used in the context model NN (916) can be an autoregressive model over latent (e.g., the quantized latent y). In an example, the hyper encoder (921), the quantizer (922), the entropy encoder (923), the entropy decoder (924), and the hyper decoder (925) form a hyper neural network (e.g., a hyperprior NN). The hyper neural network can represent information useful for correcting context-based predictions. Data from the context model NN (916) and the hyper neural network can be combined by the entropy parameter NN (917). The entropy parameter NN (917) can generate parameters, such as mean and scale parameters for the entropy model such as a conditional Gaussian entropy model (e.g., the GMM).
[0142] Referring to FIG. 9B, at an encoder side, the quantized latent y from the quantizer (912) is fed into the context model NN (916). At a decoder side, the quantized latent y from the entropy decoder (914) is fed into the context model NN (916). The context model NN (916) can be implemented using a neural network, such as a CNN. The context model NN (916) can generate an output ocm i based on a context y<L that is the quantized latent y available to the context model NN (916). The context y<L can include previously
quantized latent at the encoder side or previously entropy decoded quantized latent at the decoder side. A relationship between the output ocm i and the input (e.g., y<i) of the context model NN (916) can be described using Eq. 4. oCm,i = h y<i ,· <¾) Eq. 4 where a parameter q3 represents parameters, such as weights used in convolution kernels in the context model NN (916) and biases (if biases are used in the context model NN (916)).
[0143] The output ocm i from the context model NN (916) and an output ohc from the hyper decoder (925) are fed into the entropy parameter NN (917) to generate an output oep. The entropy parameter NN (917) can be implemented using a neural network, such as a CNN. A relationship between the output oep and the inputs (e.g., ocm i and ohc) of the entropy parameter NN (917) can be described using Eq. 5.
where a parameter q4 represents parameters, such as weights used in convolution kernels in the entropy parameter NN (917) and biases (if biases are used in the entropy parameter NN (917)). The output oep of the entropy parameter NN (917) can be used in determining (e.g., conditioning) the entropy model, and thus the conditioned entropy model can be dependent on the input block x, for example, via the output ohc from the hyper decoder (925). In an example, the output oep includes parameters, such as the mean and scale parameters, used to condition the entropy model (e.g., GMM). Referring to FIG. 9B, the entropy model (e.g., the conditioned entropy model) can be employed by the entropy encoder (913) and the entropy decoder (914) in entropy coding and entropy decoding, respectively.
[0144] The second sub-NN (952) can be described below. The latent y can be fed into the hyper encoder (921) to generate a hyper latent z. In an example, the hyper encoder (921) is implemented using a neural network, such as a CNN. A relationship between the hyper latent z and the latent y can be described using Eq. 6. z = /5(y; <¾) Eq. 6 where a parameter q5 represents parameters, such as weights used in convolution kernels in the hyper encoder (921) and biases (if biases are used in the hyper encoder (921)).
[0145] The hyper latent z is quantized by the quantizer (922) to generate a quantized latent z. The quantized latent z can be compressed, for example, using lossless compression by the entropy encoder (923) to generate side information, such as encoded bits (932) from the hyper neural network. The entropy encoder (923) can use entropy coding techniques such as Huffman coding, arithmetic coding, or the like. In an example, the entropy encoder (923)
uses arithmetic encoding and is an arithmetic encoder. In an example, the side information, such as the encoded bits (932), can be transmitted in the coded bitstream, for example, together with the encoded block (931).
[0146] The side information, such as the encoded bits (932), can be decompressed (e.g., entropy decoded) by the entropy decoder (924) to generate an output. The entropy decoder (924) can use entropy coding techniques such as Huffman coding, arithmetic coding, or the like. In an example, the entropy decoder (924) uses arithmetic decoding and is an arithmetic decoder. In an example, lossless compression is used in the entropy encoder (923), lossless decompression is used in the entropy decoder (924), and noises, such as due to the transmission of the side information are omissible, the output from the entropy decoder (924) can be the quantized latent z. The hyper decoder (925) can decode the quantized latent z to generate the output ohc. A relationship between the output ohc and the quantized latent z can be described using Eq. 7.
Ohc = fe (z; <¾) Eq. 7 where a parameter q6 represents parameters, such as weights used in convolution kernels in the hyper decoder (925) and biases (if biases are used in the hyper decoder (925)).
[0147] As described above, the compressed or encoded bits (932) can be added to the coded bitstream as the side information, which enables the entropy decoder (914) to use the conditional entropy model. Thus, the entropy model can be block-dependent and spatially adaptive, and thus can be more accurate than a fixed entropy model.
[0148] The NIC framework (900) can be suitably adapted, for example, to omit one or more components shown in FIG. 9, to modify one or more components shown in FIG. 9, and/or to include one or more components not shown in FIG. 9. In an example, a NIC framework using a fixed entropy model includes the first sub-NN (951), and does not include the second sub-NN (952). In an example, a NIC framework includes the components in the NIC framework (900) except the entropy encoder (923) and the entropy decoder (924).
[0149] In an embodiment, one or more components in the NIC framework (900) shown in FIG. 9 are implemented using neural network(s), such as CNN(s). Each NN-based component (e.g., the main encoder network (911), the main decoder network (915), the context model NN (916), the entropy parameter NN (917), the hyper encoder (921), or the hyper decoder (925)) in a NIC framework (e.g., the NIC framework (900)) can include any suitable architecture (e.g., have any suitable combinations of layers), include any suitable types of parameters (e.g., weights, biases, a combination of weights and biases, and/or the like), and include any suitable number of parameters.
[0150] In an embodiment, the main encoder network (911), the main decoder network (915), the context model NN (916), the entropy parameter NN (917), the hyper encoder (921), and the hyper decoder (925) are implemented using respective CNNs.
[0151] FIG. 10 shows an exemplary CNN of the main encoder network (911) according to an embodiment of the disclosure. For example, the main encoder network (911) includes four sets of layers where each set of layers includes a convolution layer 5x5 cl92 s2 followed by a GDN layer. One or more layers shown in FIG. 10 can be modified and /or omitted. Additional layer(s) can be added to the main encoder network (911).
[0152] FIG. 11 shows an exemplary CNN of the main decoder network (915) according to an embodiment of the disclosure. For example, the main decoder network (915) includes three sets of layers where each set of layers includes a deconvolution layer 5x5 cl92 s2 followed by an IGDN layer. In addition, the three sets of layers are followed by a deconvolution layer 5x5 c3 s2 followed by an IGDN layer. One or more layers shown in FIG. 11 can be modified and /or omitted. Additional layer(s) can be added to the main decoder network (915).
[0153] FIG. 12 shows an exemplary CNN of the hyper encoder (921) according to an embodiment of the disclosure. For example, the hyper encoder (921) includes a convolution layer 3x3 cl92 si followed by a leaky ReLU, a convolution layer 5x5 cl92 s2 followed by a leaky ReLU, and a convolution layer 5x5 cl92 s2. One or more layers shown in FIG. 12 can be modified and /or omitted. Additional layer(s) can be added to the hyper encoder (921).
[0154] FIG. 13 shows an exemplary CNN of the hyper decoder (925) according to an embodiment of the disclosure. For example, the hyper decoder (925) includes a deconvolution layer 5x5 cl92 s2 followed by a leaky ReLU, a deconvolution layer 5x5 c288 s2 followed by a leaky ReLU, and a deconvolution layer 3x3 c384 si. One or more layers shown in FIG. 13 can be modified and /or omitted. Additional layer(s) can be added to the hyper encoder (925).
[0155] FIG. 14 shows an exemplary CNN of the context model NN (916) according to an embodiment of the disclosure. For example, the context model NN (916) includes a masked convolution 5x5 c384 si for context prediction, and thus the context y<L in Eq. 4 includes a limited context (e.g., a 5x5 convolution kernel). The convolution layer in FIG. 14 can be modified. Additional layer(s) can be added to the context model NN (916).
[0156] FIG. 15 shows an exemplary CNN of the entropy parameter NN (917) according to an embodiment of the disclosure. For example, the entropy parameter NN (917) includes a convolution layer lxl c640 si followed by a leaky ReLU, a convolution layer lxl
c512 si followed by leaky ReLU, and a convolution layer lxl c384 si. One or more layers shown in FIG. 15 can be modified and /or omitted. Additional layer(s) can be added to the entropy parameter NN (917).
[0157] The NIC framework (900) can be implemented using CNNs, as described with reference to FIGs. 10-15. The NIC framework (900) can be suitably adapted such that one or more components (e.g., (911), (915), (916), (917), (921), and/or (925)) in the NIC framework (900) are implemented using any suitable types of neural networks (e.g., CNNs or non-CNN based neural networks). One or more other components the NIC framework (900) can be implemented using neural network(s).
[0158] The NIC framework (900) that includes neural networks (e.g., CNNs) can be trained to leam the parameters used in the neural networks. For example, when CNNs are used, the parameters represented by q1-q6, such as the weights used in the convolution kernels in the main encoder network (911) and biases (if biases are used in the main encoder network (911)), the weights used in the convolution kernels in the main decoder network (915) and biases (if biases are used in the main decoder network (915)), the weights used in the convolution kernels in the hyper encoder (921) and biases (if biases are used in the hyper encoder (921)), the weights used in the convolution kernels in the hyper decoder (925) and biases (if biases are used in the hyper decoder (925)), the weights used in the convolution kernel(s) in the context model NN (916) and biases (if biases are used in the context model NN (916)), and the weights used in the convolution kernels in the entropy parameter NN (917) and biases (if biases are used in the entropy parameter NN (917)), respectively, can be learned in the training process.
[0159] In an example, referring to FIG. 10, the main encoder network (911) includes four convolution layers where each convolution layer has a convolution kernel of 5x5 and 192 channels. Thus, a number of the weights used in the convolution kernels in the main encoder network (911) is 19200 (i.e., 4x5x5x192). The parameters used in the main encoder network (911) include the 19200 weights and optional biases. Additional parameter(s) can be included when biases and/or additional NN(s) are used in the main encoder network (911).
[0160] Referring to FIG. 9B, the NIC framework (900) includes at least one component or module built on neural network(s). The at least one component can include one or more of the main encoder network (911), the main decoder network (915), the hyper encoder (921), the hyper decoder (925), the context model NN (916), and the entropy parameter NN (917). The at least one component can be trained individually. In an example, the training process is used to learn the parameters for each component separately. The at
least one component can be trained jointly as a group. In an example, the training process is used to learn the parameters for a subset of the at least one component jointly. In an example, the training process is used to learn the parameters for all of the at least one component, and thus is referred to as an E2E optimization.
[0161] In the training process for one or more components in the NIC framework (900), the weights (or the weight coefficients) of the one or more components can be initialized. In an example, the weights are initialized based on pre-trained corresponding neural network model(s) (e.g., DNN models, CNN models). In an example, the weights are initialized by setting the weights to random numbers.
[0162] A set of training blocks can be employed to train the one or more components, for example, after the weights are initialized. The set of training blocks can include any suitable blocks having any suitable size(s). In some examples, the set of training blocks includes blocks from raw images, natural images, computer-generated images, and/or the like that are in the spatial domain. In some examples, the set of training blocks includes blocks from residue blocks or residue images having residue data in the spatial domain. The residue data can be calculated by a residue calculator (e.g., the residue calculator (723)). In some examples, raw images and/or residue images including residue data can be used directly to train neural networks in a NIC framework. Thus, raw images, residue images, blocks from raw images, and/or blocks from residue images can be used to train neural networks in a NIC framework.
[0163] For purposes of brevity, the training process below is described using a training block as an example. The description can be suitably adapted to a training image. A training block t of the set of training blocks can be passed through the encoding process in FIG. 9B to generate a compressed representation (e.g., encoded information, for example, to a bitstream). The encoded information can be passed through the decoding process described in FIG. 9B to compute and reconstruct a reconstructed block t.
[0164] For the NIC framework (900), two competing targets, e.g., a reconstruction quality and a bit consumption are balanced. A quality loss function (e.g., a distortion or distortion loss) D(t, t) can be used to indicate the reconstruction quality, such as a difference between the reconstruction (e.g., the reconstructed block t) and an original block (e.g., the training block t). A rate (or a rate loss) R can be used to indicate the bit consumption of the compressed representation. In an example, the rate loss R further includes the side information, for example, used in determining a context model.
For neural image compression, differentiable approximations of quantization can be used in E2E optimization. In various examples, in the training process of neural network-based image compression, noise injection is used to simulate quantization, and thus quantization is simulated by the noise injection instead of being performed by a quantizer (e.g., the quantizer (912)). Thus, training with noise injection can approximate the quantization error variationally. A bits per pixel (BPP) estimator can be used to simulate an entropy coder, and thus entropy coding is simulated by the BPP estimator instead of being performed by an entropy encoder (e.g., (913)) and an entropy decoder (e.g., (914)). Therefore, the rate loss R in the loss function L shown in Eq. 1 during the training process can be estimated, for example, based on the noise injection and the BPP estimator. In general, a higher rate R can allow for a lower distortion D, and a lower rate R can lead to a higher distortion D. Thus, a trade-off hyperparameter l in Eq. 1 can be used to optimize a joint R-D loss L where L as a summation of /.D and R can be optimized. The training process can be used to adjust the parameters of the one or more components (e.g., (911) (915)) in the NIC framework (900) such that the joint R-D loss L is minimized or optimized. In an example, a trade-off hyperparameter l can be used to optimize the joint Rate-Distortion (R-D) loss as:
L(x, x, rx, ... , rN, y) = (x, x) + R(å? st , å? iq) + bE Eq. 8 where E measures the distortion of the decoded block residuals compared with the original block residuals before encoding, which acts as regularization loss for the residual encoding/decoding DNNs and the encoding/decoding DNNs. b is a hyperparameter used to balance the importance of the regularization loss.
[0165] Various models can be used to determine the distortion loss D and the rate loss R, and thus to determine the joint R-D loss L in Eq. 1. In an example, the distortion loss D ( t , t) is expressed as a peak signal-to-noise ratio (PSNR) that is a metric based on mean squared error, a multiscale structural similarity (MS-SSIM) quality index, a weighted combination of the PSNR and MS-SSIM, or the like.
[0166] In an example, the target of the training process is to train the encoding neural network (e.g., the encoding DNN), such as a video encoder to be used on an encoder side and the decoding neural network (e.g., the decoding DNN), such as a video decoder to be used on a decoder side. In an example, referring to FIG. 9B, the encoding neural network can include the main encoder network (911), the hyper encoder (921), the hyper decoder (925), the context model NN (916), and the entropy parameter NN (917). The decoding neural network
can include the main decoder network (915), the hyper decoder (925), the context model NN (916), and the entropy parameter NN (917). The video encoder and/or the video decoder can include other component(s) that are based on NN(s) and/or not based on NN(s).
[0167] The NIC framework (e.g., the NIC framework (900)) can be trained in an E2E fashion. In an example, the encoding neural network and the decoding neural network are updated jointly in the training process based on backpropagated gradients in an E2E fashion.
[0168] After the parameters of the neural networks in the NIC framework (900) are trained, one or more components in the NIC framework (900) can be used to encode and/or decode blocks. In an embodiment, on the encoder side, the video encoder is configured to encode the input block x into the encoded block (931) to be transmitted in the bitstream. The video encoder can include multiple components in the NIC framework (900). In an embodiment, on the decoder side, the corresponding video decoder is configured to decode the encoded block (931) in the bitstream into the reconstructed block x. The video decoder can include multiple components in the NIC framework (900).
[0169] In an example, the video encoder includes all the components in the NIC framework (900), for example, when content-adaptive online training is employed.
[0170] FIG. 16A shows an exemplary video encoder (1600 A) according to an embodiment of the disclosure. The video encoder (1600 A) includes the main encoder network (911), the quantizer (912), the entropy encoder (913), and the second sub-NN (952) that are described with reference to FIG. 9B and detailed descriptions are omitted for purposes of brevity. FIG. 16B shows an exemplary video decoder (1600B) according to an embodiment of the disclosure. The video decoder (1600B) can correspond to the video encoder (1600A). The video decoder (1600B) can include the main decoder network (915), the entropy decoder (914), the context model NN (916), the entropy parameter NN (917), the entropy decoder (924), and the hyper decoder (925). Referring to FIGs. 16A-16B, on the encoder side, the video encoder (1600A) can generate the encoded block (931) and the encoded bits (932) to be transmitted in the bitstream. On the decoder side, the video decoder (1600B) can receive and decode the encoded block (931) and the encoded bits (932).
[0171] FIGs. 17-18 show an exemplary video encoder (1700) and a corresponding video decoder (1800), respectively, according to embodiments of the disclosure. Referring to FIG. 17, the encoder (1700) includes the main encoder network (911), the quantizer (912), and the entropy encoder (913). Examples of the main encoder network (911), the quantizer (912), and the entropy encoder (913) are described with reference to FIG. 9B. Referring to FIG. 18, the video decoder (1800) includes the main decoder network (915) and the entropy
decoder (914). Examples of the main decoder network (915) and the entropy decoder (914) are described with reference to FIG. 9B. Referring to FIGs. 17 and 18, the video encoder (1700) can generate the encoded block (931) to be transmitted in the bitstream. The video decoder (1800) can receive and decode the encoded block (931).
[0172] As described above, the NIC framework (900) including the video encoder and the video decoder can be trained based on images and/or blocks in the set of training images. In some examples, one or more blocks to be compressed (e.g., encoded) and/or transmitted have properties that are significantly different from the set of training blocks. Thus, encoding and decoding the one or more blocks using the video encoder and the video decoder trained based on the set of training blocks, respectively, can lead to a relatively poor R-D loss F (e.g., a relatively large distortion and/or a relatively large bit rate). Therefore, aspects of the disclosure describe a content-adaptive online training method for NIC, such as a block- wise content-adaptive online training method for NIC.
[0173] In the block-wise content- adaptive online training method, an input image can be split into blocks and one or more of the blocks can be used to update one or more parameters in a pretrained NIC framework to be one or more replacement parameters by optimizing rate-distortion performance. Neural network update information indicating the one or more replacement parameters or a subset of the one or more replacement parameters can be encoded into a bitstream along with the encoded one or more of the blocks. At a decoder side, a video decoder can decode the encoded one or more of the blocks and can achieve better compression performance by using the one or more replacement parameters or the subset of the one or more replacement parameters. The block-wise content-adaptive online training method can be used as a preprocessing step (e.g., a pre-encoding step) for boosting the compression performance of a pretrained E2E NIC compression method.
[0174] In order to differentiate the training process based on the set of training blocks and the content- adaptive online training process based on the one or more blocks to be compressed (e.g., encoded) and/or transmitted, the NIC framework (900), the video encoder, and the video decoder that are trained by the set of training blocks are referred to as the pretrained NIC framework (900), the pretrained video encoder, and the pretrained video decoder, respectively. Parameters in the pretrained NIC framework (900), the pretrained video encoder, or the pretrained video decoder are referred to as NIC pretrained parameters, encoder pretrained parameters, and decoder pretrained parameters, respectively. In an example, the NIC pretrained parameters includes the encoder pretrained parameters and the decoder pretrained parameters. In an example, the encoder pretrained parameters and the
decoder pretrained parameters do not overlap where none of the encoder pretrained parameters is included in the decoder pretrained parameters. For example, the encoder pretrained parameters (e.g., pretrained parameters in the main encoder network (911)) in (1700) and the decoder pretrained parameters (e.g., pretrained parameters in the main decoder network (915)) in (1800) do not overlap. In an example, the encoder pretrained parameters and the decoder pretrained parameters overlap where at least one of the encoder pretrained parameters is included in the decoder pretrained parameters. For example, the encoder pretrained parameters (e.g., pretrained parameters in the context model NN (916)) in (1600 A) and the decoder pretrained parameters (e.g., the pretrained parameters in the context model NN (916)) in (1600B) overlap. The NIC pretrained parameters can be obtained based on blocks and/or images in the set of training blocks.
[0175] The content- adaptive online training process can be referred to as a finetuning process and is described below. One or more pretrained parameters in the NIC pretrained parameters in the pretrained NIC framework (900) can further be trained (e.g., finetuned) based on the one or more blocks to be encoded and/or transmitted where the one or more blocks can be different from the set of training blocks. The one or more pretrained parameters used in the NIC pretrained parameters can be finetuned by optimizing the joint R-D loss L based on the one or more blocks. The one or more pretrained parameters that have been finetuned by the one or more blocks are referred to as the one or more replacement parameters or the one or more finetuned parameters. In an embodiment, after the one or more pretrained parameters in the NIC pretrained parameters have been finetuned (e.g., replaced) by the one or more replacement parameters, neural network update information is encoded into a bitstream to indicate the one or more replacement parameters or a subset of the one or more replacement parameters. In an example, the NIC framework (900) is updated (or finetuned) where the one or more pretrained parameters are replaced by the one or more replacement parameters, respectively.
[0176] In a first scenario, the one or more pretrained parameters includes a first subset of the one or more pretrained parameters and a second subset of the one or more pretrained parameters. The one or more replacement parameters includes a first subset of the one or more replacement parameters and a second subset of the one or more replacement parameters.
[0177] The first subset of the one or more pretrained parameters is used in the pretrained video encoder and is replaced by the first subset of the one or more replacement parameters, for example, in the training process. Thus, the pretrained video encoder is
updated to the updated video encoder by the training process. The neural network update information can indicate the second subset of the one or more replacement parameters that is to replace the second subset of the one or more replacement parameters. The one or more blocks can be encoded using the updated video encoder and transmitted in the bitstream with the neural network update information.
[0178] On the decoder side, the second subset of the one or more pretrained parameters is used in the pretrained video decoder. In an embodiment, the pretrained video decoder receives and decodes the neural network update information to determine the second subset of the one or more replacement parameters. The pretrained video decoder is updated to the updated video decoder when the second subset of the one or more pretrained parameters in the pretrained video decoder is replaced by the second subset of the one or more replacement parameters. The one or more encoded blocks can be decoded using the updated video decoder.
[0179] FIGs. 16A-16B show an example of the first scenario. For example, the one or more pretrained parameters include N 1 pretrained parameters in the pretrained context model NN (916) and N2 pretrained parameters in the pretrained main decoder network (915). Thus, the first subset of the one or more pretrained parameters include the N 1 pretrained parameters, and the second subset of the one or more pretrained parameters are identical to the one or more pretrained parameters. Accordingly, the N1 pretrained parameters in the pretrained context model NN (916) can be replaced by N1 corresponding replacement parameters such that the pretrained video encoder (1600A) can be updated to the updated video encoder (1600A). The pretrained context model NN (916) is also updated to be the updated context model NN (916). On the decoder side, the N1 pretrained parameters can be replaced by the N 1 corresponding replacement parameters and the N2 pretrained parameters can be replaced by N2 corresponding replacement parameters, updating the pretrained context model NN (916) to be the updated context model NN (916) and updating the pretrained main decoder network (915) to be the updated main decoder network (915). Thus, the pretrained video decoder (1600B) can be updated to the updated video decoder (1600B).
[0180] In a second scenario, none of the one or more pretrained parameters is used in the pretrained video encoder on the encoder side. Rather, the one or more pretrained parameters is used in the pretrained video decoder on the decoder side. Thus, the pretrained video encoder is not updated and continues to be the pretrained video encoder after the training process. In an embodiment, the neural network update information indicates the one
or more replacement parameters. The one or more blocks can be encoded using the pretrained video encoder and transmitted in the bitstream with the neural network update information.
[0181] On the decoder side, the pretrained video decoder can receive and decode the neural network update information to determine the one or more replacement parameters. The pretrained video decoder is updated to the updated video decoder when the one or more pretrained parameters in the pretrained video decoder is replaced by the one or more replacement parameters. The one or more encoded blocks can be decoded using the updated video decoder.
[0182] FIGs. 16A-16B show an example of the second scenario. For example, the one or more pretrained parameters include N2 pretrained parameters in the pretrained main decoder network (915). Thus, none of the one or more pretrained parameters is used in the pretrained video encoder (e.g., the pretrained video encoder (1600A)) on the encoder side. Thus, the pretrained video encoder (1600A) continues to be the pretrained video encoder after the training process. On the decoder side, the N2 pretrained parameters can be replaced by N2 corresponding replacement parameters, which updates the pretrained main decoder network (915) to the updated main decoder network (915). Thus, the pretrained video decoder (1600B) can be updated to the updated video decoder (1600B).
[0183] In a third scenario, the one or more pretrained parameters are used in the pretrained video encoder and are replaced by the one or more replacement parameters, for example, in the training process. Thus, the pretrained video encoder is updated to the updated video encoder by the training process. The one or more blocks can be encoded using the updated video encoder and transmitted in the bitstream. No neural network update information is encoded in the bitstream. On the decoder side, the pretrained video decoder is not updated and remains to be the pretrained video decoder. The one or more encoded blocks can be decoded using the pretrained video decoder.
[0184] FIGs. 16A-16B show an example of the third scenario. For example, the one or more pretrained parameters are in the pretrained main encoder network (911).
Accordingly, the one or more pretrained parameters in the pretrained main encoder network (911) can be replaced by the one or more replacement parameters such that the pretrained video encoder (1600 A) can be updated to be the updated video encoder (1600A). The pretrained main encoder network (911) is also updated to be the updated main encoder network (911). On the decoder side, the pre trained video decoder (1600B) is not updated.
[0185] In various example, such as described in the first, second, and third scenarios, video decoding may be performed by pretrained decoders having different capabilities, including decoders with and without capabilities to update the pretrained parameters.
[0186] In an example, compression performance can be increased by coding the one or more blocks with the updated video encoder and/or the updated video decoder as compared to coding the one or more blocks with the pretrained video encoder and the pretrained video decoder. Therefore, the content-adaptive online training method can be used to adapt a pretrained NIC framework (e.g., the pretrained NIC framework (900)) to target block content (e.g., the one or more blocks to be transmitted), and thus finetuning the pretrained NIC framework. Accordingly, the video encoder on the encoder side and/or the video decoder on the decoder side can be updated.
[0187] The content- adaptive online training method can be used as a preprocessing step (e.g., pre-encoding step) for boosting the compression performance of a pretrained E2E NIC compression method.
[0188] In an embodiment, the one or more blocks include a single input block, and the finetuning process is performed with the single input block. The NIC framework (900) is trained and updated (e.g., finetuned) based on the single input block. The updated video encoder on the encoder side and/or the updated video decoder on the decoder side can be used to code the single input block and optionally other input blocks. The neural network update information can be encoded into the bitstream together with the encoded single input block.
[0189] In an embodiment, the one or more blocks include multiple input blocks, and the finetuning process is performed with the multiple input blocks. The NIC framework (900) is trained and updated (e.g., finetuned) based on the multiple input blocks. The updated video encoder on the encoder side and/or the updated decoder on the decoder side can be used to code the multiple input blocks and optionally other input blocks. The neural network update information can be encoded into the bitstream together with the encoded multiple input blocks.
[0190] The rate loss R can increase with the signaling of the neural network update information in the bitstream. When the one or more blocks include the single input block, the neural network update information is signaled for each encoded block, and a first increase to the rate loss R is used to indicate the increase to the rate loss R due to the signaling of the neural network update information per block. When the one or more blocks include the multiple input blocks, the neural network update information is signaled for and shared by the
multiple input blocks, and a second increase to the rate loss R is used to indicate the increase to the rate loss R due to the signaling of the neural network update information per block. Because the neural network update information is shared by the multiple input blocks, the second increase to the rate loss R can be less than the first increase to the rate loss R. Thus, in some examples, it can be advantageous to finetune the NIC framework using the multiple input blocks.
[0191] In an embodiment, the one or more pretrained parameters to be updated are in one component of the pretrained NIC framework (900). Thus, the one component of the pretrained NIC framework (900) is updated based on the one or more replacement parameters, and other components of the pretrained NIC framework (900) are not updated.
[0192] The one component can be the pretrained context model NN (916), the pretrained entropy parameter NN (917), the pretrained main encoder network (911), the pretrained main decoder network (915), the pretrained hyper encoder (921), or the pretrained hyper decoder (925). The pretrained video encoder and/or the pretrained video decoder can be updated depending on which of the components in the pretrained NIC framework (900) is updated.
[0193] In an example, the one or more pretrained parameters to be updated are in the pretrained context model NN (916), and thus the pretrained context model NN (916) is updated and the remaining components (911), (915), (921), (917), and (925) are not updated. In an example, the pretrained video encoder on the encoder side and the pretrained video decoder on the decoder side include the pretrained context model NN (916), and thus both the pretrained video encoder and the pretrained video decoder are updated.
[0194] In an example, the one or more pretrained parameters to be updated are in the pretrained hyper decoder (925), and thus the pretrained hyper decoder (925) is updated and the remaining components (911), (915), (916), (917), and (921) are not updated. Thus, the pretrained video encoder is not updated and the pretrained video decoder is updated.
[0195] In an embodiment, the one or more pretrained parameters to be updated are in multiple components of the pretrained NIC framework (900). Thus, the multiple components of the pretrained NIC framework (900) are updated based on the one or more replacement parameters. In an example, the multiple components of the pretrained NIC framework (900) include all the components configured with neural networks (e.g., DNNs, CNNs). In an example, the multiple components of the pretrained NIC framework (900) include the CNN- based components: the pretrained main encoder network (911), the pretrained main decoder
network (915), the pretrained context model NN (916), the pretrained entropy parameter NN (917), the pretrained hyper encoder (921), and the pretrained hyper decoder (925).
[0196] As described above, in an example, the one or more pretrained parameters to be updated are in the pre trained video encoder of the pretrained NIC framework (900). In an example, the one or more pretrained parameters to be updated are in the pretrained video decoder of the NIC framework (900). In an example, the one or more pretrained parameters to be updated are in the pretrained video encoder and the pretrained video decoder of the pretrained NIC framework (900).
[0197] The NIC framework (900) can be based on neural networks, for example, one or more components in the NIC framework (900) can include neural networks, such as CNNs, DNNs, and/or the like. As described above, the neural networks can be specified by different types of parameters, such as weights, biases, and the like. Each neural network- based component (e.g., the context model NN (916), the entropy parameter NN (917), the main encoder network (911), the main decoder network (915), the hyper encoder (921), or the hyper decoder (925)) in the NIC framework (900) can be configured with suitable parameters, such as respective weights, biases, or a combination of weights and biases. When CNN(s) are used, the weights can include elements in convolution kernels. One or more types of parameters can be used to specify the neural networks. In an embodiment, the one or more pretrained parameters to be updated are bias term(s), and only the bias term(s) are replaced by the one or more replacement parameters. In an embodiment, the one or more pretrained parameters to be updated are weights, and only the weights are replaced by the one or more replacement parameters. In an embodiment, the one or more pretrained parameters to be updated include the weights and bias term(s), and all the pretrained parameters including the weights and bias term(s) are replaced by the one or more replacement parameters. In an embodiment, other parameters can be used to specify the neural networks, and the other parameters can be finetuned.
[0198] The finetuning process can include multiple epochs (e.g., iterations) where the one or more pretrained parameters are updated in an iterative finetuning process. The finetuning process can stop when a training loss has flattened or is about to flatten. In an example, the finetuning process stops when the training loss (e.g., a R-D loss L) is below a first threshold. In an example, the finetuning process stops when a difference between two successive training losses is below a second threshold.
[0199] Two hyperparameters (e.g., a step size and a maximum number of steps) can be used in the finetuning process together with a loss function (e.g., an R-D loss L). The
maximum number of iterations can be used as a threshold of a maximum number of iterations to terminate the finetuning process. In an example, the finetuning process stops when a number of iterations reaches the maximum number of iterations.
[0200] The step size can indicate a learning rate of the online training process (e.g., the online finetuning process). The step size can be used in a gradient descent algorithm or a backpropagation calculation performed in the finetuning process. A step size can be determined using any suitable method.
[0201] The step size for each block in an image can be different. In an embodiment, different step sizes can be assigned for an image in order to achieve a better compression result (e.g., a better R-D loss L).
[0202] In some examples, a video encoder and a video decoder based on a NIC framework (e.g., the NIC framework (900)) can encode and decode an image directly. Thus, the block-wise content-adaptive online training method can be adapted to update certain parameters in the NIC framework and thus the video encoder and/or the video decoder by using one or more images directly. Different images can have different step sizes to achieve optimized compression result.
[0203] In an embodiment, different step sizes are used for blocks with different types of contents to achieve optimal results. Different types can refer to different variances. In an example, the step size is determined based on a variance of a block used to update a NIC framework. For example, a step size of a block having a high variance is larger than a step size of a block having a low variance where the high variance is larger than the low variance.
[0204] In an embodiment, a step size is chosen based on characteristics of a block or an image, such as RGB variance of the block. In an embodiment, a step size is chosen based on RD performance (e.g., a R-D loss L) of the block. Multiple sets of replacement parameter(s) can be generated based on different step sizes, and the set with the better compression performance (e.g., a smaller R-D loss) can be chosen.
[0205] In an embodiment, a first step size can be used to ran a certain number (e.g., 100) iterations. Then, a second step size (e.g., the first step size plus or minus a size increment) can be used to ran the certain number of iterations. Results from the first step size and the second step size can be compared to determine a step size to be used. More than two step sizes may be tested to determine an optimal step size.
[0206] A step size can vary during the finetuning process. The step size can have an initial value at an onset of the finetuning process, and the initial value can be reduced (e.g., halved) at a later stage of the finetuning process, for example, after a certain number of
iterations to achieve a finer tuning. The step size or the learning rate can be varied by a scheduler during the iterative online training. The scheduler can include a parameter adjustment method used to adjust the step size. The scheduler can determine a value for the step size such that the step size can increase, decrease, or remain constant in a number of intervals. In an example, the learning rate is altered in each step by the scheduler. A single scheduler or multiple different schedulers can be used for different blocks. Thus, multiple sets of replacement parameter(s) can be generated based on the multiple schedulers, and one of the multiple sets of replacement parameter(s) with the better compression performance (e.g., a smaller R-D loss) can be chosen.
[0207] In an embodiment, multiple learning rate schedules are assigned for different blocks in order to achieve better compression result. In an embodiment, all blocks in an image share a same learning rate schedule. In an embodiment, selection of learning rate schedules is based on characteristics of a block, such as a RGB variance of the block. In an embodiment, selection of learning rate schedules is based on the RD performance of the block.
[0208] In an embodiment, different blocks can be used to update different parameters in different components (e.g., the context model NN (916) or the hyper decoder (925)) in the NIC framework. For example, a first block is used to update parameters in the context model NN (916), and a second block is used to update parameters in the hyper decoder (925).
[0209] In an embodiment, different blocks can be used to update different types of parameters (e.g., biases or weights) in the NIC framework. For example, a first block is used to update at least one bias in one or more neural networks in the NIC framework, and a second block is used to update at least one weight in one or more neural networks in the NIC framework.
[0210] In an embodiment, multiple blocks (e.g., all blocks) in an image update the same one or more parameters.
[0211] In an embodiment, the one or more parameters to be updated are chosen based on a characteristic of a block, such as a RGB variance of the block. In an embodiment, the one or more parameters to be updated are chosen based on a RD performance of the block.
[0212] At the end of the finetuning process, one or more updated parameters can be computed for the respective one or more replacement parameters. In an embodiment, the one or more updated parameters are calculated as differences between the one or more replacement parameters and the corresponding one or more pretrained parameters. In an
embodiment, the one or more updated parameters are the one or more replacement parameters, respectively.
[0213] In an embodiment, the one or more updated parameters can be generated from the one or more replacement parameters, for example, using a certain linear or nonlinear transform, and the one or more updated parameters are representative parameter(s) generated based on the one or more replacement parameters. The one or more replacement parameters are transformed into the one or more updated parameters for better compression.
[0214] A first subset of the one or more updated parameters corresponds to the first subset of the one or more replacement parameters, and a second subset of the one or more updated parameters corresponds to the second subset of the one or more replacement parameters.
[0215] In an embodiment, different blocks have different relationships between the one or more updated parameters and the one or more replacement parameters. For example, for a first block the one or more updated parameters are calculated as differences between the one or more replacement parameters and the corresponding one or more pretrained parameters. For a second block, the one or more updated parameters are the one or more replacement parameters, respectively.
[0216] In an embodiment, multiple blocks (e.g., all blocks) in an image have a same relationship between the one or more updated parameters and the one or more replacement parameters.
[0217] In an embodiment, the relationship between the one or more updated parameters and the one or more replacement parameters is chosen based on characteristics of a block, such as a RGB variance of the block. In an embodiment, the relationship between the one or more updated parameters and the one or more replacement parameters is chosen based on a RD performance of the block.
[0218] In an example, the one or more updated parameters can be compressed, for example, using LZMA2 that is a variation of a Lempel-Ziv-Markov chain algorithm (LZMA), a bzip2 algorithm, or the like. In an example, compression is omitted for the one or more updated parameters. In some embodiments, the one or more updated parameters or the second subset of the one or more updated parameters can be encoded into the bitstream as the neural network update information where the neural network update information indicates the one or more replacement parameters or the second subset of the one or more replacement parameters.
[0219] In an embodiment, compression methods for the one or more updated parameters are different for different blocks. For example, for a first block, the LZMA2 is used to compress the one or more updated parameters, and for a second block, the bzip2 is used to compress the one or more updated parameters. In an embodiment, a same compression method is used to compress the one or more updated parameters for multiple blocks (e.g., all blocks) in an image. In an embodiment, a compression method is chosen based on characteristics of a block, such as a RGB variance of the block. In an embodiment, a compression method is chosen based on RD performance of the block.
[0220] After the finetuning process, in some examples, the pretrained video encoder on the encoder side can be updated or finetuned based on (i) the first subset of the one or more replacement parameters or (ii) the one or more replacement parameters. An input block (e.g., one of the one or more blocks used to in the finetuning process) can be encoded into the bitstream using the updated video encoder. Thus, the bitstream includes both the encoded block and the neural network update information.
[0221] If applicable, in an example, the neural network update information is decoded (e.g., decompressed) by the pretrained video decoder to obtain the one or more updated parameters or the second subset of the one or more updated parameters. In an example, the one or more replacement parameters or the second subset of the one or more replacement parameters can be obtained based on the relationship between the one or more updated parameters and the one or more replacement parameters described above. The pretrained video decoder can be finetuned and the updated video decoded can be used to decode the encoded block, as described above.
[0222] The NIC framework can include any type of neural networks and use any neural network-based image compression methods, such as a context-hyperprior encoder- decoder framework (e.g., the NIC framework shown in FIG. 9B), a scale-hyperprior encoder- decoder framework, a Gaussian Mixture Likelihoods framework and variants of the Gaussian Mixture Likelihoods framework, an RNN-based recursive compression method and variants of the RNN-based recursive compression method, and the like.
[0223] Compared with related E2E image compression methods, the content-adaptive online training methods and apparatus in the disclosure can have the following benefits. Adaptive online training mechanisms are exploited to improve the NIC coding efficiency.
Use of a flexible and general framework can accommodate various types of pretrained frameworks and quality metrics. For example, certain pretrained parameters in the various
types of pretrained frameworks can be replaced by using online training with blocks to be encoded and transmitted·
[0224] FIG. 19 shows a flow chart outlining a process (1900) according to an embodiment of the disclosure. The process (1900) can be used to encode a block, such as a block in a raw image or a block in a residue image. In various embodiments, the process (1900) is executed by processing circuitry, such as the processing circuitry in the terminal devices (310), (320), (330) and (340), the processing circuitry that performs functions of the video encoder (1600A), the processing circuitry that performs functions of the video encoder (1700). In an example, the processing circuitry performs a combination of functions of (i) one of the video encoder (403), (603), and (703) and (ii) one of the video encoder (1600A) and the video encoder (1700). In some embodiments, the process (1900) is implemented in software instructions, thus when the processing circuitry executes the software instructions, the processing circuitry performs the process (1900). The process starts at (S1901). In an example, an NIC framework is based on neural networks. In an example, the NIC framework is the NIC framework (900) described with reference to FIG. 9B. The NIC framework can be based on CNNs, such as described with reference to FIGs. 10-15. A video encoder (e.g., (1600A) or (1700)) and a corresponding video decoder (e.g., (1600B) or (1800)) can include multiple components in the NIC framework, as described above. The NIC framework based on neural networks are pretrained, and thus the video encoder and the video decoder are pretrained. The process (1900) proceeds to (S1910).
[0225] At (S 1910), a finetuning process is performed on the NIC framework based on one or more blocks (or input block(s)). The input block(s) can be any suitable block(s) having any suitable size(s). In some examples, the input block(s) include block(s) in a raw image, a natural image, a computer-generated image, and/or the like that are in the spatial domain.
[0226] In some examples, the input block(s) include residue data in the spatial domain, for example, calculated by a residue calculator (e.g., the residue calculator (723)). Components in various apparatuses can be suitably combined to achieve (S1910), for example, referring to FIGs. 7 and 9, the residue block from the residue calculator can be fed into the main encoder network (911) in the NIC framework.
[0227] One of more parameters (e.g., one or more pretrained parameters) in one or more neural networks (e.g., one or more pretrained neural networks) in the NIC framework (e.g., the pretrained NIC framework) can be updated to be one or more replacement parameters, respectively, as described above. In an embodiment, the one of more parameters
in the one or more neural networks are being updated during the training process described in (S1910), for example, in each step.
[0228] In an embodiment, at least one neural network in the video encoder (e.g., the pretrained video encoder) is configured with a first subset of the one or more pretrained parameters, and thus the at least one neural network in the video encoder can be updated based on a corresponding first subset of the one or more replacement parameters. In an example, the first subset of the one or more replacement parameters includes all of the one or more replacement parameters. In an example, the at least one neural network in the video encoder is updated when the first subset of the one or more pretrained parameters is replaced with the first subset of the one or more replacement parameters, respectively. In an example, the at least one neural network in the video encoder is updated iteratively in the finetuning process. In an example, none of the one or more pretrained parameters are included in the video encoder, and thus the video encoder is not updated and remains the pretrained video encoder.
[0229] At (SI 920), one of the one or more blocks can be encoded using the video encoder having the at least one updated neural network where the video encoder is configured with the first subset of the one or more replacement parameters. In an example, the one of the one or more blocks is encoded after the at least one neural network in the video encoder is updated.
[0230] The step (SI 920) can be suitably adapted. For example, the video encoder is not updated when none of the one or more replacement parameters are included in the at least one neural network in the video encoder, and thus the one of the one or more blocks can be encoded using the pretrained video encoder (e.g., the video encoder including the at least one pretrained neural network).
[0231] At (S1930), neural network update information indicating a second subset of the one or more replacement parameters can be encoded into the bitstream. In an example, the second subset of the one or more replacement parameters is to be used to update at least one neural network in the video decoder on the decoder side. The step (S1930) can be omitted, and none of the neural networks in the video decoder is updated, for example, if the second subset of the one or more replacement parameters includes no parameters and no neural network update information is signaled in the bitstream.
[0232] At (SI 940), the bitstream including the encoded one of the one or more blocks and the neural network update information can be transmitted. The step (S1940) can be suitably adapted. For example, if the step (S1930) is omitted, the bitstream does not include
the neural network update information. The process (1900) proceeds to (S1999), and terminates.
[0233] The process (1900) can be suitably adapted to various scenarios and steps in the process (1900) can be adjusted accordingly. One or more of the steps in the process (1900) can be adapted, omitted, repeated, and/or combined. Any suitable order can be used to implement the process (1900). Additional step(s) can be added. For example, in addition to encoding the one of the one or more blocks, the one or more blocks are encoded in (S1920), and are transmitted in (S1940).
[0234] In some examples of the process (1900), the one of the one or more blocks is encoded by the updated video encoder and transmitted in the bitstream. As the finetuning process is based on the one or more blocks, the finetuning process is based on the context to be encoded, and thus is context-based.
[0235] In some examples, the neural network update information further indicates what parameter(s) the second subset of the one or more pretrained parameters (or the corresponding second subset of the one or more replacement parameters) are so that corresponding pretrained parameter(s) in the video decoder can be updated. The neural network update information can indicate component information (e.g., (915)), layer information (e.g., the fourth layer DeConv: 5x5 c3 s2), channel information (e.g., the second channel), and/or the like of the second subset of the one or more pretrained parameters. Therefore, referring to FIG. 11, the second subset of the one or more replacement parameters includes the convolution kernel of the second channel of DeConv: 5x5 c3 s2 in the main decoder network (915). Thus, the convolution kernel of the second channel of DeConv: 5x5 c3 s2 in the pretrained main decoder network (915) is updated. In some examples, the component information (e.g., (915)), the layer information (e.g., the fourth layer DeConv:
5x5 c3 s2), the channel information (e.g., the second channel), and/or the like of the second subset of the one or more pretrained parameters are pre-determined and stored in the pretrained video decoder, and thus are not signaled.
[0236] FIG. 20 shows a flow chart outlining a process (2000) according to an embodiment of the disclosure. The process (2000) can be used in the reconstruction of an encoded block. In various embodiments, the process (2000) is executed by processing circuitry, such as the processing circuitry in the terminal devices (310), (320), (330) and (340), the processing circuitry that performs functions of the video decoder (1600B), the processing circuitry that performs functions of the video decoder (1800). In an example, the processing circuitry performs a combination of functions of (i) one of the video decoder
(410), the video decoder (510), and the video decoder (810) and (ii) one of the video decoder (1600B) or the video decoder (1800). In some embodiments, the process (2000) is implemented in software instructions, thus when the processing circuitry executes the software instructions, the processing circuitry performs the process (2000). The process starts at (S2001). In an example, an NIC framework is based on neural networks. In an example, the NIC framework is the NIC framework (900) described with reference to FIG. 9B. The NIC framework can be based on CNNs, such as described with reference to FIGs. 10-15. A video decoder (e.g., (1600B) or (1800)) can include multiple components in the NIC framework, as described above. The NIC framework based on neural networks can be pretrained. The video decoder can be pretrained with pretrained parameters. The process (2000) proceeds to (S2010).
[0237] At (S2010), first neural network update information in a coded bitstream can be decoded. The first neural network update information can be for a first neural network in the video decoder. The first neural network can be configured with first pretrained parameters. The first neural network update information can correspond to a first block in an image to be reconstructed and indicate a first replacement parameter corresponding to a first pretrained parameter in the first pretrained parameters.
[0238] In an example, the first pretrained parameter is a pretrained bias term.
[0239] In an example, the first pretrained parameter is a pretrained weight coefficient.
[0240] In an embodiment, the video decoder includes multiple neural networks. The first neural network update information can indicate update information for one or more remaining neural networks of the multiple neural networks. For example, the first neural network update information further indicates one or more replacement parameters for the one or more remaining neural networks in the multiple neural networks. The one or more replacement parameters correspond to one or more respective pretrained parameters for the one or more remaining neural networks. In an example, each of the first pre trained parameter and the one or more pretrained parameters is a respective pretrained bias term. In an example, each of the first pretrained parameter and the one or more pretrained parameters is a respective pretrained weight coefficient. In an example, the first pretrained parameter and the one or more pretrained parameters include one or more pretrained bias terms and one or more pretrained weight coefficients in the multiple neural networks.
[0241] In an example, the first neural network update information indicates update information for a subset of the multiple neural networks, and a remaining subset of the multiple neural networks is not updated.
[0242] In an example, the video decoder is the video decoder (1800) shown in FIG. 18. The first neural network is the main decoder network (915).
[0243] In an example, the video decoder is the video decoder (1600B) shown in FIG. 16B. The multiple neural networks in the video decoder include the main decoder network (915), the context model NN (916), the entropy parameter NN (917), and the hyper decoder (925). The first neural network is one of the main decoder network (915), the context model NN (916), the entropy parameter NN (917), and the hyper decoder (925), such as the context model NN (916). In an example, the first neural network update information further includes the one or more replacement parameters for the one or more remaining neural networks (e.g., the main decoder network (915), the entropy parameter NN (917), and/or the hyper decoder (925)) in the video decoder.
[0244] At (S2020), the first replacement parameter can be determined based on the first neural network update information. In an embodiment, an updated parameter is obtained from the first neural network update information. In an example, the updated parameter can be obtained from the first neural network update information by decompression (e.g., the LZMA2 or the bzip2 algorithm).
[0245] In an example, the first neural network update information indicates the updated parameter being a difference between the first replacement parameter and the first pretrained parameter. The first replacement parameter can be calculated according to a sum of the updated parameter and the first pretrained parameter.
[0246] In an embodiment, the first replacement parameter is determined to be the updated parameter.
[0247] In an embodiment, the updated parameter is a representative parameter generated (e.g., using a linear or a nonlinear transform) based on the first replacement parameter on an encoder side, and the first replacement parameter is obtained based on the representative parameter.
[0248] At (S2030), the first neural network in the video decoder can be updated (or finetuned) based on the first replacement parameter, for example, by replacing the first pretrained parameter with the first replacement parameter in the first neural network. If the video decoder includes the multiple neural networks, and the first neural network update information indicates the update information (e.g., additional replacement parameter(s)) for the multiple neural networks, the multiple neural networks can be updated. For example, the first neural network update information further includes the one or more replacement parameters for the one or more remaining neural networks in the video decoder, and the one
or more remaining neural networks can be updated based on the one or more replacement parameters.
[0249] At (S2040), the encoded first block in the bitstream can be decoded by the video decoder that is updated, for example, based on the updated first neural network. An output block generated at (S2040) can be any suitable block having any suitable size. In some examples, the output block is a reconstructed block in a reconstructed image in the spatial domain.
[0250] In some examples, the output block of the video decoder includes residue data in the spatial domain, and thus further processing can be used to generate a reconstructed block based on the output block. For example, the reconstruction module (874) is configured to combine, in the spatial domain, the residue data and prediction results (as output by the inter or intra prediction modules) to form reconstructed blocks that may be part of a reconstructed image. Additional suitable operations, such as a deblocking operation and the like, can be performed to improve the visual quality. Components in various apparatuses can be suitably combined to achieve (S2040), for example, referring to FIGs. 8 and 9, the residue data from the main decoder network (915) in the video decoder and the corresponding prediction results are fed into the reconstruction module (874) to generate the reconstructed image.
[0251] In an example, the bitstream further includes one or more encoded bits used to determine a context model for decoding the encoded block. The video decoder can include a main decoder network (e.g., (911)), a context model network (e.g., (916)), an entropy parameter network (e.g., (917)), and a hyper decoder network (e.g., (925)). The neural network is one of the main decoder network, the context model network, the entropy parameter NN, and the hyper decoder network. The one or more encoded bits can be decoded using the hyper decoder network. An entropy model (e.g., a context model) can be determined using the context model network and the entropy parameter network based on the decoded bits and quantized latent of the encoded block that is available to the context model network. The encoded block can be decoded using the main decoder network and the entropy model.
[0252] The process (2000) proceeds to (S2099), and terminates·
[0253] The process (2000) can be suitably adapted to various scenarios and steps in the process (2000) can be adjusted accordingly. One or more of the steps in the process (2000) can be adapted, omitted, repeated, and/or combined. Any suitable order can be used to implement the process (2000). Additional step(s) can be added.
[0254] In an example, at (S2040), another block in the coded bitstream is decoded based on the updated first neural network for the first block.
[0255] In an example, at (S2010), second neural network update information in the coded bitstream for a second neural network in the video decoder is decoded. The second neural network is configured with second pretrained parameters. The second neural network update information corresponds to a second block in the image to be reconstructed and indicates a second replacement parameter corresponding to a second pretrained parameter in the second pretrained parameters. The second neural network (e.g., the context model NN (916)) can be different from the first neural network (e.g., the main decoder network (915)). At (S2030), The second neural network in the video decoder can be updated based on the second replacement parameter. At (S2040), the second block can be decoded based on the updated second neural network for the second block. In an example, the first pretrained parameter is a one of a pretrained weight coefficient and a pretrained bias term. In an example, the second pretrained parameter is another one of the pretrained weight coefficient and the pretrained bias term.
[0256] Embodiments in the disclosure may be used separately or combined in any order. Further, each of the methods (or embodiments), an encoder, and a decoder may be implemented by processing circuitry (e.g., one or more processors or one or more integrated circuits). In one example, the one or more processors execute a program that is stored in a non-transitory computer-readable medium.
[0257] This disclosure does not put any restrictions on methods used for an encoder such as a neural network based encoder, a decoder such as a neural network based decoder. Neural network(s) used in an encoder, a decoder, and/or the like can be any suitable types of neural network(s), such as a DNN, a CNN, and the like.
[0258] Thus, the content- adaptive online training methods of this disclosure can accommodate different types of NIC frameworks, e.g., different types of encoding DNNs, decoding DNNs, encoding CNNs, decoding CNNs, and/or the like.
[0259] The techniques described above, can be implemented as computer software using computer-readable instructions and physically stored in one or more computer-readable media. For example, FIG. 21 shows a computer system (2100) suitable for implementing certain embodiments of the disclosed subject matter.
[0260] The computer software can be coded using any suitable machine code or computer language, that may be subject to assembly, compilation, linking, or like mechanisms to create code comprising instructions that can be executed directly, or through
interpretation, micro-code execution, and the like, by one or more computer central processing units (CPUs), Graphics Processing Units (GPUs), and the like.
[0261] The instructions can be executed on various types of computers or components thereof, including, for example, personal computers, tablet computers, servers, smartphones, gaming devices, internet of things devices, and the like.
[0262] The components shown in FIG. 21 for computer system (2100) are exemplary in nature and are not intended to suggest any limitation as to the scope of use or functionality of the computer software implementing embodiments of the present disclosure. Neither should the configuration of components be interpreted as having any dependency or requirement relating to any one or combination of components illustrated in the exemplary embodiment of a computer system (2100).
[0263] Computer system (2100) may include certain human interface input devices. Such a human interface input device may be responsive to input by one or more human users through, for example, tactile input (such as: keystrokes, swipes, data glove movements), audio input (such as: voice, clapping), visual input (such as: gestures), olfactory input (not depicted). The human interface devices can also be used to capture certain media not necessarily directly related to conscious input by a human, such as audio (such as: speech, music, ambient sound), images (such as: scanned images, photographic images obtain from a still image camera), video (such as two-dimensional video, three-dimensional video including stereoscopic video).
[0264] Input human interface devices may include one or more of (only one of each depicted): keyboard (2101), mouse (2102), trackpad (2103), touch screen (2110), data-glove (not shown), joystick (2105), microphone (2106), scanner (2107), camera (2108).
[0265] Computer system (2100) may also include certain human interface output devices. Such human interface output devices may be stimulating the senses of one or more human users through, for example, tactile output, sound, light, and smell/taste. Such human interface output devices may include tactile output devices (for example tactile feedback by the touch-screen (2110), data-glove (not shown), or joystick (2105), but there can also be tactile feedback devices that do not serve as input devices), audio output devices (such as: speakers (2109), headphones (not depicted)), visual output devices (such as screens (2110) to include CRT screens, LCD screens, plasma screens, OLED screens, each with or without touch-screen input capability, each with or without tactile feedback capability — some of which may be capable to output two dimensional visual output or more than three
dimensional output through means such as stereographic output; virtual-reality glasses (not depicted), holographic displays and smoke tanks (not depicted)), and printers (not depicted).
[0266] Computer system (2100) can also include human accessible storage devices and their associated media such as optical media including CD/DVD ROM/RW (2120) with CD/DVD or the like media (2121), thumb-drive (2122), removable hard drive or solid state drive (2123), legacy magnetic media such as tape and floppy disc (not depicted), specialized ROM/ASIC/PLD based devices such as security dongles (not depicted), and the like.
[0267] Those skilled in the art should also understand that term “computer readable media” as used in connection with the presently disclosed subject matter does not encompass transmission media, carrier waves, or other transitory signals.
[0268] Computer system (2100) can also include an interface (2154) to one or more communication networks (2155). Networks can for example be wireless, wireline, optical. Networks can further be local, wide-area, metropolitan, vehicular and industrial, real-time, delay-tolerant, and so on. Examples of networks include local area networks such as Ethernet, wireless LANs, cellular networks to include GSM, 3G, 4G, 5G, LTE and the like, TV wireline or wireless wide area digital networks to include cable TV, satellite TV, and terrestrial broadcast TV, vehicular and industrial to include CANBus, and so forth. Certain networks commonly require external network interface adapters that attached to certain general purpose data ports or peripheral buses (2149) (such as, for example USB ports of the computer system (2100)); others are commonly integrated into the core of the computer system (2100) by attachment to a system bus as described below (for example Ethernet interface into a PC computer system or cellular network interface into a smartphone computer system). Using any of these networks, computer system (2100) can communicate with other entities. Such communication can be uni-directional, receive only (for example, broadcast TV), uni-directional send-only (for example CANbus to certain CANbus devices), or bi directional, for example to other computer systems using local or wide area digital networks. Certain protocols and protocol stacks can be used on each of those networks and network interfaces as described above.
[0269] Aforementioned human interface devices, human-accessible storage devices, and network interfaces can be attached to a core (2140) of the computer system (2100).
[0270] The core (2140) can include one or more Central Processing Units (CPU) (2141), Graphics Processing Units (GPU) (2142), specialized programmable processing units in the form of Field Programmable Gate Areas (FPGA) (2143), hardware accelerators for certain tasks (2144), graphics adapters (2150), and so forth. These devices, along with Read-
only memory (ROM) (2145), Random-access memory (2146), internal mass storage such as internal non-user accessible hard drives, SSDs, and the like (2147), may be connected through a system bus (2148). In some computer systems, the system bus (2148) can be accessible in the form of one or more physical plugs to enable extensions by additional CPUs, GPU, and the like. The peripheral devices can be attached either directly to the core’s system bus (2148), or through a peripheral bus (2149). In an example, the screen (2110) can be connected to the graphics adapter (2150). Architectures for a peripheral bus include PCI, USB, and the like.
[0271] CPUs (2141), GPUs (2142), FPGAs (2143), and accelerators (2144) can execute certain instructions that, in combination, can make up the aforementioned computer code. That computer code can be stored in ROM (2145) or RAM (2146). Transitional data can be also be stored in RAM (2146), whereas permanent data can be stored for example, in the internal mass storage (2147). Fast storage and retrieve to any of the memory devices can be enabled through the use of cache memory, that can be closely associated with one or more CPU (2141), GPU (2142), mass storage (2147), ROM (2145), RAM (2146), and the like.
[0272] The computer readable media can have computer code thereon for performing various computer-implemented operations. The media and computer code can be those specially designed and constructed for the purposes of the present disclosure, or they can be of the kind well known and available to those having skill in the computer software arts.
[0273] As an example and not by way of limitation, the computer system having architecture (2100), and specifically the core (2140) can provide functionality as a result of processor(s) (including CPUs, GPUs, FPGA, accelerators, and the like) executing software embodied in one or more tangible, computer-readable media. Such computer-readable media can be media associated with user-accessible mass storage as introduced above, as well as certain storage of the core (2140) that are of non-transitory nature, such as core-internal mass storage (2147) or ROM (2145). The software implementing various embodiments of the present disclosure can be stored in such devices and executed by core (2140). A computer- readable medium can include one or more memory devices or chips, according to particular needs. The software can cause the core (2140) and specifically the processors therein (including CPU, GPU, FPGA, and the like) to execute particular processes or particular parts of particular processes described herein, including defining data structures stored in RAM (2146) and modifying such data structures according to the processes defined by the software. In addition or as an alternative, the computer system can provide functionality as a result of logic hardwired or otherwise embodied in a circuit (for example: accelerator (2144)), which
can operate in place of or together with software to execute particular processes or particular parts of particular processes described herein. Reference to software can encompass logic, and vice versa, where appropriate. Reference to a computer-readable media can encompass a circuit (such as an integrated circuit (IC)) storing software for execution, a circuit embodying logic for execution, or both, where appropriate. The present disclosure encompasses any suitable combination of hardware and software.
Appendix A: Acronyms JEM: joint exploration model VVC: versatile video coding BMS: benchmark set MV: Motion Vector HEVC: High Efficiency Video Coding SEI: Supplementary Enhancement Information VUI: Video Usability Information GOPs: Groups of Pictures TUs: Transform Units,
PUs: Prediction Units
CTUs: Coding Tree Units
CTBs: Coding Tree Blocks
PBs: Prediction Blocks
HRD: Hypothetical Reference Decoder
SNR: Signal Noise Ratio
CPUs: Central Processing Units
GPUs: Graphics Processing Units
CRT: Cathode Ray Tube
LCD: Liquid-Crystal Display
OLED: Organic Light-Emitting Diode
CD: Compact Disc
DVD: Digital Video Disc
ROM: Read-Only Memory
RAM: Random Access Memory
ASIC: Application-Specific Integrated Circuit
PLD: Programmable Logic Device
LAN: Local Area Network
GSM: Global System for Mobile communications
LTE: Long-Term Evolution
CANBus: Controller Area Network Bus
USB: Universal Serial Bus
PCI: Peripheral Component Interconnect
FPGA: Field Programmable Gate Areas
SSD: solid-state drive
IC: Integrated Circuit
CU: Coding Unit
NIC: Neural Image Compression
R-D: Rate-Distortion
E2E: End to End
ANN: Artificial Neural Network
DNN: Deep Neural Network
CNN: Convolution Neural Network
[0274] While this disclosure has described several exemplary embodiments, there are alterations, permutations, and various substitute equivalents, which fall within the scope of the disclosure. It will thus be appreciated that those skilled in the art will be able to devise numerous systems and methods which, although not explicitly shown or described herein, embody the principles of the disclosure and are thus within the spirit and scope thereof.
Claims
1. A method for video decoding in a video decoder, comprising: decoding first neural network update information in a coded bitstream for a first neural network in the video decoder, the first neural network being configured with first pretrained parameters, the first neural network update information corresponding to a first block in an image to be reconstructed and indicating a first replacement parameter corresponding to a first pretrained parameter in the first pretrained parameters; updating the first neural network in the video decoder based on the first replacement parameter; and decoding the first block based on the updated first neural network for the first block.
2. The method of claim 1, wherein the first neural network update information further indicates one or more replacement parameters for one or more remaining neural networks in the video decoder, and the updating further includes updating the one or more remaining neural networks based on the one or more replacement parameters.
3. The method of claim 1, further comprising: decoding second neural network update information in the coded bitstream for a second neural network in the video decoder, the second neural network being configured with second pretrained parameters, the second neural network update information corresponding to a second block in the image to be reconstructed and indicating a second replacement parameter corresponding to a second pretrained parameter in the second pretrained parameters, the second neural network being different from the first neural network; updating the second neural network in the video decoder based on the second replacement parameter; and decoding the second block based on the updated second neural network for the second block.
4. The method of claim 3, wherein
the first pretrained parameter is a one of a pretrained weight coefficient and a pretrained bias term.
5. The method of claim 4, wherein the second pretrained parameter is another one of the pretrained weight coefficient and the pretrained bias term.
6. The method of claim 1, further comprising: decoding a second block in the coded bitstream based on the updated first neural network for the first block.
7. The method of claim 1, wherein the first neural network update information indicates a difference between the first replacement parameter and the first pretrained parameter, and the method further includes determining the first replacement parameter according to a sum of the difference and the first pretrained parameter.
8. The method of claim 3, wherein the decoding the first neural network update information includes decoding the first neural network update information based on one of a variation of a Lempel-Ziv- Markov chain algorithm (LZMA2) and a bzip2 algorithm.
9. The method of claim 8, wherein the decoding the second neural network update information includes decoding the second neural network update information based on another one of the LZMA2 and the bzip2 algorithm.
10. An apparatus for video decoding, comprising processing circuitry configured to: decode first neural network update information in a coded bitstream for a first neural network in a video decoder, the first neural network being configured with first pretrained parameters, the first neural network update information corresponding to a first block in an image to be reconstructed and indicating a first replacement parameter corresponding to a first pretrained parameter in the first pretrained parameters;
update the first neural network in the video decoder based on the first replacement parameter; and decode the first block based on the updated first neural network for the first block.
11. The apparatus of claim 10, wherein the first neural network update information further indicates one or more replacement parameters for one or more remaining neural networks in the video decoder, and the processing circuitry is configured to update the one or more remaining neural networks based on the one or more replacement parameters.
12. The apparatus of claim 10, wherein the processing circuitry is configured to: decode second neural network update information in the coded bitstream for a second neural network in the video decoder, the second neural network being configured with second pretrained parameters, the second neural network update information corresponding to a second block in the image to be reconstructed and indicating a second replacement parameter corresponding to a second pretrained parameter in the second pretrained parameters, the second neural network being different from the first neural network; update the second neural network in the video decoder based on the second replacement parameter; and decode the second block based on the updated second neural network for the second block.
13. The apparatus of claim 12, wherein the first pretrained parameter is a one of a pretrained weight coefficient and a pretrained bias term.
14. The apparatus of claim 13, wherein the second pretrained parameter is another one of the pretrained weight coefficient and the pretrained bias term.
15. The apparatus of claim 10, wherein the processing circuitry is configured to:
decode a second block in the coded bitstream based on the updated first neural network for the first block.
16. The apparatus of claim 10, wherein the first neural network update information indicates a difference between the first replacement parameter and the first pretrained parameter, and the processing circuitry is configured to determine the first replacement parameter according to a sum of the difference and the first pretrained parameter.
17. The apparatus of claim 12, wherein the processing circuitry is configured to: decode the first neural network update information based on one of a variation of a
Lempel-Ziv-Markov chain algorithm (LZMA2) and a bzip2 algorithm.
18. The apparatus of claim 17, wherein the processing circuitry is configured to: decode the second neural network update information based on another one of the
LZMA2 and the bzip2 algorithm.
19. A non-transitory computer-readable storage medium storing a program executable by at least one processor to perform: decoding first neural network update information in a coded bitstream for a first neural network in a video decoder, the first neural network being configured with first pretrained parameters, the first neural network update information corresponding to a first block in an image to be reconstructed and indicating a first replacement parameter corresponding to a first pretrained parameter in the first pretrained parameters; updating the first neural network in the video decoder based on the first replacement parameter; and decoding the first block based on the updated first neural network for the first block.
20. The non-transitory computer-readable storage medium of claim 19, wherein the first neural network update information further indicates one or more replacement parameters for one or more remaining neural networks in the video decoder, and
the program executable by the at least one processor performs updating the one or more remaining neural networks based on the one or more replacement parameters.
Priority Applications (4)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
KR1020227038660A KR20220163472A (en) | 2021-04-30 | 2022-04-29 | Method and apparatus for video decoding, and storage medium |
JP2022566411A JP7447311B2 (en) | 2021-04-30 | 2022-04-29 | Methods, devices and computer programs for video decoding |
CN202280003933.9A CN115552912A (en) | 2021-04-30 | 2022-04-29 | Block content adaptive online training in neural image compression |
EP22773386.2A EP4111695A4 (en) | 2021-04-30 | 2022-04-29 | Block-wise content-adaptive online training in neural image compression |
Applications Claiming Priority (4)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
US202163182366P | 2021-04-30 | 2021-04-30 | |
US63/182,366 | 2021-04-30 | ||
US17/729,978 US11889112B2 (en) | 2021-04-30 | 2022-04-26 | Block-wise content-adaptive online training in neural image compression |
US17/729,978 | 2022-04-26 |
Publications (1)
Publication Number | Publication Date |
---|---|
WO2022232841A1 true WO2022232841A1 (en) | 2022-11-03 |
Family
ID=83598651
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
PCT/US2022/072022 WO2022232841A1 (en) | 2021-04-30 | 2022-04-29 | Block-wise content-adaptive online training in neural image compression |
Country Status (6)
Country | Link |
---|---|
US (1) | US11889112B2 (en) |
EP (1) | EP4111695A4 (en) |
JP (1) | JP7447311B2 (en) |
KR (1) | KR20220163472A (en) |
CN (1) | CN115552912A (en) |
WO (1) | WO2022232841A1 (en) |
Families Citing this family (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
WO2024146645A1 (en) * | 2023-01-07 | 2024-07-11 | Douyin Vision Co., Ltd. | Method, apparatus, and medium for video processing |
Citations (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20200036986A1 (en) * | 2017-03-15 | 2020-01-30 | Realnetworks, Inc. | Condensed coding block headers in video coding systems and methods |
WO2020165493A1 (en) | 2019-02-15 | 2020-08-20 | Nokia Technologies Oy | An apparatus, a method and a computer program for video coding and decoding |
Family Cites Families (7)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
EP3310058B1 (en) * | 2015-06-12 | 2023-02-22 | Panasonic Intellectual Property Management Co., Ltd. | Image coding method, image decoding method, image coding device and image decoding device |
WO2019157228A1 (en) * | 2018-02-09 | 2019-08-15 | D-Wave Systems Inc. | Systems and methods for training generative machine learning models |
US20200160565A1 (en) * | 2018-11-19 | 2020-05-21 | Zhan Ma | Methods And Apparatuses For Learned Image Compression |
US11379978B2 (en) * | 2020-07-14 | 2022-07-05 | Canon Medical Systems Corporation | Model training apparatus and method |
US20220261631A1 (en) * | 2021-02-12 | 2022-08-18 | Nvidia Corporation | Pipelines for efficient training and deployment of machine learning models |
WO2022182265A1 (en) * | 2021-02-25 | 2022-09-01 | Huawei Technologies Co., Ltd | Apparatus and method for coding pictures using a convolutional neural network |
US11978258B2 (en) * | 2021-04-06 | 2024-05-07 | Nvidia Corporation | Techniques for identification of out-of-distribution input data in neural networks |
-
2022
- 2022-04-26 US US17/729,978 patent/US11889112B2/en active Active
- 2022-04-29 WO PCT/US2022/072022 patent/WO2022232841A1/en unknown
- 2022-04-29 CN CN202280003933.9A patent/CN115552912A/en active Pending
- 2022-04-29 EP EP22773386.2A patent/EP4111695A4/en active Pending
- 2022-04-29 KR KR1020227038660A patent/KR20220163472A/en active Search and Examination
- 2022-04-29 JP JP2022566411A patent/JP7447311B2/en active Active
Patent Citations (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20200036986A1 (en) * | 2017-03-15 | 2020-01-30 | Realnetworks, Inc. | Condensed coding block headers in video coding systems and methods |
WO2020165493A1 (en) | 2019-02-15 | 2020-08-20 | Nokia Technologies Oy | An apparatus, a method and a computer program for video coding and decoding |
Non-Patent Citations (3)
Title |
---|
"High Efficiency Video Coding", ITU-T REC. H.265, December 2016 (2016-12-01) |
LAM ET AL., COMPRESSING WEIGHT-UPDATES FOR IMAGE ARTIFACTS REMOVAL NEURAL NETWORKS |
See also references of EP4111695A4 |
Also Published As
Publication number | Publication date |
---|---|
US11889112B2 (en) | 2024-01-30 |
KR20220163472A (en) | 2022-12-09 |
EP4111695A1 (en) | 2023-01-04 |
EP4111695A4 (en) | 2023-05-24 |
JP7447311B2 (en) | 2024-03-11 |
CN115552912A (en) | 2022-12-30 |
US20220353528A1 (en) | 2022-11-03 |
JP2023527672A (en) | 2023-06-30 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
US11849118B2 (en) | Content-adaptive online training with image substitution in neural image compression | |
US11979565B2 (en) | Content-adaptive online training method and apparatus for post-filtering | |
US20220405979A1 (en) | Content-adaptive online training method and apparatus for deblocking in block-wise image compression | |
US11758168B2 (en) | Content-adaptive online training with scaling factors and/or offsets in neural image compression | |
US11889112B2 (en) | Block-wise content-adaptive online training in neural image compression | |
US11917162B2 (en) | Content-adaptive online training with feature substitution in neural image compression | |
US20220353521A1 (en) | Method and apparatus for content-adaptive online training in neural image compression | |
US20220360770A1 (en) | Block-wise content-adaptive online training in neural image compression with post filtering |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
ENP | Entry into the national phase |
Ref document number: 2022773386 Country of ref document: EP Effective date: 20220929 |
|
ENP | Entry into the national phase |
Ref document number: 2022566411 Country of ref document: JP Kind code of ref document: A |
|
ENP | Entry into the national phase |
Ref document number: 20227038660 Country of ref document: KR Kind code of ref document: A |
|
NENP | Non-entry into the national phase |
Ref country code: DE |