中文

InvertibleNetworks.jl:一个用于可扩展归一化流的 Julia 包

机器学习 2023-12-22 v1

摘要

InvertibleNetworks.jl 是一个专为归一化流的可扩展实现而设计的 Julia 包,归一化流是一种用于高维分布密度估计和采样的方法。该包通过利用归一化流固有的可逆性,在内存效率方面表现出色,与依赖自动微分框架的现有归一化流包相比,显著减少了反向传播过程中的内存需求。InvertibleNetworks.jl 已适配于多种应用,包括地震成像、医学成像和 CO2 监测,证明了其在学习高维分布方面的有效性。

关键词

引用

@article{arxiv.2312.13480,
  title  = {InvertibleNetworks.jl: A Julia package for scalable normalizing flows},
  author = {Rafael Orozco and Philipp Witte and Mathias Louboutin and Ali Siahkoohi and Gabrio Rizzuti and Bas Peters and Felix J. Herrmann},
  journal= {arXiv preprint arXiv:2312.13480},
  year   = {2023}
}

备注

Submitted to Journal of Open Source Software (JOSS)