Skip to content

sigma-py/accupy

Repository files navigation

accupy

Accurate sums and (dot) products for Python.

PyPi Version PyPI pyversions DOI GitHub stars PyPi downloads

Discord

gh-actions codecov Code style: black

Sums

Summing up values in a list can get tricky if the values are floating point numbers; digit cancellation can occur and the result may come out wrong. A classical example is the sum

1.0e16 + 1.0 - 1.0e16

The actual result is 1.0, but in double precision, this will result in 0.0. While in this example the failure is quite obvious, it can get a lot more tricky than that. accupy provides

p, exact, cond = accupy.generate_ill_conditioned_sum(100, 1.0e20)

which, given a length and a target condition number, will produce an array of floating point numbers that is hard to sum up.

Given one or two vectors, accupy can compute the condition of the sum or dot product via

accupy.cond(x)
accupy.cond(x, y)

accupy has the following methods for summation:

  • accupy.kahan_sum(p): Kahan summation

  • accupy.fsum(p): A vectorization wrapper around math.fsum (which uses Shewchuck's algorithm [1] (see also here)).

  • accupy.ksum(p, K=2): Summation in K-fold precision (from [2])

All summation methods sum the first dimension of a multidimensional NumPy array.

Let's compare them.

Accuracy comparison (sum)

As expected, the naive sum performs very badly with ill-conditioned sums; likewise for numpy.sum which uses pairwise summation. Kahan summation not significantly better; this, too, is expected.

Computing the sum with 2-fold accuracy in accupy.ksum gives the correct result if the condition is at most in the range of machine precision; further increasing K helps with worse conditions.

Shewchuck's algorithm in math.fsum always gives the correct result to full floating point precision.

Runtime comparison (sum)

We compare more and more sums of fixed size (above) and larger and larger sums, but a fixed number of them (below). In both cases, the least accurate method is the fastest (numpy.sum), and the most accurate the slowest (accupy.fsum).

Dot products

accupy has the following methods for dot products:

  • accupy.fdot(p): A transformation of the dot product of length n into a sum of length 2n, computed with math.fsum

  • accupy.kdot(p, K=2): Dot product in K-fold precision (from [2])

Let's compare them.

Accuracy comparison (dot)

accupy can construct ill-conditioned dot products with

x, y, exact, cond = accupy.generate_ill_conditioned_dot_product(100, 1.0e20)

With this, the accuracy of the different methods is compared.

As for sums, numpy.dot is the least accurate, followed by instanced of kdot. fdot is provably accurate up into the last digit

Runtime comparison (dot)

NumPy's numpy.dot is much faster than all alternatives provided by accupy. This is because the bookkeeping of truncation errors takes more steps, but mostly because of NumPy's highly optimized dot implementation.

References

  1. Richard Shewchuk, Adaptive Precision Floating-Point Arithmetic and Fast Robust Geometric Predicates, J. Discrete Comput. Geom. (1997), 18(305), 305–363

  2. Takeshi Ogita, Siegfried M. Rump, and Shin'ichi Oishi, Accurate Sum and Dot Product, SIAM J. Sci. Comput. (2006), 26(6), 1955–1988 (34 pages)

Dependencies

accupy needs the C++ Eigen library, provided in Debian/Ubuntu by libeigen3-dev.

Installation

accupy is available from the Python Package Index, so with

pip install accupy

you can install.

Testing

To run the tests, just check out this repository and type

MPLBACKEND=Agg pytest