#多项式 #有限域 #FFT

不依赖std tfhe-ark-poly

基于有限域上的FFT的高效多项式算术库

1个不稳定版本

0.4.2 2024年6月6日

#947 in 密码学

Download history 131/week @ 2024-06-02 4/week @ 2024-06-09 643/week @ 2024-06-16 1348/week @ 2024-06-23 200/week @ 2024-06-30 1411/week @ 2024-07-07 1254/week @ 2024-07-14 1167/week @ 2024-07-21 1289/week @ 2024-07-28

5,130每月下载量
4 个crate中使用了 (直接使用 2 个)

MIT/Apache

565KB
12K SLoC

ark-poly

此crate实现了多项式、域(FFT友好子集)的特性和实现,以及这些域的FFT。

多项式

polynomial模块提供了以下特性和方法来定义系数形式的多项式:

  • Polynomial:要求实现者支持多项式的常见操作,如AddSubZero、在一点上的评估、次数等,并定义了将多项式序列化为系数表示和从系数表示反序列化的方法。
  • DenseUVPolynomial:指定一个Polynomial实际上是一个一元多项式。
  • DenseMVPolynomial:指定一个Polynomial实际上是一个多元多项式。

此crate还提供了以下数据结构,实现了这些特性:

  • univariate/DensePolynomial:通过一个包含d + 1个系数的列表表示次数为d的一元多项式。此结构实现了DenseUVPolynomial特性。
  • univariate/SparsePolynomial:通过包含所有非零单项式的列表表示次数为 d 的单项式多项式。这应该只在多项式的大多数系数为零时使用。此结构体实现了 Polynomial 特性(但 不是 DenseUVPolynomial 特性)。
  • multivariate/SparsePolynomial:通过包含所有非零单项式的列表表示多元多项式。

此软件包还提供了 univariate/DenseOrSparsePolynomial 枚举,允许用户抽象底层单项式多项式的类型(密集或稀疏)。

示例

use ark_poly::{
    polynomial::multivariate::{SparsePolynomial, SparseTerm, Term},
    DenseMVPolynomial, Polynomial,
};
use ark_test_curves::bls12_381::Fq;
// Create a multivariate polynomial in 3 variables, with 4 terms:
// f(x_0, x_1, x_2) = 2*x_0^3 + x_0*x_2 + x_1*x_2 + 5
let poly = SparsePolynomial::from_coefficients_vec(
    3,
    vec![
        (Fq::from(2), SparseTerm::new(vec![(0, 3)])),
        (Fq::from(1), SparseTerm::new(vec![(0, 1), (2, 1)])),
        (Fq::from(1), SparseTerm::new(vec![(1, 1), (2, 1)])),
        (Fq::from(5), SparseTerm::new(vec![])),
    ],
);
assert_eq!(poly.evaluate(&vec![Fq::from(2), Fq::from(3), Fq::from(6)]), Fq::from(51));

评估

evaluations 模块提供数据结构来表示拉格朗日形式的单项式多项式。

evaluations 模块还提供了以下特性来定义拉格朗日形式的多元多项式

此软件包提供了一些数据结构来实现这些特性。

示例

use ark_test_curves::bls12_381::Fq;
use ark_poly::{DenseMultilinearExtension, MultilinearExtension, SparseMultilinearExtension};
use ark_poly::{
    polynomial::multivariate::{SparsePolynomial, SparseTerm, Term},
    DenseMVPolynomial, Polynomial,
};
use ark_std::One;

// Create a multivariate polynomial in 3 variables:
// f(x_0, x_1, x_2) = 2*x_0^3 + x_0*x_2 + x_1*x_2 
let f = SparsePolynomial::from_coefficients_vec(
    3,
    vec![
        (Fq::from(2), SparseTerm::new(vec![(0, 3)])),
        (Fq::from(1), SparseTerm::new(vec![(0, 1), (2, 1)])),
        (Fq::from(1), SparseTerm::new(vec![(1, 1), (2, 1)])),
        (Fq::from(0), SparseTerm::new(vec![])),
    ],
);
// g is the multilinear extension of f, defined by the evaluations of f on the Boolean hypercube:
// f(0, 0, 0) = 2*0^3 + 0*0 + 0*0 = 0
// f(1, 0, 0) = 2*1^3 + 0*0 + 0*0 = 2
// ...
// f(1, 1, 1) = 2*1^3 + 1*1 + 1*1 = 4
let g: DenseMultilinearExtension<Fq> = DenseMultilinearExtension::from_evaluations_vec(
    3, 
    vec![
        Fq::from(0),
        Fq::from(2),
        Fq::from(0),
        Fq::from(2),
        Fq::from(0),
        Fq::from(3),
        Fq::from(1),
        Fq::from(4),
    ]
);
// when evaluated at any point within the Boolean hypercube, f and g should be equal
let point_within_hypercube = &vec![Fq::from(0), Fq::from(1), Fq::from(1)];
assert_eq!(f.evaluate(&point_within_hypercube), g.evaluate(&point_within_hypercube));

// We can also define a MLE g'(x_0, x_1, x_2) by providing the list of non-zero evaluations:
let g_prime: SparseMultilinearExtension<Fq> = SparseMultilinearExtension::from_evaluations(
    3,
    &vec![
        (1, Fq::from(2)),
        (3, Fq::from(2)),
        (5, Fq::from(3)),
        (6, Fq::from(1)),
        (7, Fq::from(4)),
    ]
);
// at any random point (X0, X1, X2), g == g' with negligible probability, unless they are the same function
let random_point = &vec![Fq::from(123), Fq::from(456), Fq::from(789)];
assert_eq!(g_prime.evaluate(&random_point), g.evaluate(&random_point));

待办事项

依赖关系

~3.5–4.5MB
~85K SLoC