@article{agueh2011barycenters,
  title={Barycenters in the Wasserstein space},
  author={Agueh, Martial and Carlier, Guillaume},
  journal={SIAM Journal on Mathematical Analysis},
  volume={43},
  number={2},
  pages={904--924},
  year={2011},
  publisher={SIAM}
}

@article{hua2023dynamic,
  title={Dynamic Flows on Curved Space Generated by Labeled Data},
  author={Hua, Xinru and Nguyen, Truyen and Le, Tam and Blanchet, Jose and Nguyen, Viet Anh},
  journal={arXiv preprint arXiv:2302.00061},
  year={2023}
}

@article{yao2021meta,
  title={Meta-learning with fewer tasks through task interpolation},
  author={Yao, Huaxiu and Zhang, Linjun and Finn, Chelsea},
  journal={arXiv preprint arXiv:2106.02695},
  year={2021}
}

@article{alvarez2016fixed,
  title={A fixed-point approach to barycenters in Wasserstein space},
  author={{\'A}lvarez-Esteban, Pedro C and Del Barrio, E and Cuesta-Albertos, JA and Matr{\'a}n, C},
  journal={Journal of Mathematical Analysis and Applications},
  volume={441},
  number={2},
  pages={744--762},
  year={2016},
  publisher={Elsevier}
}

@article{peyre2019computational,
  title={Computational optimal transport: With applications to data science},
  author={Peyr{\'e}, Gabriel and Cuturi, Marco and others},
  journal={Foundations and Trends in Machine Learning},
  volume={11},
  number={5-6},
  pages={355--607},
  year={2019},
  publisher={Now Publishers, Inc.}
}

@book{Vil03,
  title     = {Topics in optimal transportation},
  author    = {Villani, C{\'e}dric},
  number    = {58},
  year      = {2003},
  publisher = {American Mathematical Soc.}
}


@book{ambrosio2008gradient,
  title     = {Gradient flows: in metric spaces and in the space of probability measures},
  author    = {Ambrosio, Luigi and Gigli, Nicola and Savar{\'e}, Giuseppe},
  year      = {2008},
  publisher = {Springer Science \& Business Media}
}

@article{brenier1991polar,
  title={Polar factorization and monotone rearrangement of vector-valued functions},
  author={Brenier, Yann},
  journal={Communications on pure and applied mathematics},
  volume={44},
  number={4},
  pages={375--417},
  year={1991},
  publisher={Wiley Online Library}
}

@article{mccann1997convexity,
  title={A convexity principle for interacting gases},
  author={McCann, Robert J},
  journal={Advances in mathematics},
  volume={128},
  number={1},
  pages={153--179},
  year={1997},
  publisher={Elsevier}
}


@article{santambrogio2017euclidean,
  title={$\{$Euclidean, metric, and Wasserstein$\}$ gradient flows: an overview},
  author={Santambrogio, Filippo},
  journal={Bulletin of Mathematical Sciences},
  volume={7},
  number={1},
  pages={87--154},
  year={2017},
  publisher={Springer}
}

@article{benamou2000computational,
  title={A computational fluid mechanics solution to the Monge-Kantorovich mass transfer problem},
  author={Benamou, Jean-David and Brenier, Yann},
  journal={Numerische Mathematik},
  volume={84},
  number={3},
  pages={375--393},
  year={2000},
  publisher={Springer-Verlag}
}

@article{santambrogio2015optimal,
  title={Optimal transport for applied mathematicians},
  author={Santambrogio, Filippo},
  journal={Birk{\"a}user, NY},
  volume={55},
  number={58-63},
  pages={94},
  year={2015},
  publisher={Springer}
}

@article{alvarez2020geometric,
  title={Geometric dataset distances via optimal transport},
  author={Alvarez-Melis, David and Fusi, Nicolo},
  journal={Advances in Neural Information Processing Systems},
  volume={33},
  pages={21428--21439},
  year={2020}
}

@inproceedings{alvarez2021dataset,
  title={Dataset dynamics via gradient flows in probability space},
  author={Alvarez-Melis, David and Fusi, Nicol{\`o}},
  booktitle={International Conference on Machine Learning},
  pages={219--230},
  year={2021},
  organization={PMLR}
}

@article{craig2016exponential,
  title={The exponential formula for the Wasserstein metric},
  author={Craig, Katy},
  journal={ESAIM: Control, Optimisation and Calculus of Variations},
  volume={22},
  number={1},
  pages={169--187},
  year={2016},
  publisher={EDP Sciences}
}

@article{srivastava2018scalable,
  title={Scalable Bayes via barycenter in Wasserstein space},
  author={Srivastava, Sanvesh and Li, Cheng and Dunson, David B},
  journal={The Journal of Machine Learning Research},
  volume={19},
  number={1},
  pages={312--346},
  year={2018},
  publisher={JMLR. org}
}

@article{korotin2021continuous,
  title={Continuous wasserstein-2 barycenter estimation without minimax optimization},
  author={Korotin, Alexander and Li, Lingxiao and Solomon, Justin and Burnaev, Evgeny},
  journal={arXiv preprint arXiv:2102.01752},
  year={2021}
}

@article{fan2020scalable,
  title={Scalable computations of wasserstein barycenter via input convex neural networks},
  author={Fan, Jiaojiao and Taghvaei, Amirhossein and Chen, Yongxin},
  journal={arXiv preprint arXiv:2007.04462},
  year={2020}
}

@article{korotin2022wasserstein,
  title={Wasserstein iterative networks for barycenter estimation},
  author={Korotin, Alexander and Egiazarian, Vage and Li, Lingxiao and Burnaev, Evgeny},
  journal={arXiv preprint arXiv:2201.12245},
  year={2022}
}

@article{fan2021scalable,
  title={Neural Monge Map estimation and its applications},
  author={Fan, Jiaojiao and Liu, Shu and Ma, Shaojun and Zhou, Haomin and Chen, Yongxin},
  journal={arXiv preprint arXiv:2106.03812},
  year={2021}
}

@article{zhu2023interpolation,
  title={Interpolation for Robust Learning: Data Augmentation on Geodesics},
  author={Zhu, Jiacheng and Qiu, Jielin and Guha, Aritra and Yang, Zhuolin and Nguyen, Xuanlong and Li, Bo and Zhao, Ding},
  journal={arXiv preprint arXiv:2302.02092},
  year={2023}
}

@article{korotin2022neural,
  title={Neural Optimal Transport},
  author={Korotin, Alexander and Selikhanovych, Daniil and Burnaev, Evgeny},
  journal={arXiv preprint arXiv:2201.12220},
  year={2022}
}

@inproceedings{rout2022generative,
    title={Generative Modeling with Optimal Transport Maps},
    author={Litu Rout and Alexander Korotin and Evgeny Burnaev},
    booktitle={International Conference on Learning Representations},
    year={2022},
}


@INPROCEEDINGS{chuang2021fair,
  title     = "Fair Mixup: Fairness via Interpolation",
  booktitle = "International Conference on Learning Representations",
  author    = "Chuang, Ching-Yao and Mroueh, Youssef",
  year      =  2021
}

@INPROCEEDINGS{bunne2022proximal,
  title     = "Proximal Optimal Transport Modeling of Population Dynamics",
  booktitle = "Proceedings of The 25th International Conference on Artificial Intelligence and Statistics",
  author    = "Bunne, Charlotte and Papaxanthos, Laetitia and Krause, Andreas and Cuturi, Marco",
  editor    = "Camps-Valls, Gustau and Ruiz, Francisco J R and Valera, Isabel",
  publisher = "PMLR",
  volume    =  151,
  pages     = "6511--6528",
  series    = "Proceedings of Machine Learning Research",
  year      =  2022
}

@inproceedings{he2022masked,
  title={Masked autoencoders are scalable vision learners},
  author={He, Kaiming and Chen, Xinlei and Xie, Saining and Li, Yanghao and Doll{\'a}r, Piotr and Girshick, Ross},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={16000--16009},
  year={2022}
}

@article{zhai2019large,
  title={A large-scale study of representation learning with the visual task adaptation benchmark},
  author={Zhai, Xiaohua and Puigcerver, Joan and Kolesnikov, Alexander and Ruyssen, Pierre and Riquelme, Carlos and Lucic, Mario and Djolonga, Josip and Pinto, Andre Susano and Neumann, Maxim and Dosovitskiy, Alexey and others},
  journal={arXiv preprint arXiv:1910.04867},
  year={2019}
}

@INPROCEEDINGS{mokrov2021large-scale,
  title     = "{Large-Scale} Wasserstein Gradient Flows",
  booktitle = "Advances in Neural Information Processing Systems",
  author    = "Mokrov, Petr and Korotin, Alexander and Li, Lingxiao and
              Genevay, Aude and Solomon, Justin M and Burnaev, Evgeny",
  editor    = "Ranzato, M and Beygelzimer, A and Dauphin, Y and Liang, P S and
              Vaughan, J Wortman",
  publisher = "Curran Associates, Inc.",
  volume    =  34,
  pages     = "15243--15256",
  year      =  2021
}

@INPROCEEDINGS{fan2022variational,
  title     = "Variational {W}asserstein gradient flow",
  booktitle = "Proceedings of the 39th International Conference on Machine
              Learning",
  author    = "Fan, Jiaojiao and Zhang, Qinsheng and Taghvaei, Amirhossein and
              Chen, Yongxin",
  editor    = "Chaudhuri, Kamalika and Jegelka, Stefanie and Song, Le and
              Szepesvari, Csaba and Niu, Gang and Sabato, Sivan",
  publisher = "PMLR",
  volume    =  162,
  pages     = "6185--6215",
  series    = "Proceedings of Machine Learning Research",
  year      =  2022
}


@INPROCEEDINGS{yoon2019pate-gan,
  title     = "{{PATE}-{GAN}}: Generating Synthetic Data with Differential Privacy Guarantees",
  booktitle = "International Conference on Learning Representations",
  author    = "Yoon, Jinsung and Jordon, James and van der Schaar, Mihaela",
  year      =  2019
}

@ARTICLE{sandfort2019data,
  title   = "Data augmentation using generative adversarial networks
             ({CycleGAN}) to improve generalizability in {CT} segmentation
             tasks",
  author  = "Sandfort, Veit and Yan, Ke and Pickhardt, Perry J and Summers,
             Ronald M",
  journal = "Sci. Rep.",
  volume  =  9,
  number  =  1,
  pages   = "16884",
  month   =  nov,
  year    =  2019,
  issn    = "2045-2322",
  doi     = "10.1038/s41598-019-52737-x"
}

@misc{bowles2018GAN,
  title         = "{GAN} Augmentation: Augmenting Training Data using Generative Adversarial Networks",
  author        = "Bowles, Christopher and Chen, Liang and Guerrero, Ricardo and Bentley, Paul and Gunn, Roger and Hammers, Alexander and Dickie, David Alexander and Hernández, Maria Valdés and Wardlaw, Joanna and Rueckert, Daniel",
  month         =  oct,
  year          =  2018,
  archivePrefix = "arXiv",
  eprint        = "1810.10863",
  primaryClass  = "cs.CV",
  arxivid       = "1810.10863"
}

@inproceedings{dvurechensky2018computational,
  title={Computational optimal transport: Complexity by accelerated gradient descent is better than by Sinkhorn’s algorithm},
  author={Dvurechensky, Pavel and Gasnikov, Alexander and Kroshnin, Alexey},
  booktitle={International conference on machine learning},
  pages={1367--1376},
  year={2018},
  organization={PMLR}
}

@article{mccann1995existence,
  title={Existence and uniqueness of monotone measure-preserving maps},
  author={McCann, Robert J},
  journal={Duke Mathematical Journal},
  volume={80},
  number={2},
  pages={309--323},
  year={1995},
  publisher={Duke University Press}
}

@inproceedings{liu2019wasserstein,
  title={Wasserstein gan with quadratic transport cost},
  author={Liu, Huidong and Gu, Xianfeng and Samaras, Dimitris},
  booktitle={Proceedings of the IEEE/CVF international conference on computer vision},
  pages={4832--4841},
  year={2019}
}


@inproceedings{ronneberger2015u,
  title={U-net: Convolutional networks for biomedical image segmentation},
  author={Ronneberger, Olaf and Fischer, Philipp and Brox, Thomas},
  booktitle={International Conference on Medical image computing and computer-assisted intervention},
  pages={234--241},
  year={2015},
  organization={Springer}
}


@article{asadulaev2022neural,
  title={Neural Optimal Transport with General Cost Functionals},
  author={Asadulaev, Arip and Korotin, Alexander and Egiazarian, Vage and Burnaev, Evgeny},
  journal={arXiv preprint arXiv:2205.15403},
  year={2022}
}

@article{bunne2022supervised,
  title={Supervised Training of Conditional Monge Maps},
  author={Bunne, Charlotte and Krause, Andreas and Cuturi, Marco},
  journal={arXiv preprint arXiv:2206.14262},
  year={2022}
}

@article{kabir2022spinalnet,
  title={Spinalnet: Deep neural network with gradual input},
  author={Kabir, HM Dipu and Abdar, Moloud and Khosravi, Abbas and Jalali, Seyed Mohammad Jafar and Atiya, Amir F and Nahavandi, Saeid and Srinivasan, Dipti},
  journal={IEEE Transactions on Artificial Intelligence},
  year={2022},
  publisher={IEEE}
}

@article{perrot2016mapping,
  title={Mapping estimation for discrete optimal transport},
  author={Perrot, Micha{\"e}l and Courty, Nicolas and Flamary, R{\'e}mi and Habrard, Amaury},
  journal={Advances in Neural Information Processing Systems},
  volume={29},
  year={2016}
}

@article{pooladian2021entropic,
  title={Entropic estimation of optimal transport maps},
  author={Pooladian, Aram-Alexandre and Niles-Weed, Jonathan},
  journal={arXiv preprint arXiv:2109.12004},
  year={2021}
}

@article{sinkhorn1967diagonal,
  title={Diagonal equivalence to matrices with prescribed row and column sums},
  author={Sinkhorn, Richard},
  journal={The American Mathematical Monthly},
  volume={74},
  number={4},
  pages={402--405},
  year={1967},
  publisher={JSTOR}
}

@article{gazdieva2022unpaired,
  title={Unpaired image super-resolution with optimal transport maps},
  author={Gazdieva, Milena and Rout, Litu and Korotin, Alexander and Filippov, Alexander and Burnaev, Evgeny},
  journal={arXiv preprint arXiv:2202.01116},
  year={2022}
}

@inproceedings{cohen2017emnist,
  title={EMNIST: Extending MNIST to handwritten letters},
  author={Cohen, Gregory and Afshar, Saeed and Tapson, Jonathan and Van Schaik, Andre},
  booktitle={2017 international joint conference on neural networks (IJCNN)},
  pages={2921--2926},
  year={2017},
  organization={IEEE}
}

@article{hull1994database,
  title={A database for handwritten text recognition research},
  author={Hull, Jonathan J.},
  journal={IEEE Transactions on pattern analysis and machine intelligence},
  volume={16},
  number={5},
  pages={550--554},
  year={1994},
  publisher={IEEE}
}

@article{jain2022data,
  title={A Data-Based Perspective on Transfer Learning},
  author={Jain, Saachi and Salman, Hadi and Khaddaj, Alaa and Wong, Eric and Park, Sung Min and Madry, Aleksander},
  journal={arXiv preprint arXiv:2207.05739},
  year={2022}
}

@INPROCEEDINGS{brown2020language,
  title     = "Language Models are {Few-Shot} Learners",
  booktitle = "Advances in Neural Information Processing Systems",
  author    = "Brown, Tom and Mann, Benjamin and Ryder, Nick and Subbiah, Melanie and Kaplan, Jared D and Dhariwal, Prafulla and Neelakantan, Arvind and Shyam, Pranav and Sastry, Girish and Askell, Amanda and Agarwal, Sandhini and Herbert-Voss, Ariel and Krueger, Gretchen and Henighan, Tom and Child, Rewon and Ramesh, Aditya and Ziegler, Daniel and Wu, Jeffrey and Winter, Clemens and Hesse, Chris and Chen, Mark and Sigler, Eric and Litwin, Mateusz and Gray, Scott and Chess, Benjamin and Clark, Jack and Berner, Christopher and McCandlish, Sam and Radford, Alec and Sutskever, Ilya and Amodei, Dario",
  editor    = "Larochelle, H and Ranzato, M and Hadsell, R and Balcan, M F and Lin, H",
  publisher = "Curran Associates, Inc.",
  volume    =  33,
  pages     = "1877--1901",
  year      =  2020
}

@BOOK{villani2008optimal,
  title     = "Optimal transport, Old and New",
  author    = "Villani, Cédric",
  publisher = "Springer Science \& Business Media",
  volume    =  338,
  year      =  2008,
  isbn      = "9783540710493"
}


@inproceedings{makkuva2020optimal,
  title={Optimal transport mapping via input convex neural networks}, 
  author={Makkuva, Ashok and Taghvaei, Amirhossein and Oh, Sewoong and Lee, Jason},
  booktitle={International Conference on Machine Learning}, 
  year={2020},
  volume={37},
}


@INPROCEEDINGS{zhang2018mixup,
  title     = "mixup: Beyond Empirical Risk Minimization",
  booktitle = "International Conference on Learning Representations",
  author    = "Zhang, Hongyi and Cisse, Moustapha and Dauphin, Yann N and Lopez-Paz, David",
  year      =  2018
}

@INCOLLECTION{mccloskey1989catastrophic,
  title     = {Catastrophic Interference in Connectionist Networks: The Sequential Learning Problem},
  booktitle = {Psychology of Learning and Motivation},
  author    = {McCloskey, Michael and Cohen, Neal J},
  editor    = "Bower, Gordon H",
  publisher = "Academic Press",
  volume    =  24,
  pages     = "109--165",
  month     =  jan,
  year      =  1989,
  doi       = "10.1016/S0079-7421(08)60536-8"
}

@INPROCEEDINGS{zhang2021how,
  title     = "How Does Mixup Help With Robustness and Generalization?",
  booktitle = "International Conference on Learning Representations",
  author    = "Zhang, Linjun and Deng, Zhun and Kawaguchi, Kenji and Ghorbani, Amirata and Zou, James",
  year      =  2021
}

@article{kirkpatrick2017overcoming,
  title    = {Overcoming catastrophic forgetting in neural networks},
  author   = {Kirkpatrick, James and Pascanu, Razvan and Rabinowitz, Neil and Veness, Joel and Desjardins, Guillaume and Rusu, Andrei A and  Milan, Kieran and Quan, John and Ramalho, Tiago and Grabska-Barwinska, Agnieszka and Hassabis, Demis and Clopath, Claudia and Kumaran, Dharshan and Hadsell, Raia},
  journal  = {PNAS},
  volume   = {114},
  year = {2017},
  number   =  {13},
  pages    = {3521--3526},
  month    =  {mar},  
}


@INPROCEEDINGS{gao2021information,
  title      = "An {Information-Geometric} Distance on the Space of Tasks",
  booktitle  = "Proceedings of the 38th International Conference on Machine
                Learning",
  author     = "Gao, Yansong and Chaudhari, Pratik",
  publisher  = "PMLR",
  year       =  2021,
  conference = "International Conference on Machine Learning"
}

@article{yeaton2022hierarchical,
  title={Hierarchical Optimal Transport for Comparing Histopathology Datasets},
  author={Yeaton, Anna and Krishnan, Rahul G and Mieloszyk, Rebecca and Alvarez-Melis, David and Huynh, Grace},
  journal={arXiv preprint arXiv:2204.08324},
  year={2022}
}