English
Related papers

Related papers: cuRegOT: A GPU-Accelerated Solver for Entropic-Reg…

200 papers

From logistics to the natural sciences, combinatorial optimisation on graphs underpins numerous real-world applications. Reinforcement learning (RL) has shown particular promise in this setting as it can adapt to specific problem structures…

Machine Learning · Computer Science 2022-05-30 Thomas D. Barrett , Christopher W. F. Parsonson , Alexandre Laterre

The mismatch capacity characterizes the highest information rate of the channel under a prescribed decoding metric and serves as a critical performance indicator in numerous practical communication scenarios. Compared to the commonly used…

Information Theory · Computer Science 2025-07-29 Shitong Wu , Wenhao Ye , Xinwei Li , Lingyi Chen , Wenyi Zhang , Huihui Wu , Hao Wu

We present a new multiscale algorithm for solving regularized Optimal Transport problems on the GPU, with a linear memory footprint. Relying on Sinkhorn divergences which are convex, smooth and positive definite loss functions, this method…

Computer Vision and Pattern Recognition · Computer Science 2021-07-06 Jean Feydy , Pierre Roussillon , Alain Trouvé , Pietro Gori

We propose integrating optimal transport (OT) into operator learning for partial differential equations (PDEs) on complex geometries. Classical geometric learning methods typically represent domains as meshes, graphs, or point clouds. Our…

Machine Learning · Computer Science 2025-07-29 Xinyi Li , Zongyi Li , Nikola Kovachki , Anima Anandkumar

Driven by Moore's Law, the dimensions of transistors have been pushed down to the nanometer scale. Advanced quantum transport (QT) solvers are required to accurately simulate such nano-devices. The non-equilibrium Green's function (NEGF)…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-01-09 Vincent Maillou , Matthias Bollhofer , Olaf Schenk , Alexandros Nikolaos Ziogas , Mathieu Luisier

Krylov methods provide a fast and highly parallel numerical tool for the iterative solution of many large-scale sparse linear systems. To a large extent, the performance of practical realizations of these methods is constrained by the…

Mathematical Software · Computer Science 2020-09-28 José I. Aliaga , Hartwig Anzt , Thomas Grützmacher , Enrique S. Quintana-Ortí , Andrés E. Tomás

Partial Optimal Transport (POT) has recently emerged as a central tool in various Machine Learning (ML) applications. It lifts the stringent assumption of the conventional Optimal Transport (OT) that input measures are of equal masses,…

Discrete optimization is a central problem in artificial intelligence. The optimization of the aggregated cost of a network of cost functions arises in a variety of problems including (W)CSP, DCOP, as well as optimization in stochastic…

Artificial Intelligence · Computer Science 2018-01-12 Ferdinando Fioretto , Enrico Pontelli , William Yeoh , Rina Dechter

We present the numerical methods and GPU-accelerated implementation underlying a Total Lagrangian finite element framework for finite-deformation flexible multibody dynamics, introduced in the companion paper [1]. The framework supports…

Computational Engineering, Finance, and Science · Computer Science 2026-05-04 Zhenhao Zhou , Ruochun Zhang , Ganesh Arivoli , Dan Negrut

Decentralized optimization with orthogonality constraints is found widely in scientific computing and data science. Since the orthogonality constraints are nonconvex, it is quite challenging to design efficient algorithms. Existing…

Optimization and Control · Mathematics 2024-01-09 Lei Wang , Xin Liu

Optimal transport is an important tool in machine learning, allowing to capture geometric properties of the data through a linear program on transport polytopes. We present a single-loop optimization algorithm for minimizing general convex…

Machine Learning · Computer Science 2023-06-21 Marin Ballu , Quentin Berthet

Three-dimensional neutron transport calculations using the Method of Characteristics (MOC) are highly regarded for their exceptional computational efficiency, precision, and stability. Nevertheless, when dealing with extensive-scale…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-03-25 Faguo Zhou , Shunde Li , Rong Xue , Lingkun Bu , Ningming Nie , Peng Shi , Jue Wang , Yun Hu , Zongguo Wang , Yangang Wang , Qinmeng Yang , Miao Yu

The ability to compare two degenerate probability distributions (i.e. two probability distributions supported on two distinct low-dimensional manifolds living in a much higher-dimensional space) is a crucial problem arising in the…

Machine Learning · Statistics 2017-10-23 Aude Genevay , Gabriel Peyré , Marco Cuturi

Large scale graph optimization problems arise in many fields. This paper presents an extensible, high performance framework (named OpenGraphGym-MG) that uses deep reinforcement learning and graph embedding to solve large graph optimization…

Distributed, Parallel, and Cluster Computing · Computer Science 2021-06-25 Weijian Zheng , Dali Wang , Fengguang Song

Many machine learning problems involve data supported on curved spaces such as spheres, rotation groups, hyperbolic spaces, and general Riemannian manifolds, where Euclidean geometry can distort distances, averages, and the resulting…

Machine Learning · Statistics 2026-05-07 Alessandro Micheli , Silvia Sapora , Anthea Monod , Samir Bhatt

Sparse tensors are prevalent in real-world applications, often characterized by their large-scale, high-order, and high-dimensional nature. Directly handling raw tensors is impractical due to the significant memory and computational…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-05-24 Zixuan Li , Mingxing Duan , Huizhang Luo , Wangdong Yang , Kenli Li , Keqin Li

Evolutionary multiobjective optimization (EMO) has made significant strides over the past two decades. However, as problem scales and complexities increase, traditional EMO algorithms face substantial performance limitations due to…

Neural and Evolutionary Computing · Computer Science 2025-07-11 Zhenyu Liang , Hao Li , Naiwei Yu , Kebin Sun , Ran Cheng

In this paper, we propose a new framework named Communication Optimal Transport (CommOT) for computing the rate distortion (RD) function. This work is motivated by observing the fact that the transition law and the relative entropy in…

Information Theory · Computer Science 2022-12-21 Shitong Wu , Wenhao Ye , Hao Wu , Huihui Wu , Wenyi Zhang , Bo Bai

The objective in statistical Optimal Transport (OT) is to consistently estimate the optimal transport plan/map solely using samples from the given source and target marginal distributions. This work takes the novel approach of posing…

Machine Learning · Computer Science 2020-11-11 J. Saketha Nath , Pratik Jawanpuria

Cross-city transfer improves prediction in label-scarce cities by leveraging labeled data from other cities, but it becomes challenging when cities adopt incompatible partitions and no ground-truth region correspondences exist. Existing…

Machine Learning · Computer Science 2026-05-12 Yuyao Wang , Min Yang , Meng Chen , Weiming Huang , Yilong Yin , Yongshun Gong