-
BlackJAX: Composable Bayesian inference in JAX
Authors:
Alberto Cabezas,
Adrien Corenflos,
Junpeng Lao,
Rémi Louf,
Antoine Carnec,
Kaustubh Chaudhari,
Reuben Cohn-Gordon,
Jeremie Coullon,
Wei Deng,
Sam Duffield,
Gerardo Durán-Martín,
Marcin Elantkowski,
Dan Foreman-Mackey,
Michele Gregori,
Carlos Iguaran,
Ravin Kumar,
Martin Lysy,
Kevin Murphy,
Juan Camilo Orduz,
Karm Patel,
Xi Wang,
Rob Zinkov
Abstract:
BlackJAX is a library implementing sampling and variational inference algorithms commonly used in Bayesian computation. It is designed for ease of use, speed, and modularity by taking a functional approach to the algorithms' implementation. BlackJAX is written in Python, using JAX to compile and run NumpPy-like samplers and variational methods on CPUs, GPUs, and TPUs. The library integrates well w…
▽ More
BlackJAX is a library implementing sampling and variational inference algorithms commonly used in Bayesian computation. It is designed for ease of use, speed, and modularity by taking a functional approach to the algorithms' implementation. BlackJAX is written in Python, using JAX to compile and run NumpPy-like samplers and variational methods on CPUs, GPUs, and TPUs. The library integrates well with probabilistic programming languages by working directly with the (un-normalized) target log density function. BlackJAX is intended as a collection of low-level, composable implementations of basic statistical 'atoms' that can be combined to perform well-defined Bayesian inference, but also provides high-level routines for ease of use. It is designed for users who need cutting-edge methods, researchers who want to create complex sampling methods, and people who want to learn how these work.
△ Less
Submitted 22 February, 2024; v1 submitted 16 February, 2024;
originally announced February 2024.
-
Wireless Content Caching for Small Cell and D2D Networks
Authors:
Maria Gregori,
Jesús Gómez-Vilardebó,
Javier Matamoros,
Deniz Gündüz
Abstract:
The fifth generation wireless networks must provide fast and reliable connectivity while co** with the ongoing traffic growth. It is of paramount importance that the required resources, such as energy and bandwidth, do not scale with traffic. While the aggregate network traffic is growing at an unprecedented rate, users tend to request the same popular contents at different time instants. Theref…
▽ More
The fifth generation wireless networks must provide fast and reliable connectivity while co** with the ongoing traffic growth. It is of paramount importance that the required resources, such as energy and bandwidth, do not scale with traffic. While the aggregate network traffic is growing at an unprecedented rate, users tend to request the same popular contents at different time instants. Therefore, caching the most popular contents at the network edge is a promising solution to reduce the traffic and the energy consumption over the backhaul links. In this paper, two scenarios are considered, where caching is performed either at a small base station, or directly at the user terminals, which communicate using \ac{D2D} communications. In both scenarios, joint design of the transmission and caching policies is studied when the user demands are known in advance. This joint design offers two different caching gains, namely, the \textit{pre-downloading} and \textit{local caching gains}. It is shown that the finite cache capacity limits the attainable gains, and creates an inherent tradeoff between the two types of gains. In this context, a continuous time optimization problem is formulated to determine the optimal transmission and caching policies that minimize a generic cost function, such as energy, bandwidth, or throughput. The jointly optimal solution is obtained by demonstrating that caching files at a constant rate is optimal, which allows to reformulate the problem as a finite-dimensional convex program. The numerical results show that the proposed joint transmission and caching policy dramatically reduces the total cost, which is particularised to the total energy consumption at the \ac{MBS}, as well as to the total economical cost for the service provider, when users demand economical incentives for delivering content to other users over the D2D links.
△ Less
Submitted 14 March, 2016;
originally announced March 2016.
-
On the precoder design of a wireless energy harvesting node in linear vector Gaussian channels with arbitrary input distribution
Authors:
Maria Gregori,
Miquel Payaró
Abstract:
A Wireless Energy Harvesting Node (WEHN) operating in linear vector Gaussian channels with arbitrarily distributed input symbols is considered in this paper. The precoding strategy that maximizes the mutual information along N independent channel accesses is studied under non-causal knowledge of the channel state and harvested energy (commonly known as offline approach). It is shown that, at each…
▽ More
A Wireless Energy Harvesting Node (WEHN) operating in linear vector Gaussian channels with arbitrarily distributed input symbols is considered in this paper. The precoding strategy that maximizes the mutual information along N independent channel accesses is studied under non-causal knowledge of the channel state and harvested energy (commonly known as offline approach). It is shown that, at each channel use, the left singular vectors of the precoder are equal to the eigenvectors of the Gram channel matrix. Additionally, an expression that relates the optimal singular values of the precoder with the energy harvesting profile through the Minimum Mean-Square Error (MMSE) matrix is obtained. Then, the specific situation in which the right singular vectors of the precoder are set to the identity matrix is considered. In this scenario, the optimal offline power allocation, named Mercury Water-Flowing, is derived and an intuitive graphical representation is presented. Two optimal offline algorithms to compute the Mercury Water- Flowing solution are proposed and an exhaustive study of their computational complexity is performed. Moreover, an online algorithm is designed, which only uses causal knowledge of the harvested energy and channel state. Finally, the achieved mutual information is evaluated through simulation.
△ Less
Submitted 28 January, 2013;
originally announced January 2013.
-
Energy-efficient transmission for wireless energy harvesting nodes
Authors:
Maria Gregori,
Miquel Payaró
Abstract:
Energy harvesting is increasingly gaining importance as a means to charge battery powered devices such as sensor nodes. Efficient transmission strategies must be developed for Wireless Energy Harvesting Nodes (WEHNs) that take into account both the availability of energy and data in the node. We consider a scenario where data and energy packets arrive to the node where the time instants and amount…
▽ More
Energy harvesting is increasingly gaining importance as a means to charge battery powered devices such as sensor nodes. Efficient transmission strategies must be developed for Wireless Energy Harvesting Nodes (WEHNs) that take into account both the availability of energy and data in the node. We consider a scenario where data and energy packets arrive to the node where the time instants and amounts of the packets are known (offline approach). In this paper, the best data transmission strategy is found for a finite battery capacity WEHN that has to fulfill some Quality of Service (QoS) constraints, as well as the energy and data causality constraints. As a result of our analysis, we can state that losing energy due to overflows of the battery is inefficient unless there is no more data to transmit and that the problem may not have a feasible solution. Finally, an algorithm that computes the data transmission curve minimizing the total transmission time that satisfies the aforementioned constraints has been developed.
△ Less
Submitted 5 December, 2012;
originally announced December 2012.