Domain Adaptation for Learned Image Compression with Supervised Adapters

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Presta, Alberto, Spadaro, Gabriele, Tartaglione, Enzo, Fiandrotti, Attilio, Grangetto, Marco
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917648598564864
author Presta, Alberto
Spadaro, Gabriele
Tartaglione, Enzo
Fiandrotti, Attilio
Grangetto, Marco
author_facet Presta, Alberto
Spadaro, Gabriele
Tartaglione, Enzo
Fiandrotti, Attilio
Grangetto, Marco
contents In Learned Image Compression (LIC), a model is trained at encoding and decoding images sampled from a source domain, often outperforming traditional codecs on natural images; yet its performance may be far from optimal on images sampled from different domains. In this work, we tackle the problem of adapting a pre-trained model to multiple target domains by plugging into the decoder an adapter module for each of them, including the source one. Each adapter improves the decoder performance on a specific domain, without the model forgetting about the images seen at training time. A gate network computes the weights to optimally blend the contributions from the adapters when the bitstream is decoded. We experimentally validate our method over two state-of-the-art pre-trained models, observing improved rate-distortion efficiency on the target domains without penalties on the source domain. Furthermore, the gate's ability to find similarities with the learned target domains enables better encoding efficiency also for images outside them.
format Preprint
id arxiv_https___arxiv_org_abs_2404_15591
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Domain Adaptation for Learned Image Compression with Supervised Adapters
Presta, Alberto
Spadaro, Gabriele
Tartaglione, Enzo
Fiandrotti, Attilio
Grangetto, Marco
Computer Vision and Pattern Recognition
Image and Video Processing
In Learned Image Compression (LIC), a model is trained at encoding and decoding images sampled from a source domain, often outperforming traditional codecs on natural images; yet its performance may be far from optimal on images sampled from different domains. In this work, we tackle the problem of adapting a pre-trained model to multiple target domains by plugging into the decoder an adapter module for each of them, including the source one. Each adapter improves the decoder performance on a specific domain, without the model forgetting about the images seen at training time. A gate network computes the weights to optimally blend the contributions from the adapters when the bitstream is decoded. We experimentally validate our method over two state-of-the-art pre-trained models, observing improved rate-distortion efficiency on the target domains without penalties on the source domain. Furthermore, the gate's ability to find similarities with the learned target domains enables better encoding efficiency also for images outside them.
title Domain Adaptation for Learned Image Compression with Supervised Adapters
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2404.15591