Data driven conditional optimal transport
作者:Esteban G. Tabak, Giulio Trigila, Wenjun Zhao
摘要
A data-driven procedure is developed to compute the optimal map between two conditional probabilities \(\rho (x|z_{1},\ldots ,z_{L})\) and \(\mu (y|z_{1},\ldots ,z_{L})\), known only through samples and depending on a set of covariates \(z_{l}\). The procedure is tested on synthetic data from the ACIC Data Analysis Challenge 2017 and it is applied to non-uniform lightness transfer between images. Exactly solvable examples and simulations are performed to highlight the differences with ordinary optimal transport.
论文关键词:Optimal transport, Conditional average treatment effect, Uncertainty quantification, Color transfer, Image restoration
论文评审过程:
论文官网地址:https://doi.org/10.1007/s10994-021-06060-0