Skip to main content

Series-reversion polynomial solver with bootstrap and deflation

Project description

geodepoly — Series-Reversion Polynomial Solver (MVP)

About

CI DOI Docs PyPI License: MIT

geodepoly is a small Python package that finds all roots of a complex polynomial using a shift–recenter + truncated series reversion with optional bootstrap iterations, and only falls back to classical iterations when strictly necessary.

This implements the impact-first MVP discussed:

  • Compositional inverse (via Lagrange inversion in coefficient form) around a local recentering point to obtain an analytic series for a nearby root.
  • Bootstrap: update the center by the series estimate and re-expand (typically a few steps).
  • Deflation: synthetic division to peel off roots one-by-one.
  • Safe fallbacks: Halley or Durand–Kerner if series degenerates (multiple root / tiny derivative).

Note: This is a minimal working scaffold you can publish and iterate on. It is self-contained (numpy optional), tested, and provides a SymPy hook.

Install (editable)

pip install -e .

Install (PyPI)

pip install geodepoly==0.1.7.post1

Quickstart

from geodepoly import solve_all

# Coefficients lowest-degree first: a0 + a1 x + ... + aN x^N
coeffs = [1, 0, -7, 6]
roots = solve_all(coeffs, method="hybrid", resum="auto")
print(roots)

CLI

python -m geodepoly.scripts.benchmark --deg 8 --seed 123 --trials 100

Examples

  • SymPy comparison: python examples/sympy_vs_nroots.py
  • JSON bridge round‑trip: python examples/json_bridge_roundtrip.py
  • Multiple root demo: python examples/multiple_root_demo.py
  • Compare methods: python examples/compare_methods.py
  • Resummation effect: python examples/resummation_effect.py
  • Eigenvalues demo: python examples/eigs_demo.py

API

  • series_solve_all(coeffs, max_order=32, boots=3, tol=1e-12, max_deflation=None, verbose=False)
  • series_one_root(coeffs, center=None, max_order=32, boots=3, tol=1e-14)
  • sympy_solve(poly) — lightweight SymPy integration (if SymPy is installed).
  • solve_eigs(A) — eigenvalues via characteristic polynomial (Faddeev–LeVerrier).

How it works (short)

Let p(x) be a degree-n polynomial. We recenter around x = μ and expand q(y) = p(μ + y) = a0 + a1 y + a2 y^2 + .... If a1 ≠ 0, solve q(y)=0 by compositional inversion of F(y) = y + β2 y^2 + β3 y^3 + ... with βk = ak/a1. Lagrange inversion gives the inverse coefficients {g_m} of F, and the nearby root is y ≈ Σ_{m≥1} g_m t^m, with t = -a0/a1. Update μ ← μ + y (bootstrap) and repeat a few times; then deflate and continue.

This repo implements the coefficient formula g_m = (1/m) * [y^{m-1}] (1 / F'(y))^m using truncated series arithmetic. No derivatives of p beyond a1 = q'(0) are used.

Caveats

  • Multiple or nearly-multiple roots are ill-conditioned for any method. We switch to a guarded Durand–Kerner step when |a1| is tiny.
  • Convergence radius depends on the local analytic structure; bootstrap helps.

License

MIT


New in this build

Friendlier API

from geodepoly import solve_poly, solve_all, solve_one
roots = solve_poly(coeffs, method="hybrid", resum="pade")

Methods: hybrid (series seeds + Aberth), aberth, dk, numpy (companion).
Resummation: None, "pade", "borel", "borel-pade".

SymPy integration

from geodepoly.sympy_plugin import sympy_solve
roots = sympy_solve(x**8 - 3*x + 1, method="hybrid", resum="pade")

Mathematica / Maple bridge (JSON CLI)

python bridges/geodepoly_cli.py <<'JSON'
{"coeffs":[-6,11,-6,1],"kwargs":{"method":"hybrid","resum":"pade"}}
JSON

In Mathematica:

payload = ExportString[<|"coeffs"->{-6,11,-6,1},"kwargs"-><|"method"->"hybrid","resum"->"pade"|>|>,"JSON"];
res = RunProcess[{"python","bridges/geodepoly_cli.py"}, "StandardInput"->payload, "StandardOutput"];
ImportString[res, "JSON"]

Benchmarks

geodepoly-bench --degrees 8 --methods hybrid,aberth,dk --trials 50 --out bench_deg8.csv

Aggregate and plot (see docs/assets/):

python scripts/bench_presets.py quick
# or run manually:
geodepoly-bench --degrees 3,5,8,12 --methods hybrid,aberth,dk --trials 10 --out docs/assets/bench.csv --agg_out docs/assets/bench_agg.csv --resum auto
python scripts/plot_bench.py --in docs/assets/bench_agg.csv --out docs/assets

Previews:

Time vs Degree

Residual vs Degree

Paper skeleton

See paper/GeodePoly_MVP.md.

Notebooks

  • Quickstart: notebooks/Quickstart.ipynb
  • Bench summary: notebooks/BenchSummary.ipynb

Bench dataset & GPU roadmap

  • GeodeBench spec: bench/geodebench_spec.md, generator: bench/generate_slices.py
  • GPU roadmap: docs/geode_gpu_spec.md

Hyper-Catalan API (S[t2,t3,...])

Utilities based on the paper's multivariate series:

from geodepoly import evaluate_hyper_catalan, evaluate_quadratic_slice, catalan_number

# Quadratic slice (Catalan series)
t2 = 0.05
alpha_approx = evaluate_quadratic_slice(t2, max_weight=20)
catalan_series = sum(catalan_number(n) * (t2**n) for n in range(12))

# Multivariate evaluation (truncated by weighted degree)
alpha_multi = evaluate_hyper_catalan({2: 0.05, 3: 0.01}, max_weight=12)

See docs/paper_guide.md for how the paper maps onto the codebase.

Cite

If you use this software, please cite:

DOI

CITATION.cff is included at the repo root.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

geodepoly-0.1.7.post1.tar.gz (25.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

geodepoly-0.1.7.post1-py3-none-any.whl (24.1 kB view details)

Uploaded Python 3

File details

Details for the file geodepoly-0.1.7.post1.tar.gz.

File metadata

  • Download URL: geodepoly-0.1.7.post1.tar.gz
  • Upload date:
  • Size: 25.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.6

File hashes

Hashes for geodepoly-0.1.7.post1.tar.gz
Algorithm Hash digest
SHA256 4552c29c20752ee8dab04f9e1567af90db75a8913da3855d376e384294ac1ec7
MD5 74a818e87e25866867f661a1c454c379
BLAKE2b-256 1184e78a158258ab6653ded662c8f2bf62b2d36131f27d4457683940ddb37e0e

See more details on using hashes here.

File details

Details for the file geodepoly-0.1.7.post1-py3-none-any.whl.

File metadata

File hashes

Hashes for geodepoly-0.1.7.post1-py3-none-any.whl
Algorithm Hash digest
SHA256 84d48dfdd1bdff29355caa801bc588591c1616afe4e9865a23566167209ae0b3
MD5 132eebce1e7020a7c5351b8028f27d51
BLAKE2b-256 f107038086182b92fb2398d9b9391454c4741524e704be84fb9bad33f264de92

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page