Domain-Scalable Unpaired Image Translation via Latent Space Anchoring

Domain-Scalable Unpaired Image Translation via Latent Space Anchoring

Huang S, An J, Wei D, Lin Z, Luo J, and Pfister H.

IEEE Transactions on Pattern Analysis and Machine Intelligence, 2023.

Unpaired image-to-image translation (UNIT) aims to map images between two visual domains without paired training data. However, given a UNIT model trained on certain domains, it is difficult for current methods to incorporate new domains because they often need to train the full model on both existing and new domains. To address this problem, we propose a new domain-scalable UNIT method, termed as latent space anchoring, which can be efficiently extended to new visual domains and does not need to fine-tune encoders and decoders of existing domains. Our method anchors images of different domains to the same latent space of frozen GANs by learning lightweight encoder and regressor models to reconstruct single-domain images. In the inference phase, the learned encoders and decoders of different domains can be arbitrarily combined to translate images between any two domains without fine-tuning. Experiments on various datasets show that the proposed method achieves superior performance on both standard and domain-scalable UNIT tasks in comparison with the state-of-the-art methods.

Acknowledgements

This work was supported by NIH grants R01HD104969 and 5U54CA225088-03.