Universal Multi-Domain Translation via Diffusion Routers

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Kieu, Duc, Do, Kien, Hoang, Tuan, Le, Thao Minh, Kieu, Tung, Nguyen, Dang, Nguyen, Thin
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918307072835584
author Kieu, Duc
Do, Kien
Hoang, Tuan
Le, Thao Minh
Kieu, Tung
Nguyen, Dang
Nguyen, Thin
author_facet Kieu, Duc
Do, Kien
Hoang, Tuan
Le, Thao Minh
Kieu, Tung
Nguyen, Dang
Nguyen, Thin
contents Multi-domain translation (MDT) aims to learn translations between multiple domains, yet existing approaches either require fully aligned tuples or can only handle domain pairs seen in training, limiting their practicality and excluding many cross-domain mappings. We introduce universal MDT (UMDT), a generalization of MDT that seeks to translate between any pair of $K$ domains using only $K-1$ paired datasets with a central domain. To tackle this problem, we propose Diffusion Router (DR), a unified diffusion-based framework that models all central$\leftrightarrow$non-central translations with a single noise predictor conditioned on the source and target domain labels. DR enables indirect non-central translations by routing through the central domain. We further introduce a novel scalable learning strategy with a variational-bound objective and an efficient Tweedie refinement procedure to support direct non-central mappings. Through evaluation on three large-scale UMDT benchmarks, DR achieves state-of-the-art results for both indirect and direct translations, while lowering sampling cost and unlocking novel tasks such as sketch$\leftrightarrow$segmentation. These results establish DR as a scalable and versatile framework for universal translation across multiple domains.
format Preprint
id arxiv_https___arxiv_org_abs_2510_03252
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Universal Multi-Domain Translation via Diffusion Routers
Kieu, Duc
Do, Kien
Hoang, Tuan
Le, Thao Minh
Kieu, Tung
Nguyen, Dang
Nguyen, Thin
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Multi-domain translation (MDT) aims to learn translations between multiple domains, yet existing approaches either require fully aligned tuples or can only handle domain pairs seen in training, limiting their practicality and excluding many cross-domain mappings. We introduce universal MDT (UMDT), a generalization of MDT that seeks to translate between any pair of $K$ domains using only $K-1$ paired datasets with a central domain. To tackle this problem, we propose Diffusion Router (DR), a unified diffusion-based framework that models all central$\leftrightarrow$non-central translations with a single noise predictor conditioned on the source and target domain labels. DR enables indirect non-central translations by routing through the central domain. We further introduce a novel scalable learning strategy with a variational-bound objective and an efficient Tweedie refinement procedure to support direct non-central mappings. Through evaluation on three large-scale UMDT benchmarks, DR achieves state-of-the-art results for both indirect and direct translations, while lowering sampling cost and unlocking novel tasks such as sketch$\leftrightarrow$segmentation. These results establish DR as a scalable and versatile framework for universal translation across multiple domains.
title Universal Multi-Domain Translation via Diffusion Routers
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2510.03252